telemetry/__main__

example/__main__.py

@@ -0,0 +1,9 @@
+from argparse import ArgumentParser
+
+parser = ArgumentParser(description="telemetry python wrapper")
+parser.add_argument("bob", help="python module to run")
+
+args = parser.parse_args()
+
+if __name__ == '__main__':
+    print(args.bob)

telemetry/__init__.py

@@ -31,6 +31,12 @@ def is_logger(meter: Telemeter):
     return hasattr(loggers, meter.__name__)
 
 
+def env_loggers():
+    for meter in REGISTRY:
+        if is_configured(meter) and is_logger(meter):
+            yield meter
+
+
 def load_env(logger=None):
     global ACTIVE_TELEMETERS
     ACTIVE_TELEMETERS = []  # TODO scan if each telemeter is loaded already.

telemetry/__main__.py

@@ -0,0 +1,22 @@
+import sys
+import runpy
+import logging
+from argparse import ArgumentParser
+
+import telemetry
+
+parser = ArgumentParser(description="telemetry python wrapper")
+parser.add_argument("module", help="python module to run")
+args, _ = parser.parse_known_args()
+
+main_logger = logging.getLogger()
+telem_logger = logging.getLogger("telemetry")
+
+loggers = telemetry.env_loggers()
+for logger in loggers:
+    logging.info(f"loading {logger.__name__}")
+    main_logger.addHandler(logger)
+    telem_logger.addHandler(logger)
+
+sys.argv = sys.argv[1:]
+runpy.run_module(args.module, run_name="__main__")