Move HandlerMode definition

hivemind/utils/logging.py

@@ -34,6 +34,9 @@ class TextStyle:
         _codes.update({_name: "" for _name in list(_codes) if _name.isupper()})
 
 
+_PACKAGE_NAME = __name__.split(".")[0]
+
+
 class CustomFormatter(logging.Formatter):
     """
     A formatter that allows a log time and caller info to be overridden via
@@ -56,7 +59,7 @@ def format(self, record: logging.LogRecord) -> str:
 
         if not hasattr(record, "caller"):
             module_path = record.name.split(".")
-            if module_path[0] == "hivemind":
+            if module_path[0] == _PACKAGE_NAME:
                 module_path = module_path[1:]
             record.caller = f"{'.'.join(module_path)}.{record.funcName}:{record.lineno}"
 
@@ -68,7 +71,11 @@ def format(self, record: logging.LogRecord) -> str:
         return super().format(record)
 
 
-_PACKAGE_NAME = __name__.split(".")[0]
+class HandlerMode(Enum):
+    NOWHERE = 0
+    IN_HIVEMIND = 1
+    IN_ROOT_LOGGER = 2
+
 
 _init_lock = threading.RLock()
 _current_mode = HandlerMode.NOWHERE  # This is the initial state before module initialization but not an actual default
@@ -116,12 +123,6 @@ def _disable_default_handler(name: str) -> None:
     logger.setLevel(logging.NOTSET)
 
 
-class HandlerMode(Enum):
-    NOWHERE = 0
-    IN_HIVEMIND = 1
-    IN_ROOT_LOGGER = 2
-
-
 def use_hivemind_log_handler(where: Union[HandlerMode, str]) -> None:
     """
     Choose loggers where the default hivemind log handler is applied. Options for the ``where`` argument are: