Update utils/logger.py

utils/logger.py

@@ -1,3 +1,4 @@
+#!/usr/bin/env python3
 """
 Logging Management Module
 ========================
@@ -28,7 +29,8 @@
 import traceback
 from functools import wraps
 import time
-from typing import Dict, List 
+from typing import Dict, List
+
 
 class ColoredFormatter(logging.Formatter):
     """Custom formatter with colors for console output"""
@@ -50,6 +52,7 @@ def format(self, record):
         
         return super().format(record)
 
+
 class BackgroundFXLogger:
     """Main logger class for BackgroundFX Pro"""
     
@@ -82,6 +85,13 @@ def __init__(self,
         # Performance tracking
         self.performance_data = {}
         self.start_times = {}
+    
+    def __getattr__(self, name):
+        """
+        Delegate unknown attributes/methods to the underlying stdlib logger.
+        This makes BackgroundFXLogger behave like logging.Logger where needed.
+        """
+        return getattr(self.logger, name)
         
     def _setup_handlers(self):
         """Setup logging handlers for console and file output"""
@@ -106,7 +116,8 @@ def _setup_handlers(self):
             file_handler = logging.handlers.RotatingFileHandler(
                 main_log_file,
                 maxBytes=10*1024*1024,  # 10MB
-                backupCount=5
+                backupCount=5,
+                encoding="utf-8"
             )
             file_handler.setLevel(self.level)
             
@@ -123,7 +134,8 @@ def _setup_handlers(self):
             error_handler = logging.handlers.RotatingFileHandler(
                 error_log_file,
                 maxBytes=5*1024*1024,  # 5MB
-                backupCount=3
+                backupCount=3,
+                encoding="utf-8"
             )
             error_handler.setLevel(logging.ERROR)
             error_handler.setFormatter(file_formatter)
@@ -247,8 +259,11 @@ def _save_performance_data(self):
             # Load existing data
             existing_data = {}
             if self.performance_log_file.exists():
-                with open(self.performance_log_file, 'r') as f:
-                    existing_data = json.load(f)
+                with open(self.performance_log_file, 'r', encoding="utf-8") as f:
+                    try:
+                        existing_data = json.load(f)
+                    except json.JSONDecodeError:
+                        existing_data = {}
             
             # Merge with new data
             existing_data.update(self.performance_data)
@@ -260,7 +275,7 @@ def _save_performance_data(self):
                 existing_data = {k: existing_data[k] for k in keep_keys}
             
             # Save updated data
-            with open(self.performance_log_file, 'w') as f:
+            with open(self.performance_log_file, 'w', encoding="utf-8") as f:
                 json.dump(existing_data, f, indent=2)
                 
         except Exception as e:
@@ -271,7 +286,7 @@ def get_log_files(self) -> Dict[str, str]:
         return {
             'main_log': str(self.logs_dir / "backgroundfx.log"),
             'error_log': str(self.logs_dir / "errors.log"), 
-            'performance_log': str(self.logs_dir / "performance.json"),
+            'performance_log': str(self.performance_log_file),
             'logs_directory': str(self.logs_dir)
         }
     
@@ -283,13 +298,13 @@ def get_recent_logs(self, lines: int = 50) -> Dict[str, List[str]]:
             # Main log
             main_log_file = self.logs_dir / "backgroundfx.log"
             if main_log_file.exists():
-                with open(main_log_file, 'r') as f:
+                with open(main_log_file, 'r', encoding="utf-8") as f:
                     logs['main'] = f.readlines()[-lines:]
             
             # Error log
             error_log_file = self.logs_dir / "errors.log"
             if error_log_file.exists():
-                with open(error_log_file, 'r') as f:
+                with open(error_log_file, 'r', encoding="utf-8") as f:
                     logs['errors'] = f.readlines()[-lines:]
             
         except Exception as e:
@@ -297,9 +312,11 @@ def get_recent_logs(self, lines: int = 50) -> Dict[str, List[str]]:
         
         return logs
 
+
 # Global logger instance
 _global_logger: Optional[BackgroundFXLogger] = None
 
+
 def setup_logging(logs_dir: str = "LOGS", 
                  level: int = logging.INFO,
                  console_output: bool = True,
@@ -318,6 +335,16 @@ def setup_logging(logs_dir: str = "LOGS",
     
     return _global_logger
 
+
+# --- Backward-compat alias for legacy imports ---
+def setup_logger(*args, **kwargs):
+    """
+    Alias for old code: `from utils.logger import setup_logger`.
+    Behaves the same as setup_logging and returns a BackgroundFXLogger.
+    """
+    return setup_logging(*args, **kwargs)
+
+
 def get_logger(name: str = None) -> BackgroundFXLogger:
     """Get logger instance"""
     if _global_logger is None:
@@ -336,6 +363,7 @@ def get_logger(name: str = None) -> BackgroundFXLogger:
     
     return _global_logger
 
+
 def log_function_call(logger: BackgroundFXLogger = None):
     """Decorator to log function calls with timing"""
     if logger is None:
@@ -362,6 +390,7 @@ def wrapper(*args, **kwargs):
         return wrapper
     return decorator
 
+
 def log_processing_pipeline():
     """Decorator for logging processing pipeline steps"""
     logger = get_logger()
@@ -384,6 +413,7 @@ def wrapper(*args, **kwargs):
         return wrapper
     return decorator
 
+
 # Convenience functions
 def log_info(message: str, **kwargs):
     """Quick info logging"""
@@ -401,10 +431,25 @@ def log_debug(message: str, **kwargs):
     """Quick debug logging"""
     get_logger().debug(message, **kwargs)
 
+
 # Initialize logging on module import
 if _global_logger is None:
     try:
         setup_logging()
         log_info("✅ Logging system initialized")
     except Exception as e:
-        print(f"⚠️ Failed to initialize logging: {e}")
+        print(f"⚠️ Failed to initialize logging: {e}")
+
+
+__all__ = [
+    "BackgroundFXLogger",
+    "setup_logging",
+    "setup_logger",   # alias for legacy code
+    "get_logger",
+    "log_function_call",
+    "log_processing_pipeline",
+    "log_info",
+    "log_error",
+    "log_warning",
+    "log_debug",
+]