-
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathbasic_log.py
More file actions
63 lines (54 loc) · 2.33 KB
/
basic_log.py
File metadata and controls
63 lines (54 loc) · 2.33 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
import logging
from pythonlogs.core.log_utils import cleanup_logger_handlers, get_format, get_level, get_timezone_function
from pythonlogs.core.memory_utils import register_logger_weakref
from pythonlogs.core.settings import get_log_settings
from pythonlogs.core.thread_safety import auto_thread_safe
@auto_thread_safe(["init"])
class BasicLog:
"""Basic logger with context manager support for automatic resource cleanup."""
def __init__(
self,
level: str | None = None,
name: str | None = None,
encoding: str | None = None,
datefmt: str | None = None,
timezone: str | None = None,
showlocation: bool | None = None,
):
_settings = get_log_settings()
self.level = get_level(level or _settings.level)
self.appname = name or _settings.appname
self.encoding = encoding or _settings.encoding
self.datefmt = datefmt or _settings.date_format
self.timezone = timezone or _settings.timezone
self.showlocation = showlocation or _settings.show_location
self.logger = None
def init(self):
logger = logging.getLogger(self.appname)
if logger.level != self.level:
logger.setLevel(self.level)
logging.Formatter.converter = get_timezone_function(self.timezone)
_format = get_format(self.showlocation, self.appname, self.timezone)
# Only add handler if logger doesn't have any handlers
if not logger.handlers:
handler = logging.StreamHandler()
formatter = logging.Formatter(_format, datefmt=self.datefmt)
handler.setFormatter(formatter)
logger.addHandler(handler)
self.logger = logger
# Register weak reference for memory tracking
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)