nc-user-terminator 0.1.0__py3-none-any.whl

nc-user-manager/__init__.py

@@ -0,0 +1,10 @@
+from client import OAuthClient
+from exceptions import OAuthError
+from models import UserResponse, OAuth2AuthorizeResponse
+
+__all__ = [
+    "OAuthClient",
+    "OAuthError",
+    "UserResponse",
+    "OAuth2AuthorizeResponse",
+]

nc-user-manager/client.py

@@ -0,0 +1,180 @@
+import asyncio
+from typing import Optional, Dict, Any
+
+from models import UserResponse, OAuth2AuthorizeResponse, CallbackResponse
+from exceptions import OAuthError
+from utils import request
+
+
+TENANT_HEADER = "X-Tenant-Code"
+
+
+class OAuthClient:
+    def __init__(
+        self,
+        base_url: str,
+        tenant_code: str,
+        redirect_url: Optional[str] = None,
+        single_session: bool = False,
+    ):
+        """
+        OAuth 客户端
+
+        :param base_url: 服务端基础地址 (例如 http://localhost:8000)
+        :param tenant_code: 租户编码
+        :param redirect_url: 可选，重定向地址
+        :param single_session: 是否单会话登录
+        """
+        self._base_url = base_url.rstrip("/")
+        self._tenant_code = tenant_code
+        self._redirect_url = redirect_url
+        self._single_session = single_session
+
+    # ----------------------
+    # 内部异步包装
+    # ----------------------
+    async def _arequest(self, *args, **kwargs) -> dict:
+        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}
+        if extra:
+            headers.update(extra)
+        return headers
+
+    # ----------------------
+    # 授权
+    # ----------------------
+    def authorize(self, platform: str) -> OAuth2AuthorizeResponse:
+        """同步获取授权地址"""
+        params = {}
+        if self._redirect_url:
+            params["redirect_url"] = self._redirect_url
+        res_dict = request(
+            "GET",
+            f"{self._base_url}/api/oauth/{platform}/authorize",
+            params=params,
+            headers=self._headers(),
+        )
+        return OAuth2AuthorizeResponse(res_dict)
+
+    async def authorize_async(self, platform: str) -> OAuth2AuthorizeResponse:
+        """异步获取授权地址"""
+        params = {}
+        if self._redirect_url:
+            params["redirect_url"] = self._redirect_url
+        res_dict = await self._arequest(
+            "GET",
+            f"{self._base_url}/api/oauth/{platform}/authorize",
+            params=params,
+            headers=self._headers(),
+        )
+        return OAuth2AuthorizeResponse(res_dict)
+
+    # ----------------------
+    # 普通回调
+    # ----------------------
+    def callback(self, platform: str, query_params: Dict[str, Any]) -> CallbackResponse:
+        """同步处理 OAuth2 回调 (非 One Tap)"""
+        if query_params.get("credential"):
+            raise OAuthError("Use google_one_tap() for One Tap login")
+
+        params = {"single_session": str(self._single_session).lower()}
+        if self._redirect_url:
+            params["redirect_url"] = self._redirect_url
+        params.update(query_params or {})
+        res_dict = request(
+            "GET",
+            f"{self._base_url}/api/oauth/{platform}/callback",
+            params=params,
+            headers=self._headers(),
+        )
+        return CallbackResponse(res_dict)
+
+    async def callback_async(self, platform: str, query_params: Dict[str, Any]) -> CallbackResponse:
+        """异步处理 OAuth2 回调 (非 One Tap)"""
+        if query_params.get("credential"):
+            raise OAuthError("Use google_one_tap_async() for One Tap login")
+
+        params = {"single_session": str(self._single_session).lower()}
+        if self._redirect_url:
+            params["redirect_url"] = self._redirect_url
+        params.update(query_params or {})
+        res_dict = await self._arequest(
+            "GET",
+            f"{self._base_url}/api/oauth/{platform}/callback",
+            params=params,
+            headers=self._headers(),
+        )
+        return CallbackResponse(res_dict)
+
+    # ----------------------
+    # Google One Tap
+    # ----------------------
+    def google_one_tap(self, credential: str) -> CallbackResponse:
+        """同步 Google One Tap 登录"""
+        params = {"credential": credential}
+        if self._redirect_url:
+            params["redirect_url"] = self._redirect_url
+        res_dict = request(
+            "GET",
+            f"{self._base_url}/api/oauth/google/callback",
+            params=params,
+            headers=self._headers(),
+        )
+        return CallbackResponse(res_dict)
+
+    async def google_one_tap_async(self, credential: str) -> CallbackResponse:
+        """异步 Google One Tap 登录"""
+        params = {"credential": credential}
+        if self._redirect_url:
+            params["redirect_url"] = self._redirect_url
+        res_dict = await self._arequest(
+            "GET",
+            f"{self._base_url}/api/oauth/google/callback",
+            params=params,
+            headers=self._headers(),
+        )
+        return CallbackResponse(res_dict)
+
+    # ----------------------
+    # 获取用户信息
+    # ----------------------
+    def say_my_name(self, token: str) -> UserResponse:
+        """同步获取当前用户"""
+        headers = self._headers({"Authorization": f"Bearer {token}"})
+        res_dict = request("GET", f"{self._base_url}/api/me", headers=headers)
+        return UserResponse(res_dict)
+
+    async def say_my_name_async(self, token: str) -> UserResponse:
+        """异步获取当前用户"""
+        headers = self._headers({"Authorization": f"Bearer {token}"})
+        res_dict = await self._arequest("GET", f"{self._base_url}/api/me", headers=headers)
+        return UserResponse(res_dict)
+
+    # 刷新过期时间
+    def reborn(self, token: str, extend_seconds: Optional[int] = None) -> dict:
+        """重新设置过期时间"""
+        headers = self._headers({"Authorization": f"Bearer {token}"})
+        if extend_seconds:
+            return request("POST", f"{self._base_url}/api/refresh-me", headers=headers, json_body={"extend_seconds": extend_seconds})
+        else:
+            return request("POST", f"{self._base_url}/api/refresh-me", headers=headers)
+
+    async def reborn_async(self, token: str, extend_seconds: Optional[int] = None) -> dict:
+        """异步重新设置过期时间"""
+        headers = self._headers({"Authorization": f"Bearer {token}"})
+        if extend_seconds:
+            return await self._arequest("POST", f"{self._base_url}/api/refresh-me", headers=headers, json_body={"extend_seconds": extend_seconds})
+        else:
+            return await self._arequest("POST", f"{self._base_url}/api/refresh-me", headers=headers)
+
+    # 登出
+    def logout(self, token: str):
+        headers = self._headers({"Authorization": f"Bearer {token}"})
+        return request("POST", f"{self._base_url}/api/auth/logout", headers=headers)
+
+    async def logout_async(self, token: str) -> dict:
+        """异步获取当前用户"""
+        headers = self._headers({"Authorization": f"Bearer {token}"})
+        return await self._arequest("POST", f"{self._base_url}/api/auth/logout", headers=headers)

nc-user-manager/exceptions.py

@@ -0,0 +1,8 @@
+class OAuthError(Exception):
+    """统一封装 OAuth 调用错误"""
+
+    def __init__(self, message: str, status_code: int = None, response: str = None):
+        self.message = message
+        self.status_code = status_code
+        self.response = response
+        super().__init__(f"[{status_code}] {message}" if status_code else message)

nc-user-manager/models.py

@@ -0,0 +1,48 @@
+from typing import Optional
+
+class OAuth2AuthorizeResponse(dict):
+    @property
+    def url(self) -> str:
+        return self.get("authorization_url")
+
+class UserResponse(dict):
+    @property
+    def id(self) -> str:
+        return self.get("id")
+
+    @property
+    def email(self) -> str:
+        return self.get("email")
+
+    @property
+    def picture(self) -> Optional[str]:
+        return self.get("picture")
+
+    @property
+    def full_name(self) -> Optional[str]:
+        return self.get("full_name")
+
+    @property
+    def role(self) -> Optional[str]:
+        return self.get("role")
+
+class CallbackResponse(dict):
+    @property
+    def token(self) -> str:
+        return self.get("access_token")
+
+    @property
+    def token_type(self) -> str:
+        return self.get("token_type")
+
+    @property
+    def user_name(self) -> str:
+        return self.get("user_name")
+
+    @property
+    def email(self) -> str:
+        return self.get("email")
+
+    @property
+    def user_picture(self) -> str:
+        return self.get("user_picture")

nc-user-manager/utils.py

@@ -0,0 +1,39 @@
+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}")

nc_user_terminator-0.1.0.dist-info/METADATA

@@ -0,0 +1,11 @@
+Metadata-Version: 2.4
+Name: nc-user-terminator
+Version: 0.1.0
+Summary: OAuth client wrapper for multi-tenant projects
+Author: bw_song
+Author-email: m132777096902@gmail.com
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Dynamic: author-email
+Dynamic: description-content-type
+Dynamic: requires-python

nc_user_terminator-0.1.0.dist-info/RECORD

@@ -0,0 +1,9 @@
+nc-user-manager/__init__.py,sha256=VHYhkgEcaHYUlOK9z0bJNiOWCBRDpBe29lTUIktZ1Ng,235
+nc-user-manager/client.py,sha256=urNJFx2ssQc5KxmW6SshXfdAJw_Hq4Sm2x208oV20kM,7074
+nc-user-manager/exceptions.py,sha256=yUMDrh1HHZF36UUadQNHvJlDgEYSYoYOObs8Q11fjrE,351
+nc-user-manager/models.py,sha256=mDK7zskIcaThxvUUTWVf6eMasw7YaA3hDGVVSd1ZJgo,1088
+nc-user-manager/utils.py,sha256=0QmJ9s3mzuYQSlwkcS8BW5XCe23lcoKskEeuqfdZHck,1245
+nc_user_terminator-0.1.0.dist-info/METADATA,sha256=rJsRtdiFbiY7LMsH6aBhdm4ntLqQjJQhroUGcqy7LIc,327
+nc_user_terminator-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nc_user_terminator-0.1.0.dist-info/top_level.txt,sha256=VJF5yCswS1mXdkOMLPZqRGrvVORiO5V3mxE5UkEJAIo,16
+nc_user_terminator-0.1.0.dist-info/RECORD,,

nc_user_terminator-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.9.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

nc_user_terminator-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+nc-user-manager