Linted

Author: angelosantos4

nodestream/cli/operations/initialize_logger.py

@@ -5,8 +5,9 @@
 from pythonjsonlogger.jsonlogger import JsonFormatter
 
 from nodestream.cli.commands.nodestream_command import NodestreamCommand
-from nodestream.metrics import Metrics
 from nodestream.logging_metrics import MetricsLoggingHandler
+from nodestream.metrics import Metrics
+
 from .operation import Operation
 
 
@@ -33,7 +34,7 @@ def record_factory(*args, **kwargs):
     )
     logger = logging.getLogger()  # Configure the root logger.
     logger.handlers[0].setFormatter(formatter)
-    
+
     # Add metrics handler to track log level counts
     metrics_handler = MetricsLoggingHandler()
     logger.addHandler(metrics_handler)

nodestream/logging_metrics.py

@@ -33,4 +33,4 @@ def emit(self, record: logging.LogRecord) -> None:
             elif record.levelno >= logging.DEBUG:
                 self.metrics.increment(LOG_DEBUG_COUNT)
         except Exception:
-            self.handleError(record)  # Handles any errors during metric increment
+            self.handleError(record)  # Handles any errors during metric increment

nodestream/metrics.py

@@ -94,10 +94,18 @@ def tick(self):
 FATAL_ERRORS = Metric("fatal_errors", "Number of fatal errors", accumulate=True)
 
 # Logging metrics
-LOG_ERROR_COUNT = Metric("log_error_count", "Number of error log messages", accumulate=True)
-LOG_WARNING_COUNT = Metric("log_warning_count", "Number of warning log messages", accumulate=True)
-LOG_INFO_COUNT = Metric("log_info_count", "Number of info log messages", accumulate=True)
-LOG_DEBUG_COUNT = Metric("log_debug_count", "Number of debug log messages", accumulate=True)
+LOG_ERROR_COUNT = Metric(
+    "log_error_count", "Number of error log messages", accumulate=True
+)
+LOG_WARNING_COUNT = Metric(
+    "log_warning_count", "Number of warning log messages", accumulate=True
+)
+LOG_INFO_COUNT = Metric(
+    "log_info_count", "Number of info log messages", accumulate=True
+)
+LOG_DEBUG_COUNT = Metric(
+    "log_debug_count", "Number of debug log messages", accumulate=True
+)
 
 NODES_UPSERTED = Metric(
     "nodes_upserted", "Number of nodes upserted to the graph", accumulate=True