refactor(BasicFormatter): improve trace ID handling in BasicFormatter

Author: Msameim181

src/chromatrace/logging_settings.py

@@ -99,6 +99,7 @@ def _format_process_id(self, process_id: int) -> str:
     def format(self, record):
         record_copy = copy(record)
         record_copy.__dict__["process"] = f'PID:{getattr(record_copy, "process", 0)}'
+        trace_id = getattr(record_copy, "trace_id", "NAN")
         if self.colored:
             record_copy.__dict__["levelname"] = self._color_level_name(
                 level_name=record_copy.levelname, level_no=record_copy.levelno
@@ -108,10 +109,8 @@ def format(self, record):
             record_copy.__dict__["process"] = self._format_process_id(
                 record_copy.process
             )
-            if not (record_copy.trace_id == "NAN" and self.remove_nan_trace):
-                record_copy.__dict__["trace_id"] = self._format_trace_id(
-                    record_copy.trace_id
-                )
+            if not (trace_id == "NAN" and self.remove_nan_trace):
+                record_copy.__dict__["trace_id"] = self._format_trace_id(trace_id)
 
         record_copy.__dict__["msg"] = (
             self.message_splitter + record_copy.__dict__["msg"]