move logger onto settings, only built once

sidecar/app/query/base.py

@@ -8,7 +8,6 @@
 import loguru
 
 from ..helpers import Role
-from ..logger import get_logger
 from ..settings import get_settings
 from .status import Status, StatusHistory
 from .step import Step
@@ -41,9 +40,8 @@ class Query:
 
     def __post_init__(self):
         settings = get_settings()
-        _logger = get_logger()
 
-        self.logger = _logger.bind(task=self.query_id)
+        self.logger = settings.logger.bind(task=self.query_id)
         self.role = settings.role
 
         self._status_history = StatusHistory(

sidecar/app/settings.py

@@ -1,7 +1,12 @@
+from __future__ import annotations
+
+import sys
 from functools import lru_cache
 from pathlib import Path
 from typing import Annotated, Any
 
+import loguru
+from loguru import logger
 from pydantic.functional_validators import BeforeValidator
 from pydantic_settings import BaseSettings
 
@@ -19,24 +24,45 @@ class Settings(BaseSettings):
     role: Role
     helper_port: int
     _helpers: dict[Role, Helper]
+    _logger: Any  # cannot use loguru.Logger here because pydantic tries to import it
 
     def model_post_init(self, __context) -> None:
         self._helpers = load_helpers_from_network_config(self.network_config_path)
+        self._logger = logger
+        self._logger.remove()
+        max_role_str_len = max(len(role.name) for role in Role)
+        role_str = f"{self.role.name.replace('_', ' ').title():>{max_role_str_len}}"
+
+        logger_format = (
+            "<green>{time:YYYY-MM-DD HH:mm:ss}</green> | "
+            "<level>{level: <8}</level> | "
+            "<blue>{extra[role]}</blue> - <level>{message}</level>"
+        )
+        self._logger.configure(extra={"role": role_str})
+        self._logger.add(
+            sys.stderr,
+            level="INFO",
+            format=logger_format,
+        )
+
+    @property
+    def logger(self) -> loguru.Logger:
+        return self._logger
 
     @property
-    def helper(self):
+    def helper(self) -> Helper:
         return self._helpers[self.role]
 
     @property
-    def helpers(self):
+    def helpers(self) -> dict[Role, Helper]:
         return self._helpers
 
     @property
-    def status_dir_path(self):
+    def status_dir_path(self) -> Path:
         return self.root_path / Path("status")
 
     @property
-    def log_dir_path(self):
+    def log_dir_path(self) -> Path:
         return self.root_path / Path("logs")