fixed double client and formatting as well

Author: vaibhavatlan

pyatlan/client/aio/__init__.py

@@ -35,7 +35,7 @@
 from .file import AsyncFileClient
 from .group import AsyncGroupClient
 from .impersonate import AsyncImpersonationClient
-from .oauth_client import AsyncOAuthClientClient
+from .oauth_client import AsyncOAuthClient
 from .open_lineage import AsyncOpenLineageClient
 from .query import AsyncQueryClient
 from .role import AsyncRoleClient
@@ -62,7 +62,7 @@
     "AsyncImpersonationClient",
     "AsyncIndexSearchResults",
     "AsyncLineageListResults",
-    "AsyncOAuthClientClient",
+    "AsyncOAuthClient",
     "AsyncOpenLineageClient",
     "AsyncQueryClient",
     "AsyncRoleClient",

pyatlan/client/aio/client.py

@@ -42,7 +42,7 @@
 from pyatlan.client.aio.group import AsyncGroupClient
 from pyatlan.client.aio.impersonate import AsyncImpersonationClient
 from pyatlan.client.aio.oauth import AsyncOAuthTokenManager
-from pyatlan.client.aio.oauth_client import AsyncOAuthClientClient
+from pyatlan.client.aio.oauth_client import AsyncOAuthClient
 from pyatlan.client.aio.open_lineage import AsyncOpenLineageClient
 from pyatlan.client.aio.query import AsyncQueryClient
 from pyatlan.client.aio.role import AsyncRoleClient
@@ -114,9 +114,7 @@ class AsyncAtlanClient(AtlanClient):
     _async_sso_client: Optional[AsyncSSOClient] = PrivateAttr(default=None)
     _async_task_client: Optional[AsyncTaskClient] = PrivateAttr(default=None)
     _async_token_client: Optional[AsyncTokenClient] = PrivateAttr(default=None)
-    _async_oauth_client_client: Optional[AsyncOAuthClientClient] = PrivateAttr(
-        default=None
-    )
+    _async_oauth_client_client: Optional[AsyncOAuthClient] = PrivateAttr(default=None)
     _async_typedef_client: Optional[AsyncTypeDefClient] = PrivateAttr(default=None)
     _async_user_client: Optional[AsyncUserClient] = PrivateAttr(default=None)
     _async_workflow_client: Optional[AsyncWorkflowClient] = PrivateAttr(default=None)
@@ -367,10 +365,10 @@ def token(self) -> AsyncTokenClient:  # type: ignore[override]
         return self._async_token_client
 
     @property
-    def oauth_client(self) -> AsyncOAuthClientClient:  # type: ignore[override]
+    def oauth_client(self) -> AsyncOAuthClient:  # type: ignore[override]
         """Get async OAuth client client with same API as sync"""
         if self._async_oauth_client_client is None:
-            self._async_oauth_client_client = AsyncOAuthClientClient(self)  # type: ignore[arg-type]
+            self._async_oauth_client_client = AsyncOAuthClient(self)  # type: ignore[arg-type]
         return self._async_oauth_client_client
 
     @property

pyatlan/client/aio/oauth_client.py

@@ -18,13 +18,10 @@
 )
 from pyatlan.errors import ErrorCode
 from pyatlan.model.aio.oauth_client import AsyncOAuthClientListResponse
-from pyatlan.model.oauth_client import (
-    OAuthClientCreateResponse,
-    OAuthClientResponse,
-)
+from pyatlan.model.oauth_client import OAuthClientCreateResponse, OAuthClientResponse
 
 
-class AsyncOAuthClientClient:
+class AsyncOAuthClient:
     """
     Async client for managing OAuth client credentials.
     """

pyatlan/client/atlan.py

@@ -49,7 +49,7 @@
 from pyatlan.client.group import GroupClient
 from pyatlan.client.impersonate import ImpersonationClient
 from pyatlan.client.oauth import OAuthTokenManager
-from pyatlan.client.oauth_client import OAuthClientClient
+from pyatlan.client.oauth_client import OAuthClient
 from pyatlan.client.open_lineage import OpenLineageClient
 from pyatlan.client.query import QueryClient
 from pyatlan.client.role import RoleClient
@@ -152,7 +152,7 @@ class AtlanClient(BaseSettings):
     _asset_client: Optional[AssetClient] = PrivateAttr(default=None)
     _typedef_client: Optional[TypeDefClient] = PrivateAttr(default=None)
     _token_client: Optional[TokenClient] = PrivateAttr(default=None)
-    _oauth_client_client: Optional[OAuthClientClient] = PrivateAttr(default=None)
+    _oauth_client_client: Optional[OAuthClient] = PrivateAttr(default=None)
     _user_client: Optional[UserClient] = PrivateAttr(default=None)
     _impersonate_client: Optional[ImpersonationClient] = PrivateAttr(default=None)
     _query_client: Optional[QueryClient] = PrivateAttr(default=None)
@@ -346,9 +346,9 @@ def token(self) -> TokenClient:
         return self._token_client
 
     @property
-    def oauth_client(self) -> OAuthClientClient:
+    def oauth_client(self) -> OAuthClient:
         if self._oauth_client_client is None:
-            self._oauth_client_client = OAuthClientClient(client=self)
+            self._oauth_client_client = OAuthClient(client=self)
         return self._oauth_client_client
 
     @property

pyatlan/client/oauth_client.py

@@ -23,7 +23,7 @@
 )
 
 
-class OAuthClientClient:
+class OAuthClient:
     """
     This class can be used to manage OAuth client credentials.
     This class does not need to be instantiated directly but can be

pyatlan/model/oauth_client.py

@@ -155,7 +155,9 @@ def current_page(self) -> Optional[List[OAuthClientResponse]]:
         """Get the current page of OAuth clients."""
         return self.records
 
-    def next_page(self, start: Optional[int] = None, size: Optional[int] = None) -> bool:
+    def next_page(
+        self, start: Optional[int] = None, size: Optional[int] = None
+    ) -> bool:
         """
         Retrieve the next page of results.