nc-user-terminator 0.1.4__tar.gz → 0.1.5__tar.gz

{nc_user_terminator-0.1.4 → nc_user_terminator-0.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nc-user-terminator
-Version: 0.1.4
+Version: 0.1.5
 Summary: OAuth client wrapper for multi-tenant projects
 Author: bw_song
 Author-email: m132777096902@gmail.com
@@ -8,5 +8,8 @@ Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 Requires-Dist: cachetools==6.2.0
 Dynamic: author-email
-Dynamic: description-content-type
 Dynamic: requires-python
+
+# 更新
++ V0.1.5
+  - 新增本地缓存机制

nc_user_terminator-0.1.5/README.md

@@ -0,0 +1,3 @@
+# 更新
++ V0.1.5
+  - 新增本地缓存机制

{nc_user_terminator-0.1.4 → nc_user_terminator-0.1.5}/nc_user_manager/client.py

@@ -7,7 +7,7 @@ from exceptions import OAuthError
 from utils import request
 
 
-TENANT_HEADER = "X-Tenant-Code"
+PRODUCT_HEADER = "X-Product-Code"
 DEFAULT_TTL = 3600
 
 
@@ -15,7 +15,7 @@ class OAuthClient:
     def __init__(
         self,
         base_url: str,
-        tenant_code: str,
+        product_code: str,
         redirect_url: Optional[str] = None,
         single_session: bool = False,
         cache: Optional[BaseCache] = None,
@@ -24,12 +24,12 @@ class OAuthClient:
         OAuth 客户端
 
         :param base_url: 服务端基础地址 (例如 http://localhost:8000)
-        :param tenant_code: 租户编码
+        :param product_code: 产品编码
         :param redirect_url: 可选，重定向地址
         :param single_session: 是否单会话登录
         """
         self._base_url = base_url.rstrip("/")
-        self._tenant_code = tenant_code
+        self._product_code = product_code
         self._redirect_url = redirect_url
         self._single_session = single_session
         self._cache = cache or MemoryCache(maxsize=10000, ttl=DEFAULT_TTL)
@@ -43,7 +43,7 @@ class OAuthClient:
         return await asyncio.to_thread(request, *args, **kwargs)
 
     def _headers(self, extra: Optional[Dict[str, str]] = None) -> Dict[str, str]:
-        headers = {TENANT_HEADER: self._tenant_code}
+        headers = {PRODUCT_HEADER: self._product_code}
         if extra:
             headers.update(extra)
         return headers
@@ -209,7 +209,8 @@ class OAuthClient:
         headers = self._headers({"Authorization": f"Bearer {token}"})
         res_dict = request("GET", f"{self._base_url}/api/me", headers=headers)
 
-        self._cache_user(token, res_dict)
+        if res_dict.get("success", True):
+            self._cache_user(token, res_dict)
         return UserResponse(res_dict)
 
     async def say_my_name_async(self, token: str) -> UserResponse:
@@ -224,8 +225,9 @@ class OAuthClient:
 
         headers = self._headers({"Authorization": f"Bearer {token}"})
         res_dict = await self._arequest("GET", f"{self._base_url}/api/me", headers=headers)
+        if res_dict.get("success", True):
+            await self._cache_user_async(token, res_dict)
 
-        await self._cache_user_async(token, res_dict)
         return UserResponse(res_dict)
 
     # 刷新过期时间
@@ -264,3 +266,30 @@ class OAuthClient:
         await self._uncache_user_async(token, data)
         return resp
 
+async def main():
+    def get_redis_url() -> str:
+        return f'redis://default:1q2w3e@localhost:6379/1'
+
+
+    import redis.asyncio as redis
+    # —— Redis 连接与策略（Bearer + Redis）——
+    _redis = redis.from_url(get_redis_url(), decode_responses=True)
+
+    redis_cache = AsyncRedisCache(_redis)
+
+    client = OAuthClient("http://localhost:8000/", "bankgpt", "http://localhost:8000/auth/google/custom_callback", single_session=True, cache=redis_cache)
+    # authorize = client.authorize("google")
+    # print(authorize)
+
+
+    name = await client.say_my_name_async("Tu72u4SVue4ezadi1v5Ui9r9HjxH7YqbkY_yf5jFtPQ")
+    print(name)
+    name = await client.say_my_name_async("Tu72u4SVue4ezadi1v5Ui9r9HjxH7YqbkY_yf5jFtPQ")
+    print(name)
+    name = await client.logout_async("Tu72u4SVue4ezadi1v5Ui9r9HjxH7YqbkY_yf5jFtPQ")
+    print(name)
+
+
+if __name__ == '__main__':
+    asyncio.run(main())
+

nc_user_terminator-0.1.5/nc_user_manager/utils.py

@@ -0,0 +1,77 @@
+import json
+import urllib.parse
+import urllib.request
+import urllib.error
+from typing import Optional, Dict, Any
+
+
+def request(
+    method: str,
+    url: str,
+    params: Optional[Dict[str, Any]] = None,
+    headers: Optional[Dict[str, str]] = None,
+    json_body: Optional[dict] = None
+) -> dict:
+    """
+    发送 HTTP 请求并返回统一结果，不抛异常
+    返回格式:
+    {
+        "status": int,           # HTTP 状态码
+        "success": bool,         # 是否成功 (2xx)
+        "body": dict or str,     # JSON解析后的body, 或原始body
+        "error": Optional[str]   # 错误信息
+    }
+    """
+    # 拼接 GET 参数
+    if params:
+        query = urllib.parse.urlencode(params)
+        url = f"{url}?{query}"
+
+    data = None
+    if json_body is not None:
+        data = json.dumps(json_body).encode("utf-8")
+        if headers is None:
+            headers = {}
+        headers["Content-Type"] = "application/json"
+
+    req = urllib.request.Request(url, method=method.upper(), data=data)
+    if headers:
+        for k, v in headers.items():
+            req.add_header(k, v)
+
+    result = {
+        "status": 0,
+        "success": False,
+        "message": None,
+        "error": None
+    }
+
+    try:
+        with urllib.request.urlopen(req) as resp:
+            body = resp.read().decode()
+            if not body:
+                return {}
+            try:
+                return json.loads(body)
+            except json.JSONDecodeError:
+                result["status"] = 200
+                result["success"] = False
+                result["message"] = body
+                result["error"] = "Invalid JSON response"
+                return result
+    except urllib.error.HTTPError as e:
+        body = e.read().decode() if e.fp else None
+        result["status"] = e.code
+        try:
+            result["message"] = json.loads(body) if body else None
+        except json.JSONDecodeError:
+            result["message"] = body
+        result["error"] = "HTTPError"
+        result["success"] = False
+    except urllib.error.URLError as e:
+        result["status"] = 0
+        result["message"] = None
+        result["error"] = f"NetworkError: {e.reason}"
+        result["success"] = False
+
+    return result

{nc_user_terminator-0.1.4 → nc_user_terminator-0.1.5}/nc_user_terminator.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nc-user-terminator
-Version: 0.1.4
+Version: 0.1.5
 Summary: OAuth client wrapper for multi-tenant projects
 Author: bw_song
 Author-email: m132777096902@gmail.com
@@ -8,5 +8,8 @@ Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 Requires-Dist: cachetools==6.2.0
 Dynamic: author-email
-Dynamic: description-content-type
 Dynamic: requires-python
+
+# 更新
++ V0.1.5
+  - 新增本地缓存机制

{nc_user_terminator-0.1.4 → nc_user_terminator-0.1.5}/pyproject.toml

@@ -4,11 +4,12 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "nc-user-terminator"
-version = "0.1.4"
+version = "0.1.5"
 description = "OAuth client wrapper for multi-tenant projects"
 authors = [
     { name = "bw_song" }
 ]
+readme = "README.md"
 requires-python = ">=3.8"
 dependencies = [
     "cachetools==6.2.0"

nc_user_terminator-0.1.4/nc_user_manager/utils.py

@@ -1,39 +0,0 @@
-import json
-import urllib.parse
-import urllib.request
-import urllib.error
-
-from exceptions import OAuthError
-
-
-def request(method: str, url: str, params=None, headers=None, json_body=None) -> dict:
-    # 拼接 GET 参数
-    if params:
-        query = urllib.parse.urlencode(params)
-        url = f"{url}?{query}"
-
-    data = None
-    if json_body is not None:
-        data = json.dumps(json_body).encode("utf-8")
-        if headers is None:
-            headers = {}
-        headers["Content-Type"] = "application/json"
-
-    req = urllib.request.Request(url, method=method.upper(), data=data)
-
-    if headers:
-        for k, v in headers.items():
-            req.add_header(k, v)
-
-    try:
-        with urllib.request.urlopen(req) as resp:
-            body = resp.read().decode()
-            try:
-                return json.loads(body)
-            except json.JSONDecodeError:
-                raise OAuthError("Invalid JSON response", resp.getcode(), body)
-    except urllib.error.HTTPError as e:
-        body = e.read().decode() if e.fp else None
-        raise OAuthError("HTTP request failed", e.code, body)
-    except urllib.error.URLError as e:
-        raise OAuthError(f"Network error: {e.reason}")