huace-aigc-auth-client 1.1.30__py3-none-any.whl → 1.1.31__py3-none-any.whl

huace_aigc_auth_client/__init__.py

@@ -54,7 +54,20 @@ from .sdk import (
     create_fastapi_auth_dependency
 )
 
-from .user_context import get_current_user
+from .user_context import (
+    get_current_user, add_to_request_context, clear_request_context, set_request_context, get_request_context, get_client_ip, get_user_agent, get_request_id, get_trace_id, get_request_token, get_request_app_id, get_request_app_secret
+)
+
+from .api_stats_collector import (
+    init_api_stats_collector,
+    get_api_stats_collector,
+    collect_api_stat,
+    stop_api_stats_collector
+)
+
+from .auth_request import (
+    auth_request,
+)
 
 from .legacy_adapter import (
     LegacySystemAdapter,
@@ -130,6 +143,12 @@ def setLogger(log):
     except Exception as e:
         print(f"Failed to set logger for api_stats_collector module: {e}")
     
+    try:
+        from .auth_request import setLogger as auth_request_setLogger
+        auth_request_setLogger(log)
+    except Exception as e:
+        print(f"Failed to set logger for auth_request module: {e}")
+    
     if _fastapi_available:
         try:
             from .webhook import setLogger as webhook_setLogger
@@ -177,5 +196,23 @@ __all__ = [
     "setLogger",
     # 用户上下文
     "get_current_user",
+    "add_to_request_context",
+    "clear_request_context",
+    "set_request_context",
+    "get_request_context",
+    "get_client_ip",
+    "get_user_agent",
+    "get_request_id",
+    "get_trace_id",
+    "get_request_token",
+    "get_request_app_id",
+    "get_request_app_secret",
+    # API 统计收集器
+    "init_api_stats_collector",
+    "get_api_stats_collector",
+    "collect_api_stat",
+    "stop_api_stats_collector",
+    # 认证请求封装
+    "auth_request",
 ]
-__version__ = "1.1.30"
+__version__ = "1.1.31"

huace_aigc_auth_client/auth_request.py

@@ -12,7 +12,15 @@ from .user_context import get_request_context
 from .api_stats_collector import get_api_stats_collector
 
 logger = logging.getLogger(__name__)
-
+def setLogger(log):
+    """
+    统一设置所有模块的 logger
+    
+    Args:
+        log: logging.Logger 实例
+    """
+    global logger
+    logger = log
 
 def auth_request(
     method: str,

{huace_aigc_auth_client-1.1.30.dist-info → huace_aigc_auth_client-1.1.31.dist-info}/METADATA

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

{huace_aigc_auth_client-1.1.30.dist-info → huace_aigc_auth_client-1.1.31.dist-info}/RECORD

@@ -1,13 +1,13 @@
-huace_aigc_auth_client/__init__.py,sha256=9wZM_9CM2SukN3pw2JpiXe-oVqXh0FD3h0y4S8ueckQ,4857
+huace_aigc_auth_client/__init__.py,sha256=coLaEiBYIxrTEunRc53gOiyETrH3_EoTPt_7n_QcDaI,5959
 huace_aigc_auth_client/api_stats_collector.py,sha256=ADpjpHXMqn80YI4UltWHbzAO_szykU9ZCvwXgBRWFIM,11046
-huace_aigc_auth_client/auth_request.py,sha256=f7KgmyamoEk1HWP-47nCxUBnOgRbQjYnnNZ7yRzb4j0,9285
+huace_aigc_auth_client/auth_request.py,sha256=0Nfe4pAywOvaE5l6m42JpyKBG0MLcpAf54i8I5KKhf0,9444
 huace_aigc_auth_client/legacy_adapter.py,sha256=TVCBAKejE2z2HQFsEwDW8LMiaIkXNfz3Mxv6_E-UJFY,24102
 huace_aigc_auth_client/sdk.py,sha256=rproo913OAi37wz_rMYgxzP3F1YyY3nc5e35JS5WvoY,37751
 huace_aigc_auth_client/user_context.py,sha256=IqdX6Xd2jJwvij6Hc2qWAFWj5pn3wHqk0RBsaXKLP8g,6795
 huace_aigc_auth_client/webhook.py,sha256=XQZYEbMoqIdqZWCGSTcedeDKJpDbUVSq5g08g-6Qucg,4124
 huace_aigc_auth_client/webhook_flask.py,sha256=Iosu4dBtRhQZM_ytn-bn82MpVsyOiV28FBnt7Tfh31U,7225
-huace_aigc_auth_client-1.1.30.dist-info/licenses/LICENSE,sha256=z7dgC7KljhBLNvKjN15391nMj3aLt0gbud8-Yf1F8EQ,1063
-huace_aigc_auth_client-1.1.30.dist-info/METADATA,sha256=3rtqM-PxSFyXEc0l_UW2gy6R5rN-J81VpKfuMgVNfYI,23629
-huace_aigc_auth_client-1.1.30.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-huace_aigc_auth_client-1.1.30.dist-info/top_level.txt,sha256=kbv0nQ6PQ0JVneWPH7O2AbtlJnP7AjvFJ6JjM6ZEBxo,23
-huace_aigc_auth_client-1.1.30.dist-info/RECORD,,
+huace_aigc_auth_client-1.1.31.dist-info/licenses/LICENSE,sha256=z7dgC7KljhBLNvKjN15391nMj3aLt0gbud8-Yf1F8EQ,1063
+huace_aigc_auth_client-1.1.31.dist-info/METADATA,sha256=qAOEVKk5_5X45kmGsP2HMWMJSGfAYc9Nsc_WI0MiJIs,23629
+huace_aigc_auth_client-1.1.31.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+huace_aigc_auth_client-1.1.31.dist-info/top_level.txt,sha256=kbv0nQ6PQ0JVneWPH7O2AbtlJnP7AjvFJ6JjM6ZEBxo,23
+huace_aigc_auth_client-1.1.31.dist-info/RECORD,,