V6.0.0

Author: ddc

pythonLogs/core/factory.py

@@ -398,7 +398,26 @@ def create_timed_rotating_logger(
 
 
 # Public API wrapper classes - act like logging.Logger with context manager support
-class BasicLog:
+class _LoggerMixin:
+    """Mixin providing common logger wrapper functionality with context manager support."""
+
+    _logger: logging.Logger
+
+    def __getattr__(self, name: str):
+        """Delegate attribute access to the underlying logger."""
+        return getattr(self._logger, name)
+
+    def __enter__(self):
+        """Context manager entry."""
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        """Context manager exit with automatic cleanup."""
+        cleanup_logger_handlers(self._logger)
+        return False
+
+
+class BasicLog(_LoggerMixin):
     """Basic logger wrapper that acts like logging.Logger with context manager support.
 
     Usage:
@@ -430,21 +449,8 @@ def __init__(
         )
         self._name = name or get_log_settings().appname
 
-    def __getattr__(self, name: str):
-        """Delegate attribute access to the underlying logger."""
-        return getattr(self._logger, name)
-
-    def __enter__(self):
-        """Context manager entry."""
-        return self
-
-    def __exit__(self, exc_type, exc_val, exc_tb):
-        """Context manager exit with automatic cleanup."""
-        cleanup_logger_handlers(self._logger)
-        return False
 
-
-class SizeRotatingLog:
+class SizeRotatingLog(_LoggerMixin):
     """Size-based rotating logger wrapper that acts like logging.Logger with context manager support.
 
     Usage:
@@ -486,21 +492,8 @@ def __init__(
         )
         self._name = name or get_log_settings().appname
 
-    def __getattr__(self, name: str):
-        """Delegate attribute access to the underlying logger."""
-        return getattr(self._logger, name)
-
-    def __enter__(self):
-        """Context manager entry."""
-        return self
-
-    def __exit__(self, exc_type, exc_val, exc_tb):
-        """Context manager exit with automatic cleanup."""
-        cleanup_logger_handlers(self._logger)
-        return False
-
 
-class TimedRotatingLog:
+class TimedRotatingLog(_LoggerMixin):
     """Time-based rotating logger wrapper that acts like logging.Logger with context manager support.
 
     Usage:
@@ -546,19 +539,6 @@ def __init__(
         )
         self._name = name or get_log_settings().appname
 
-    def __getattr__(self, name: str):
-        """Delegate attribute access to the underlying logger."""
-        return getattr(self._logger, name)
-
-    def __enter__(self):
-        """Context manager entry."""
-        return self
-
-    def __exit__(self, exc_type, exc_val, exc_tb):
-        """Context manager exit with automatic cleanup."""
-        cleanup_logger_handlers(self._logger)
-        return False
-
 
 # Convenience functions
 def clear_logger_registry() -> None:

pythonLogs/core/log_utils.py

@@ -3,6 +3,7 @@
 import errno
 from functools import lru_cache
 import gzip
+import logging
 import logging.handlers
 import os
 from pathlib import Path
@@ -14,6 +15,29 @@
 from typing import Callable, Optional, Set
 from zoneinfo import ZoneInfo
 
+
+class RotatingLogMixin:
+    """Mixin providing common rotating logger functionality with context manager support."""
+
+    logger: logging.Logger | None
+
+    def __enter__(self):
+        """Context manager entry."""
+        if not hasattr(self, 'logger') or self.logger is None:
+            self.init()
+        return self.logger
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        """Context manager exit with automatic cleanup."""
+        if hasattr(self, 'logger'):
+            cleanup_logger_handlers(self.logger)
+
+    @staticmethod
+    def cleanup_logger(logger: logging.Logger) -> None:
+        """Static method for cleaning up logger resources."""
+        cleanup_logger_handlers(logger)
+
+
 # Global cache for checked directories with thread safety and size limits
 _checked_directories: Set[str] = set()
 _directory_lock = threading.Lock()

pythonLogs/size_rotating.py

@@ -3,9 +3,9 @@
 from pathlib import Path
 from pythonLogs.core.constants import MB_TO_BYTES
 from pythonLogs.core.log_utils import (
+    RotatingLogMixin,
     check_directory_permissions,
     check_filename_instance,
-    cleanup_logger_handlers,
     get_level,
     get_log_path,
     get_logger_and_formatter,
@@ -21,7 +21,7 @@
 
 
 @auto_thread_safe(['init'])
-class SizeRotatingLog:
+class SizeRotatingLog(RotatingLogMixin):
     """Size-based rotating logger with context manager support for automatic resource cleanup."""
 
     def __init__(
@@ -86,22 +86,6 @@ def init(self):
         register_logger_weakref(logger)
         return logger
 
-    def __enter__(self):
-        """Context manager entry."""
-        if not hasattr(self, 'logger') or self.logger is None:
-            self.init()
-        return self.logger
-
-    def __exit__(self, exc_type, exc_val, exc_tb):
-        """Context manager exit with automatic cleanup."""
-        if hasattr(self, 'logger'):
-            cleanup_logger_handlers(self.logger)
-
-    @staticmethod
-    def cleanup_logger(logger: logging.Logger) -> None:
-        """Static method for cleaning up logger resources."""
-        cleanup_logger_handlers(logger)
-
 
 class GZipRotatorSize:
     def __init__(self, dir_logs: str, daystokeep: int):

pythonLogs/timed_rotating.py

@@ -1,9 +1,9 @@
 import logging.handlers
 import os
 from pythonLogs.core.log_utils import (
+    RotatingLogMixin,
     check_directory_permissions,
     check_filename_instance,
-    cleanup_logger_handlers,
     get_level,
     get_log_path,
     get_logger_and_formatter,
@@ -17,7 +17,7 @@
 
 
 @auto_thread_safe(['init'])
-class TimedRotatingLog:
+class TimedRotatingLog(RotatingLogMixin):
     """
     Time-based rotating logger with context manager support for automatic resource cleanup.
 
@@ -94,22 +94,6 @@ def init(self):
         register_logger_weakref(logger)
         return logger
 
-    def __enter__(self):
-        """Context manager entry."""
-        if not hasattr(self, 'logger') or self.logger is None:
-            self.init()
-        return self.logger
-
-    def __exit__(self, exc_type, exc_val, exc_tb):
-        """Context manager exit with automatic cleanup."""
-        if hasattr(self, 'logger'):
-            cleanup_logger_handlers(self.logger)
-
-    @staticmethod
-    def cleanup_logger(logger: logging.Logger) -> None:
-        """Static method for cleaning up logger resources."""
-        cleanup_logger_handlers(logger)
-
 
 class GZipRotatorTimed:
     def __init__(self, dir_logs: str, days_to_keep: int):