first commit

logger.py

@@ -0,0 +1,145 @@
+"""
+Mô-đun theo dõi và nhật ký - Hệ thống nhật ký cấp sản xuất
+
+Bản quyền © 2026 Công ty TNHH Công nghệ An ninh mạng Huyễn Thành Tân Cương (Công nghệ Huyễn Thành)
+Tác giả: Huyễn Thành
+"""
+import logging
+import logging.handlers
+import os
+from datetime import datetime
+from typing import Optional
+
+LOG_DIR = "logs"
+os.makedirs(LOG_DIR, exist_ok=True)
+
+# Cấu hình tệp nhật ký
+LOG_FILE = os.path.join(LOG_DIR, f"novel_tool_{datetime.now().strftime('%Y%m%d')}.log")
+ERROR_LOG_FILE = os.path.join(LOG_DIR, f"errors_{datetime.now().strftime('%Y%m%d')}.log")
+
+
+def setup_logger(
+    name: str,
+    log_level: int = logging.INFO,
+    log_to_file: bool = True,
+    force_reconfigure: bool = False
+) -> logging.Logger:
+    """
+    Khởi tạo logger cấp sản xuất
+    
+    Args:
+        name: Tên logger
+        log_level: Cấp độ nhật ký
+        log_to_file: Có xuất ra tệp hay không
+        force_reconfigure: Có bắt buộc cấu hình lại không (xóa các handler cũ)
+    
+    Returns:
+        Biến logger đã được cấu hình
+    """
+    logger = logging.getLogger(name)
+    logger.setLevel(log_level)
+    
+    # Nếu bắt buộc cấu hình lại, xóa bỏ handler hiện có
+    if force_reconfigure:
+        for handler in logger.handlers[:]:
+            logger.removeHandler(handler)
+    # Tránh thêm handler lặp lại
+    elif logger.handlers:
+        return logger
+    
+    # Định dạng đầu ra bảng điều khiển
+    console_formatter = logging.Formatter(
+        '[%(asctime)s] [%(name)s] [%(levelname)s] %(message)s',
+        datefmt='%Y-%m-%d %H:%M:%S'
+    )
+    
+    console_handler = logging.StreamHandler()
+    console_handler.setLevel(logging.INFO)
+    console_handler.setFormatter(console_formatter)
+    logger.addHandler(console_handler)
+    
+    # Đầu ra tệp (nếu được bật)
+    if log_to_file:
+        file_formatter = logging.Formatter(
+            '[%(asctime)s] [%(name)s] [%(levelname)s] [%(filename)s:%(lineno)d] %(message)s',
+            datefmt='%Y-%m-%d %H:%M:%S'
+        )
+        
+        # Nhật ký chung
+        file_handler = logging.handlers.RotatingFileHandler(
+            LOG_FILE,
+            maxBytes=10*1024*1024,  # 10MB
+            backupCount=5,
+            encoding='utf-8'
+        )
+        file_handler.setLevel(logging.DEBUG)
+        file_handler.setFormatter(file_formatter)
+        logger.addHandler(file_handler)
+        
+        # Nhật ký lỗi - Luôn thêm vào, đảm bảo chắc chắn sẽ thu thập được nhật ký cấp độ ERROR
+        error_handler = logging.handlers.RotatingFileHandler(
+            ERROR_LOG_FILE,
+            maxBytes=10*1024*1024,
+            backupCount=5,
+            encoding='utf-8'
+        )
+        error_handler.setLevel(logging.ERROR)
+        error_handler.setFormatter(file_formatter)
+        logger.addHandler(error_handler)
+    
+    return logger
+
+
+class PerformanceMonitor:
+    """Công cụ theo dõi hiệu suất"""
+    
+    def __init__(self):
+        self.logger = setup_logger("PerformanceMonitor")
+        self.metrics = {}
+    
+    def record_metric(self, name: str, value: float, unit: str = "ms") -> None:
+        """Ghi lại số liệu hiệu suất"""
+        if name not in self.metrics:
+            self.metrics[name] = []
+        
+        self.metrics[name].append(value)
+        
+        # Chỉ lưu 1000 bản ghi gần nhất
+        if len(self.metrics[name]) > 1000:
+            self.metrics[name] = self.metrics[name][-1000:]
+    
+    def get_average(self, name: str) -> Optional[float]:
+        """Lấy giá trị trung bình"""
+        if name not in self.metrics or not self.metrics[name]:
+            return None
+        return sum(self.metrics[name]) / len(self.metrics[name])
+    
+    def report(self) -> str:
+        """Tạo báo cáo hiệu suất"""
+        if not self.metrics:
+            return "No performance data yet"
+        
+        report = "=== Performance Report ===\n"
+        for name, values in self.metrics.items():
+            if values:
+                avg = sum(values) / len(values)
+                max_val = max(values)
+                min_val = min(values)
+                report += f"{name}: avg={avg:.2f}ms, max={max_val:.2f}ms, min={min_val:.2f}ms, count={len(values)}\n"
+        
+        return report
+
+
+# Phiên bản toàn cục
+_logger = setup_logger("NovelTool")
+_performance_monitor = PerformanceMonitor()
+
+
+def get_logger(name: str = "NovelTool") -> logging.Logger:
+    """Lấy biến instance của logger"""
+    return logging.getLogger(name)
+
+
+def get_performance_monitor() -> PerformanceMonitor:
+    """Lấy trình theo dõi hiệu suất"""
+    return _performance_monitor