pixelarraythirdparty 1.1.3__tar.gz → 1.1.4__tar.gz

{pixelarraythirdparty-1.1.3 → pixelarraythirdparty-1.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pixelarraythirdparty
-Version: 1.1.3
+Version: 1.1.4
 Summary: PixelArray 第三方微服务客户端
 Author-email: Lu qi <qi.lu@pixelarrayai.com>
 License-Expression: MIT

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

@@ -11,7 +11,7 @@ PixelArray 第三方微服务客户端
 - unified_login: 统一登录模块
 """
 
-__version__ = "1.1.3"
+__version__ = "1.1.4"
 __author__ = "Lu qi"
 __email__ = "qi.lu@pixelarrayai.com"
 

{pixelarraythirdparty-1.1.3 → pixelarraythirdparty-1.1.4}/pixelarraythirdparty/unified_login/unified_login.py

@@ -376,3 +376,86 @@ class DouyinLogin(AsyncClient):
             await asyncio.sleep(interval)
 
         return {}, False
+
+
+class GitLabLogin(AsyncClient):
+    """
+    GitLab OAuth2 登录客户端
+
+    使用示例:
+    ```
+    gitlab = GitLabLogin(api_key="your_api_key")
+    user_info, success = await gitlab.login()
+    ```
+    """
+
+    def __init__(self, api_key: str):
+        super().__init__(api_key)
+
+    async def _get_auth_url(self) -> Tuple[Optional[Dict[str, str]], bool]:
+        data, success = await self._request(
+            "POST", "/api/unified-login/gitlab/auth-url"
+        )
+        if not success:
+            return None, False
+        auth_url = data.get("auth_url")
+        if not auth_url:
+            return None, False
+        return data, True
+
+    async def login(self, timeout: int = 180) -> Tuple[Dict, bool]:
+        """
+        仿 Supabase CLI 的一键登录流程：打开浏览器完成授权，
+        终端端轮询等待登录结果
+
+        :param timeout: 等待用户完成授权的超时时间（秒）
+        """
+        auth_data, success = await self._get_auth_url()
+        if not success or not auth_data:
+            return {}, False
+
+        auth_url = auth_data.get("auth_url")
+        state = auth_data.get("state") or self._extract_state(auth_url)
+
+        if not auth_url or not state:
+            return {}, False
+
+        webbrowser.open(auth_url, new=2)
+
+        return await self._wait_for_gitlab_login(state, timeout)
+
+    def _extract_state(self, auth_url: Optional[str]) -> Optional[str]:
+        if not auth_url:
+            return None
+        try:
+            parsed = urllib.parse.urlparse(auth_url)
+            query = urllib.parse.parse_qs(parsed.query)
+            values = query.get("state")
+            if values:
+                return values[0]
+        except Exception:
+            return None
+        return None
+
+    async def _wait_for_gitlab_login(
+        self, state: str, timeout: int
+    ) -> Tuple[Dict, bool]:
+        interval = 2
+        total_checks = max(1, timeout // interval) if timeout > 0 else 1
+
+        for _ in range(total_checks):
+            status, response = await self._request_raw(
+                "POST",
+                "/api/unified-login/gitlab/wait-login",
+                json={"state": state},
+            )
+
+            if status == 200 and response.get("success") is True:
+                return response.get("data", {}), True
+
+            if status in (400, 408):
+                break
+
+            await asyncio.sleep(interval)
+
+        return {}, False

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pixelarraythirdparty
-Version: 1.1.3
+Version: 1.1.4
 Summary: PixelArray 第三方微服务客户端
 Author-email: Lu qi <qi.lu@pixelarrayai.com>
 License-Expression: MIT

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

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pixelarraythirdparty"
-version = "1.1.3"
+version = "1.1.4"
 authors = [
     {name = "Lu qi", email = "qi.lu@pixelarrayai.com"},
 ]