ezKit 1.12.29__py3-none-any.whl → 1.12.30__py3-none-any.whl

ezKit/fastapix.py

@@ -0,0 +1,78 @@
+# FastAPI Extensions
+# pylint: disable=W0613
+
+import logging
+import sys
+from typing import Any
+
+from fastapi import FastAPI, Request
+from fastapi.exceptions import RequestValidationError
+from fastapi.responses import JSONResponse
+from loguru import logger
+from pydantic import ValidationError
+from starlette.exceptions import HTTPException as StarletteHTTPException
+
+# 移除默认控制台日志器
+logger.remove()
+
+# 日志输出到控制台
+logger.add(sys.stdout, level="INFO", format="<green>{time:YYYY-MM-DD HH:mm:ss.SSS}</green> | <level>{level}</level> | <cyan>{name}</cyan>:<cyan>{function}</cyan> - <level>{message}</level>")
+
+# 日志输出到文件
+logger.add(
+    sink="logs/runtime.log",  # 日志文件
+    rotation="10 MB",  # 文件达到 10MB 自动轮转
+    retention="7 days",  # 保留 7 天日志
+    compression="zip",  # 超过的日志自动压缩
+    level="INFO",  # 记录等级
+    encoding="utf-8",  # 文件编码
+    enqueue=True,  # 多线程、多进程安全
+    diagnose=True,  # 显示变量值
+    backtrace=True,  # 捕获堆栈追踪
+)
+
+
+def Response(code: int = 200, data: Any = None, message: str | None = None, status_code: int = 200):
+    return JSONResponse(content={"code": code, "data": data, "message": message}, status_code=status_code)
+
+
+def exceptions(app: FastAPI):
+
+    @app.exception_handler(StarletteHTTPException)
+    async def http_exception_handler(request: Request, exc: StarletteHTTPException):
+        logger.warning(f"HTTP Exception: {exc.detail}")
+        return Response(status_code=exc.status_code, code=exc.status_code, message=exc.detail)
+
+    # 参数验证错误
+    @app.exception_handler(RequestValidationError)
+    async def validation_exception_handler(request: Request, exc: RequestValidationError):
+        logger.warning(f"Request Validation Error: {exc.errors()}")
+        return Response(code=422, data=exc.errors(), message="Request Validation Error")
+
+    # 参数验证错误
+    @app.exception_handler(ValidationError)
+    async def pydantic_validation_error_handler(request: Request, exc: ValidationError):
+        logger.warning(f"Pydantic Validation Error: {exc.errors()}")
+        return Response(code=422, data=exc.errors(), message="Pydantic Validation Error")
+
+    # 服务器内部错误
+    @app.exception_handler(Exception)
+    async def global_exception_handler(request: Request, exc: Exception):
+        logger.exception("Unhandled Exception")
+        return Response(code=500, data=None, message="Server Internal Error")
+
+
+# 兼容 FastAPI/Uvicorn 的 logging（重要）
+class InterceptHandler(logging.Handler):
+    """Intercept Handler"""
+
+    def emit(self, record):
+        logger_opt = logger.opt(depth=6, exception=record.exc_info)
+        logger_opt.log(record.levelname, record.getMessage())
+
+    def write(self, message: str):
+        if message.strip():
+            logger.info(message.strip())
+
+    def flush(self):
+        pass

{ezkit-1.12.29.dist-info → ezkit-1.12.30.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ezKit
-Version: 1.12.29
+Version: 1.12.30
 Summary: Easy Kit
 Author: septvean
 Author-email: septvean@gmail.com

{ezkit-1.12.29.dist-info → ezkit-1.12.30.dist-info}/RECORD

@@ -6,6 +6,7 @@ ezKit/cipher.py,sha256=7jBarRH7ukSYzkz-Anl8B8JzluhnRz4CLHidPRRj_cg,2939
 ezKit/database.py,sha256=5MsIYDhyN6VtQ8MJy7pbMGqeaXguy9DzRHgJ6NJD5Gs,30126
 ezKit/dockerhub.py,sha256=j-wQO-71BsOgExHZhYynuy2k_hCX3on-vg0TH7QCit4,1996
 ezKit/errors.py,sha256=wLLzJ-YWCdAUqdLzsHM-jfnfZJL9pHtZoG_O_bVNI_A,1334
+ezKit/fastapix.py,sha256=59wqVLqJxNVesZTAwAqkzkiMQHKQQbVUijFP-IDu4ws,2935
 ezKit/http.py,sha256=zhNxJF-x91UqGncXWxVXnhZVpFo_wmmpGnMXVT11y9E,1832
 ezKit/markdown_to_html.template,sha256=21G2sSVGJn6aJvHd0NN4zY5YiDteKe4UtW36AzBwSdk,22274
 ezKit/mongo.py,sha256=vsRCjJ2uWbNp-1bnGKICPohNYx25El8XnHp10o-lsM4,2397
@@ -16,8 +17,8 @@ ezKit/token.py,sha256=Ac-i9xfq4TqpGyfCzakjrh4NYzxHiN2sCQrMk1tzVi8,1716
 ezKit/utils.py,sha256=U457ahFkxIXuB-qWvS3995xJs-LlkFIX5_ZWVgmL5cY,43130
 ezKit/xftp.py,sha256=-XQXyhMqeigT63P6sXkSS7r4GROXyqqlkzKxITLWG-g,8278
 ezKit/zabbix.py,sha256=PkMnfu7mcuotwwIIsHaC9FsNg-gap6hD1xvm0AwSL1Y,33777
-ezkit-1.12.29.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-ezkit-1.12.29.dist-info/METADATA,sha256=8eEFRDzl6V1laRNNs2E1maD3xcSGOXBlNBVTr9s2geQ,317
-ezkit-1.12.29.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ezkit-1.12.29.dist-info/top_level.txt,sha256=aYLB_1WODsqNTsTFWcKP-BN0KCTKcV-HZJ4zlHkCFw8,6
-ezkit-1.12.29.dist-info/RECORD,,
+ezkit-1.12.30.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+ezkit-1.12.30.dist-info/METADATA,sha256=WEAe_vtj_2i15McEoZWDdEOc94jA84vjbJ6-76Xji08,317
+ezkit-1.12.30.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ezkit-1.12.30.dist-info/top_level.txt,sha256=aYLB_1WODsqNTsTFWcKP-BN0KCTKcV-HZJ4zlHkCFw8,6
+ezkit-1.12.30.dist-info/RECORD,,