wip

mila-iqia · Nov 5, 2024 · 1a0a280 · 1a0a280
1 parent 88f1dc3
commit 1a0a280
Showing 1 changed file with 32 additions and 22 deletions.
diff --git a/sarc/logging.py b/sarc/logging.py
@@ -79,35 +79,45 @@ def setupLogging():
     from opentelemetry.sdk._logs.export import BatchLogRecordProcessor
     from opentelemetry.sdk.resources import Resource
 
-    logger_provider = LoggerProvider(
-        resource=Resource.create(
-            {
-                "service.name": config().loki.service_name,
-                "service.instance.id": os.uname().nodename,
-            }
-        ),
-    )
-    set_logger_provider(logger_provider)
-
-    endpoint = config().loki.endpoint
-
-    otlp_exporter = OTLPLogExporter(endpoint)
-    logger_provider.add_log_record_processor(BatchLogRecordProcessor(otlp_exporter))
-    handler = LoggingHandler(level=logging.NOTSET, logger_provider=logger_provider)
-
     logging_levels = {
         "DEBUG": logging.DEBUG,
         "INFO": logging.INFO,
         "WARNING": logging.WARNING,
         "ERROR": logging.ERROR,
         "CRITICAL": logging.CRITICAL,
     }
-    log_level = logging_levels.get(config().logging.log_level, logging.INFO)
-    logging.basicConfig(
-        handlers=[handler, logging.StreamHandler()],
-        format="%(asctime)-15s::%(levelname)s::%(name)s::%(message)s",
-        level=log_level,
-    )
+    log_level = logging.WARNING # default log level
+    if config().logging:
+        log_level = logging_levels.get(config().logging.log_level, logging.WARNING)
+
+    if (config().loki):
+        logger_provider = LoggerProvider(
+            resource=Resource.create(
+                {
+                    "service.name": config().loki.service_name,
+                    "service.instance.id": os.uname().nodename,
+                }
+            ),
+        )
+        set_logger_provider(logger_provider)
+
+        endpoint = config().loki.endpoint
+
+        otlp_exporter = OTLPLogExporter(endpoint)
+        logger_provider.add_log_record_processor(BatchLogRecordProcessor(otlp_exporter))
+        handler = LoggingHandler(level=logging.NOTSET, logger_provider=logger_provider)
+
+        logging.basicConfig(
+            handlers=[handler, logging.StreamHandler()],
+            format="%(asctime)-15s::%(levelname)s::%(name)s::%(message)s",
+            level=log_level,
+        )
+    else:
+        logging.basicConfig(
+            handlers=[logging.StreamHandler()],
+            format="%(asctime)-15s::%(levelname)s::%(name)s::%(message)s",
+            level=logging.INFO,
+        )
 
     # logging.info("SARC Test info log")
     # logging.debug("SARC Test debug log")