Yusarina
c31d25dd01
You can choose between errors, warning, info or full debug, errors will always log to ensure we don't have silent failures with debug on or off.
51 lines
1.8 KiB
Python
51 lines
1.8 KiB
Python
import logging
|
|
import traceback
|
|
from typing import Optional, Any
|
|
from bpy.types import Context
|
|
|
|
logger = logging.getLogger('avatar_toolkit')
|
|
_original_error = logger.error
|
|
|
|
def configure_logging(enabled: bool = False, level: str = "WARNING") -> None:
|
|
"""Configure logging for Avatar Toolkit """
|
|
level_map = {
|
|
"DEBUG": logging.DEBUG,
|
|
"INFO": logging.INFO,
|
|
"WARNING": logging.WARNING,
|
|
"ERROR": logging.ERROR
|
|
}
|
|
|
|
log_level = level_map.get(level, logging.WARNING)
|
|
|
|
if enabled:
|
|
logger.setLevel(log_level)
|
|
else:
|
|
logger.setLevel(logging.ERROR) # We should still log errors when logging is disabled so we don't have silent failures
|
|
|
|
for handler in logger.handlers[:]:
|
|
logger.removeHandler(handler)
|
|
|
|
if enabled:
|
|
handler = logging.StreamHandler()
|
|
handler.setLevel(log_level)
|
|
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
|
|
handler.setFormatter(formatter)
|
|
logger.addHandler(handler)
|
|
|
|
def error_with_traceback(msg, *args, **kwargs):
|
|
if kwargs.get('exc_info', False) or isinstance(msg, Exception):
|
|
full_msg = f"{msg}\n{traceback.format_exc()}"
|
|
_original_error(full_msg, *args, **{**kwargs, 'exc_info': False})
|
|
else:
|
|
_original_error(msg, *args, **kwargs)
|
|
|
|
logger.error = error_with_traceback
|
|
|
|
def update_logging_state(self: Any, context: Context) -> None:
|
|
"""Update logging state based on user preference"""
|
|
from .addon_preferences import save_preference
|
|
enabled = self.enable_logging
|
|
level = self.log_level if hasattr(self, "log_level") else "WARNING"
|
|
save_preference("enable_logging", enabled)
|
|
configure_logging(enabled, level)
|