feat(logging): 添加日志扩展模块并集成到HTTP服务中

添加了`logging_extension.py`模块，用于初始化日志记录器，支持日志文件的按天轮转和控制台输出。将该模块集成到HTTP服务中，以便在服务启动时自动初始化日志记录器，并在错误处理时记录日志信息。

Author: v833

internal/extension/__init__.py

@@ -1,4 +1,5 @@
 from .migrate_extensive import migrate
 from .database_extension import db
+from .logging_extension import logging_init_app
 
-__all__ = ["db", "migrate"]
+__all__ = ["db", "migrate", "logging_init_app"]

internal/extension/logging_extension.py

@@ -0,0 +1,37 @@
+import logging
+import os.path
+from logging.handlers import TimedRotatingFileHandler
+
+from flask import Flask
+
+
+def logging_init_app(app: Flask):
+    """日志记录器初始化"""
+    # 1.设置日志存储的文件夹，如果不存在则创建
+    log_folder = os.path.join(os.getcwd(), "storage", "log")
+    if not os.path.exists(log_folder):
+        os.makedirs(log_folder)
+
+    # 2.定义日志的文件名
+    log_file = os.path.join(log_folder, "app.log")
+
+    # 3.设置日志的格式，并且让日志每天更新一次
+    handler = TimedRotatingFileHandler(
+        log_file,
+        when="midnight",
+        interval=1,
+        backupCount=30,
+        encoding="utf-8",
+    )
+    formatter = logging.Formatter(
+        "[%(asctime)s.%(msecs)03d] %(filename)s -> %(funcName)s line:%(lineno)d [%(levelname)s]: %(message)s"
+    )
+    handler.setLevel(logging.DEBUG)
+    handler.setFormatter(formatter)
+    logging.getLogger().addHandler(handler)
+
+    # 4.在开发环境下同时将日志输出到控制台
+    if app.debug or os.getenv("FLASK_ENV") == "development":
+        console_handler = logging.StreamHandler()
+        console_handler.setFormatter(formatter)
+        logging.getLogger().addHandler(console_handler)

internal/server/http.py

@@ -1,49 +1,67 @@
+import logging
 from flask import Flask, Response
 from internal.router import Router
 from config import Config
 from internal.exception import CustomException
-from pkg.response import json,Response,HttpCode
+from internal.extension import logging_init_app
+from pkg.response import json, Response, HttpCode
 import os
 from pkg.sqlalchemy import SQLAlchemy
 from flask_migrate import Migrate
 from flask_cors import CORS
 
 
 class Http(Flask):
-  def __init__(self, *args,config:Config,db: SQLAlchemy, 
-               migrate: Migrate, router:Router, **kwargs):
-    super().__init__(*args, **kwargs)
-    
-    if config:  
-      self.config.from_object(config)
-    
-    self.register_error_handler(Exception, self._register_error_handler)
-    
-    db.init_app(self)
-
-    
-    
-    # with self.app_context():
-      # db.drop_all()
-    #   db.create_all()
-    
-    migrate.init_app(self, db, directory="internal/migration")
-
-    CORS(self, resources={
-            r"/*": {
-                "origins": "*",
-                "supports_credentials": True,
-            }
-        })
-    
-    router.register_router(self)
-  
-    
-  def _register_error_handler(self, err: Exception):
-    if isinstance(err, CustomException):
-      return json(Response(code=err.code, message=err.message, data=err.data if err.data is not None else {}))
-    
-    if self.debug or os.getenv("FLASK_DEBUG"):
-      raise err
-    else:
-      return json(Response(code=HttpCode.FAIL, message=str(err), data={}))
+    def __init__(
+        self,
+        *args,
+        config: Config,
+        db: SQLAlchemy,
+        migrate: Migrate,
+        router: Router,
+        **kwargs
+    ):
+        super().__init__(*args, **kwargs)
+
+        if config:
+            self.config.from_object(config)
+
+        self.register_error_handler(Exception, self._register_error_handler)
+
+        db.init_app(self)
+
+        # with self.app_context():
+        # db.drop_all()
+        #   db.create_all()
+
+        migrate.init_app(self, db, directory="internal/migration")
+        logging_init_app(self)
+
+        CORS(
+            self,
+            resources={
+                r"/*": {
+                    "origins": "*",
+                    "supports_credentials": True,
+                }
+            },
+        )
+
+        router.register_router(self)
+
+    def _register_error_handler(self, err: Exception):
+        logging.error("An error occurred: %s", err, exc_info=True)
+
+        if isinstance(err, CustomException):
+            return json(
+                Response(
+                    code=err.code,
+                    message=err.message,
+                    data=err.data if err.data is not None else {},
+                )
+            )
+
+        if self.debug or os.getenv("FLASK_DEBUG"):
+            raise err
+        else:
+            return json(Response(code=HttpCode.FAIL, message=str(err), data={}))