huace-aigc-auth-client 1.1.3__tar.gz → 1.1.4__tar.gz

{huace_aigc_auth_client-1.1.3/huace_aigc_auth_client.egg-info → huace_aigc_auth_client-1.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: huace-aigc-auth-client
-Version: 1.1.3
+Version: 1.1.4
 Summary: 华策AIGC Auth Client - 提供 Token 验证、用户信息获取、权限检查、旧系统接入等功能
 Author-email: Huace <support@huace.com>
 License: MIT

{huace_aigc_auth_client-1.1.3 → huace_aigc_auth_client-1.1.4}/huace_aigc_auth_client/__init__.py

@@ -90,4 +90,4 @@ __all__ = [
     "create_sync_config",
     "create_default_field_mappings",
 ]
-__version__ = "1.1.3"
+__version__ = "1.1.4"

{huace_aigc_auth_client-1.1.3 → huace_aigc_auth_client-1.1.4}/huace_aigc_auth_client/sdk.py

@@ -577,7 +577,7 @@ class AuthMiddleware:
             return None
         return authorization[7:]
 
-    async def fastapi_middleware(self, request, call_next):
+    async def fastapi_middleware(self, request, call_next, user_info_callback: Callable = None):
         """
         FastAPI 中间件
 
@@ -613,22 +613,23 @@ class AuthMiddleware:
             forwarded_proto = request.headers.get("x-forwarded-proto")
             if forwarded_proto:
                 request.scope["scheme"] = forwarded_proto
+            if user_info_callback:
+                await user_info_callback(request, user_info)
         except AigcAuthError as e:
             return JSONResponse(
                 status_code=401,
                 content={"code": e.code, "message": e.message, "data": None}
             )
-
         return await call_next(request)
 
-    def flask_before_request(self):
+    def flask_before_request(self, user_info_callback: Callable = None):
         """
         Flask before_request 处理器
 
         使用方法:
             @app.before_request
             def before_request():
-                return auth_middleware.flask_before_request()
+                return auth_middleware.flask_before_request(user_info_callback=user_info_callback)
         """
         from flask import request, jsonify, g
 
@@ -654,6 +655,8 @@ class AuthMiddleware:
             user_info = self.client.get_user_info(token)
             # 将用户信息存储到 flask.g
             g.user_info = user_info
+            if user_info_callback:
+                user_info_callback(request, user_info)
         except AigcAuthError as e:
             return jsonify({
                 "code": e.code,

{huace_aigc_auth_client-1.1.3 → huace_aigc_auth_client-1.1.4/huace_aigc_auth_client.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: huace-aigc-auth-client
-Version: 1.1.3
+Version: 1.1.4
 Summary: 华策AIGC Auth Client - 提供 Token 验证、用户信息获取、权限检查、旧系统接入等功能
 Author-email: Huace <support@huace.com>
 License: MIT

{huace_aigc_auth_client-1.1.3 → huace_aigc_auth_client-1.1.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "huace-aigc-auth-client"
-version = "1.1.3"
+version = "1.1.4"
 description = "华策AIGC Auth Client - 提供 Token 验证、用户信息获取、权限检查、旧系统接入等功能"
 readme = "README.md"
 requires-python = ">=3.7"