huace-aigc-auth-client 1.1.3__py3-none-any.whl → 1.1.4__py3-none-any.whl

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/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.dist-info → huace_aigc_auth_client-1.1.4.dist-info}/METADATA

@@ -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.4.dist-info/RECORD

@@ -0,0 +1,8 @@
+huace_aigc_auth_client/__init__.py,sha256=isMBleAvICiYwHXv0fm6qf-nrvdIMIePGZbrrG_rQws,2163
+huace_aigc_auth_client/legacy_adapter.py,sha256=lCxpwROHZd0RG0dVp6AidQQRpFv6H8DvVzjrYHRDZFg,21515
+huace_aigc_auth_client/sdk.py,sha256=4Azj2TPOCLWv6dTaAqw--_Uh-Gzpp0KhpxOEBGONG48,22749
+huace_aigc_auth_client-1.1.4.dist-info/licenses/LICENSE,sha256=z7dgC7KljhBLNvKjN15391nMj3aLt0gbud8-Yf1F8EQ,1063
+huace_aigc_auth_client-1.1.4.dist-info/METADATA,sha256=ly5bJH-amh0jsbg_ArABrq8mvqOfCQ0nYz1UtdD0Ka4,20084
+huace_aigc_auth_client-1.1.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+huace_aigc_auth_client-1.1.4.dist-info/top_level.txt,sha256=kbv0nQ6PQ0JVneWPH7O2AbtlJnP7AjvFJ6JjM6ZEBxo,23
+huace_aigc_auth_client-1.1.4.dist-info/RECORD,,

huace_aigc_auth_client-1.1.3.dist-info/RECORD

@@ -1,8 +0,0 @@
-huace_aigc_auth_client/__init__.py,sha256=WwdgR8RP5SkueieSv6HCjIirIZdwTDpyGTaBzpENK5g,2163
-huace_aigc_auth_client/legacy_adapter.py,sha256=lCxpwROHZd0RG0dVp6AidQQRpFv6H8DvVzjrYHRDZFg,21515
-huace_aigc_auth_client/sdk.py,sha256=46kkNDmWcce4BkSAT9zn5n8XdURDns3YtXIIfe8TGDU,22453
-huace_aigc_auth_client-1.1.3.dist-info/licenses/LICENSE,sha256=z7dgC7KljhBLNvKjN15391nMj3aLt0gbud8-Yf1F8EQ,1063
-huace_aigc_auth_client-1.1.3.dist-info/METADATA,sha256=nATeZZKa_zxe1vVU5KF6jNiASCWPwbDS3yWqpv7FtzE,20084
-huace_aigc_auth_client-1.1.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-huace_aigc_auth_client-1.1.3.dist-info/top_level.txt,sha256=kbv0nQ6PQ0JVneWPH7O2AbtlJnP7AjvFJ6JjM6ZEBxo,23
-huace_aigc_auth_client-1.1.3.dist-info/RECORD,,