第一次提交

app/utils/logger.py

@@ -0,0 +1,51 @@
+from flask import request, current_app
+from flask_login import current_user
+from app.models import AuditLog
+from functools import wraps
+import json
+
+def log_operation(action, target_type, target_id=None, details=None):
+    """记录操作日志的工具函数"""
+    try:
+        # 获取当前用户信息
+        admin_id = getattr(current_user, 'admin_id', None) if hasattr(current_user, 'is_authenticated') and current_user.is_authenticated else None
+        
+        # 获取客户端IP
+        ip_address = request.headers.get('X-Forwarded-For', request.remote_addr)
+        
+        # 获取用户代理
+        user_agent = request.headers.get('User-Agent', '')
+        
+        # 记录审计日志
+        AuditLog.log_action(
+            admin_id=admin_id,
+            action=action,
+            target_type=target_type,
+            target_id=target_id,
+            details=details,
+            ip_address=ip_address,
+            user_agent=user_agent
+        )
+    except Exception as e:
+        if hasattr(current_app, 'logger'):
+            current_app.logger.error(f"记录操作日志失败: {str(e)}")
+
+def log_operations(action, target_type):
+    """记录操作日志的装饰器"""
+    def decorator(f):
+        @wraps(f)
+        def decorated_function(*args, **kwargs):
+            try:
+                # 执行原函数
+                result = f(*args, **kwargs)
+                
+                # 记录成功日志
+                log_operation(action, target_type)
+                
+                return result
+            except Exception as e:
+                # 记录错误日志
+                log_operation(f"{action}_ERROR", target_type, details={'error': str(e)})
+                raise e
+        return decorated_function
+    return decorator