init

backend/api/core/Exceptions.py

@@ -1,21 +1,12 @@
-from datetime import datetime
-import logging
-import traceback
-from fastapi import Request, status
-from fastapi.responses import JSONResponse
-# from fastapi import Request, status
-
-from api.core.exception import ResponseException
-
-logFormatter = logging.Formatter(
-    """
-[start]
-%(asctime)s
-<%(name)s>
-%(message)s
-[end]
-"""
-)
+import logging
+import traceback
+from fastapi import Request, status
+from fastapi.responses import JSONResponse
+# from fastapi import Request, status
+
+from api.core.exception import ResponseException
+
+logger = logging.getLogger('dcalc.exception')
 
 
 def format_exception() -> str:
@@ -30,34 +21,12 @@ async def all_exception_handler(request: Request, ex: Exception):
     info = ''
     if not isinstance(ex, ResponseException):
         info = format_exception()
-        if request.client.host != '127.0.0.1':
-            from fastapi.logger import logger
-            logger.setLevel(logging.ERROR)
-            logger.name = 'dcalc'
-            log_directory = 'logs'
-            import os
-
-            if not os.path.exists('./logs'):
-                os.makedirs('./logs')
-            time_str = datetime.now().strftime('%Y_%m_%d')
-            fileHandler = logging.FileHandler(f'{log_directory}/{logger.name}_{time_str}.log', encoding='utf-8')
-            fileHandler.setFormatter(logFormatter)
-            fileHandler.setLevel(logging.ERROR)
-            logger.addHandler(fileHandler)
-            try:
-                import pathlib
-                pathlib.Path(log_directory).mkdir(parents=True, exist_ok=True)
-            except PermissionError:
-                print('创建日志目录logs失败，请确认是否限制了基础的运行权限')
-            logFunc = logger.error
-            try:
-                logFunc(info)
-                info = '服务器内部错误'
-            finally:
-                logger.removeHandler(fileHandler)
-                fileHandler.close()
-    else:
-        info = ex.args[0]
+        client_host = request.client.host if request.client else ''
+        if client_host != '127.0.0.1':
+            logger.error('Unhandled exception for %s', request.url.path, exc_info=(type(ex), ex, ex.__traceback__))
+            info = '服务器内部错误'
+    else:
+        info = ex.args[0]
     # traceback.print_last(sys.exc_info())
     return JSONResponse(
         status_code=status.HTTP_200_OK,