gitcode-api 1.1.0__py3-none-any.whl → 1.1.2__py3-none-any.whl

gitcode_api/_base_client.py

@@ -5,7 +5,7 @@ payload cleanup, and response parsing for both sync and async clients.
 """
 
 import os
-from typing import Any, Dict, Optional, Tuple, Union
+from typing import Any, Callable, Dict, Optional, Tuple, Union
 from urllib.parse import quote
 
 import httpx
@@ -30,6 +30,7 @@ class BaseGitCodeClient:
     :param repo: Default repository name for repository-scoped calls.
     :param base_url: Base URL for the GitCode REST API.
     :param timeout: Request timeout in seconds.
+    :param decrypt: Optional decryption function for encrypted access token.
     """
 
     def __init__(
@@ -40,20 +41,23 @@ class BaseGitCodeClient:
         repo: Optional[str] = None,
         base_url: str = DEFAULT_BASE_URL,
         timeout: Optional[float] = None,
+        decrypt: Optional[Callable] = None,
     ) -> None:
         """Store client configuration and resolve authentication."""
-        self.api_key = self._resolve_api_key(api_key)
+        self.api_key = self._resolve_api_key(api_key, decrypt)
         self.owner = owner
         self.repo = repo
         self.base_url = base_url.rstrip("/")
         self.timeout = timeout if timeout is not None else DEFAULT_TIMEOUT
 
-    def _resolve_api_key(self, api_key: Optional[str]) -> str:
+    def _resolve_api_key(self, api_key: Optional[str], decrypt: Optional[Callable] = None) -> str:
         """Resolve the access token from an argument or environment variable."""
         token = api_key or os.getenv(DEFAULT_TOKEN_ENV)
+        if callable(decrypt):
+            token = decrypt(token)
         if not token:
             raise GitCodeConfigurationError("No API key provided. Pass api_key=... or set GITCODE_ACCESS_TOKEN.")
-        return token
+        return str(token)
 
     def _resolve_repo_context(
         self,
@@ -187,6 +191,7 @@ class SyncAPIClient(BaseGitCodeClient):
     :param base_url: Base URL for the GitCode REST API.
     :param timeout: Request timeout in seconds.
     :param http_client: Optional pre-configured ``httpx.Client`` instance.
+    :param decrypt: Optional decryption function for encrypted access token.
     """
 
     def __init__(
@@ -198,9 +203,10 @@ class SyncAPIClient(BaseGitCodeClient):
         base_url: str = DEFAULT_BASE_URL,
         timeout: Optional[float] = None,
         http_client: Optional[httpx.Client] = None,
+        decrypt: Optional[Callable] = None,
     ) -> None:
         """Create or reuse an ``httpx.Client`` for synchronous requests."""
-        super().__init__(api_key=api_key, owner=owner, repo=repo, base_url=base_url, timeout=timeout)
+        super().__init__(api_key=api_key, owner=owner, repo=repo, base_url=base_url, timeout=timeout, decrypt=decrypt)
         self._owns_client = http_client is None
         self._client = http_client or httpx.Client(timeout=self.timeout)
 
@@ -238,9 +244,8 @@ class SyncAPIClient(BaseGitCodeClient):
         return self._parse_response(response, raw=raw)
 
     def close(self) -> None:
-        """Close the underlying HTTP client if this instance created it."""
-        if self._owns_client:
-            self._client.close()
+        """Close the underlying HTTP client."""
+        self._client.close()
 
     def __enter__(self) -> "SyncAPIClient":
         """Enter a context manager and return the client instance."""
@@ -260,6 +265,7 @@ class AsyncAPIClient(BaseGitCodeClient):
     :param base_url: Base URL for the GitCode REST API.
     :param timeout: Request timeout in seconds.
     :param http_client: Optional pre-configured ``httpx.AsyncClient`` instance.
+    :param decrypt: Optional decryption function for encrypted access token.
     """
 
     def __init__(
@@ -271,9 +277,10 @@ class AsyncAPIClient(BaseGitCodeClient):
         base_url: str = DEFAULT_BASE_URL,
         timeout: Optional[float] = None,
         http_client: Optional[httpx.AsyncClient] = None,
+        decrypt: Optional[Callable] = None,
     ) -> None:
         """Create or reuse an ``httpx.AsyncClient`` for asynchronous requests."""
-        super().__init__(api_key=api_key, owner=owner, repo=repo, base_url=base_url, timeout=timeout)
+        super().__init__(api_key=api_key, owner=owner, repo=repo, base_url=base_url, timeout=timeout, decrypt=decrypt)
         self._owns_client = http_client is None
         self._client = http_client or httpx.AsyncClient(timeout=self.timeout)
 
@@ -311,9 +318,8 @@ class AsyncAPIClient(BaseGitCodeClient):
         return self._parse_response(response, raw=raw)
 
     async def close(self) -> None:
-        """Close the underlying async HTTP client if this instance created it."""
-        if self._owns_client:
-            await self._client.aclose()
+        """Close the underlying async HTTP client."""
+        await self._client.aclose()
 
     async def __aenter__(self) -> "AsyncAPIClient":
         """Enter an async context manager and return the client instance."""

gitcode_api/_client.py

@@ -4,7 +4,7 @@ These client classes expose grouped resource helpers that mirror the
 published GitCode REST API documentation.
 """
 
-from typing import Optional
+from typing import Callable, Optional
 
 import httpx
 
@@ -54,6 +54,7 @@ class GitCode(SyncAPIClient):
     :param base_url: Base URL for the GitCode REST API.
     :param timeout: Request timeout in seconds.
     :param http_client: Optional pre-configured ``httpx.Client`` instance.
+    :param decrypt: Optional decryption function for encrypted access token.
     """
 
     repos: ReposResource
@@ -113,6 +114,7 @@ class GitCode(SyncAPIClient):
         base_url: str = DEFAULT_BASE_URL,
         timeout: Optional[float] = None,
         http_client: Optional[httpx.Client] = None,
+        decrypt: Optional[Callable] = None,
     ) -> None:
         """Create a synchronous client and attach resource groups."""
         super().__init__(
@@ -122,6 +124,7 @@ class GitCode(SyncAPIClient):
             base_url=base_url,
             timeout=timeout,
             http_client=http_client,
+            decrypt=decrypt,
         )
         self.repos = ReposResource(self)
         self.contents = RepoContentsResource(self)
@@ -140,6 +143,9 @@ class GitCode(SyncAPIClient):
         self.search = SearchResource(self)
         self.oauth = OAuthResource(self)
 
+    def __enter__(self) -> "GitCode":
+        return self
+
 
 class AsyncGitCode(AsyncAPIClient):
     """Asynchronous GitCode API client.
@@ -150,6 +156,7 @@ class AsyncGitCode(AsyncAPIClient):
     :param base_url: Base URL for the GitCode REST API.
     :param timeout: Request timeout in seconds.
     :param http_client: Optional pre-configured ``httpx.AsyncClient`` instance.
+    :param decrypt: Optional decryption function for encrypted access token.
     """
 
     repos: AsyncReposResource
@@ -209,6 +216,7 @@ class AsyncGitCode(AsyncAPIClient):
         base_url: str = DEFAULT_BASE_URL,
         timeout: Optional[float] = None,
         http_client: Optional[httpx.AsyncClient] = None,
+        decrypt: Optional[Callable] = None,
     ) -> None:
         """Create an asynchronous client and attach resource groups."""
         super().__init__(
@@ -218,6 +226,7 @@ class AsyncGitCode(AsyncAPIClient):
             base_url=base_url,
             timeout=timeout,
             http_client=http_client,
+            decrypt=decrypt,
         )
         self.repos = AsyncReposResource(self)
         self.contents = AsyncRepoContentsResource(self)
@@ -235,3 +244,6 @@ class AsyncGitCode(AsyncAPIClient):
         self.orgs = AsyncOrgsResource(self)
         self.search = AsyncSearchResource(self)
         self.oauth = AsyncOAuthResource(self)
+
+    async def __aenter__(self) -> "AsyncGitCode":
+        return self