workos 1.13.0__py3-none-any.whl → 5.38.0__py3-none-any.whl

workos/directory_sync.py

@@ -1,212 +1,435 @@
-import workos
-from workos.utils.pagiantion_order import Order
-from workos.utils.request import (
-    RequestHelper,
+from typing import Optional, Protocol
+
+from workos.types.directory_sync.list_filters import (
+    DirectoryGroupListFilters,
+    DirectoryListFilters,
+    DirectoryUserListFilters,
+)
+from workos.typing.sync_or_async import SyncOrAsync
+from workos.utils.http_client import AsyncHTTPClient, SyncHTTPClient
+from workos.utils.pagination_order import PaginationOrder
+from workos.utils.request_helper import (
+    DEFAULT_LIST_RESPONSE_LIMIT,
     REQUEST_METHOD_DELETE,
     REQUEST_METHOD_GET,
 )
-from workos.utils.validation import DIRECTORY_SYNC_MODULE, validate_settings
-from workos.resources.directory_sync import WorkOSDirectoryGroup
+from workos.types.directory_sync import (
+    DirectoryGroup,
+    Directory,
+    DirectoryUserWithGroups,
+)
+from workos.types.list_resource import ListMetadata, ListPage, WorkOSListResource
 
-RESPONSE_LIMIT = 10
+DirectoryUsersListResource = WorkOSListResource[
+    DirectoryUserWithGroups, DirectoryUserListFilters, ListMetadata
+]
 
+DirectoryGroupsListResource = WorkOSListResource[
+    DirectoryGroup, DirectoryGroupListFilters, ListMetadata
+]
 
-class DirectorySync(object):
-    """Offers methods through the WorkOS Directory Sync service."""
+DirectoriesListResource = WorkOSListResource[
+    Directory, DirectoryListFilters, ListMetadata
+]
 
-    @validate_settings(DIRECTORY_SYNC_MODULE)
-    def __init__(self):
-        pass
 
-    @property
-    def request_helper(self):
-        if not getattr(self, "_request_helper", None):
-            self._request_helper = RequestHelper()
-        return self._request_helper
+class DirectorySyncModule(Protocol):
+    """Offers methods through the WorkOS Directory Sync service."""
 
     def list_users(
         self,
-        directory=None,
-        group=None,
-        limit=RESPONSE_LIMIT,
-        before=None,
-        after=None,
-        order=None,
-    ):
+        *,
+        directory_id: Optional[str] = None,
+        group_id: Optional[str] = None,
+        limit: int = DEFAULT_LIST_RESPONSE_LIMIT,
+        before: Optional[str] = None,
+        after: Optional[str] = None,
+        order: PaginationOrder = "desc",
+    ) -> SyncOrAsync[DirectoryUsersListResource]:
         """Gets a list of provisioned Users for a Directory.
 
-        Note, either 'directory' or 'group' must be provided.
+        Note, either 'directory_id' or 'group_id' must be provided.
+
+        Kwargs:
+            directory_id (str): Directory unique identifier. (Optional)
+            group_id (str): Directory Group unique identifier. (Optional)
+            limit (int): Maximum number of records to return. (Optional)
+            before (str): Pagination cursor to receive records before a provided Directory ID. (Optional)
+            after (str): Pagination cursor to receive records after a provided Directory ID. (Optional)
+            order (Literal["asc","desc"]): Sort records in either ascending or descending (default) order by created_at timestamp. (Optional)
+
+        Returns:
+            DirectoryUsersListResource: Directory Users response from WorkOS.
+        """
+        ...
+
+    def list_groups(
+        self,
+        *,
+        directory_id: Optional[str] = None,
+        user_id: Optional[str] = None,
+        limit: int = DEFAULT_LIST_RESPONSE_LIMIT,
+        before: Optional[str] = None,
+        after: Optional[str] = None,
+        order: PaginationOrder = "desc",
+    ) -> SyncOrAsync[DirectoryGroupsListResource]:
+        """Gets a list of provisioned Groups for a Directory .
+
+        Note, either 'directory_id' or 'user_id' must be provided.
+
+        Kwargs:
+            directory_id (str): Directory unique identifier. (Optional)
+            user_id (str): Directory User unique identifier. (Optional)
+            limit (int): Maximum number of records to return. (Optional)
+            before (str): Pagination cursor to receive records before a provided Directory ID. (Optional)
+            after (str): Pagination cursor to receive records after a provided Directory ID. (Optional)
+            order (Literal["asc","desc"]): Sort records in either ascending or descending (default) order by created_at timestamp. (Optional)
+
+        Returns:
+            DirectoryGroupsListResource: Directory Groups response from WorkOS.
+        """
+        ...
+
+    def list_directories(
+        self,
+        *,
+        search: Optional[str] = None,
+        limit: int = DEFAULT_LIST_RESPONSE_LIMIT,
+        before: Optional[str] = None,
+        after: Optional[str] = None,
+        organization_id: Optional[str] = None,
+        order: PaginationOrder = "desc",
+    ) -> SyncOrAsync[DirectoriesListResource]:
+        """Gets details for existing Directories.
+
+        Kwargs:
+            organization_id: ID of an Organization (Optional)
+            search (str): Searchable text for a Directory. (Optional)
+            limit (int): Maximum number of records to return. (Optional)
+            before (str): Pagination cursor to receive records before a provided Directory ID. (Optional)
+            after (str): Pagination cursor to receive records after a provided Directory ID. (Optional)
+            order (Literal["asc","desc"]): Sort records in either ascending or descending (default) order by created_at timestamp. (Optional)
+
+        Returns:
+            DirectoriesListResource: Directories response from WorkOS.
+        """
+        ...
+
+    def get_user(self, user_id: str) -> SyncOrAsync[DirectoryUserWithGroups]:
+        """Gets details for a single provisioned Directory User.
 
         Args:
-            directory (str): Directory unique identifier.
-            group (str): Directory Group unique identifier.
-            limit (int): Maximum number of records to return.
-            before (str): Pagination cursor to receive records before a provided Directory ID.
-            after (str): Pagination cursor to receive records after a provided Directory ID.
-            order (Order): Sort records in either ascending or descending order by created_at timestamp.
+            user_id (str): Directory User unique identifier.
 
         Returns:
-            dict: Directory Users response from WorkOS.
+            DirectoryUserWithGroups: Directory User response from WorkOS.
         """
-        params = {
+        ...
+
+    def get_group(self, group_id: str) -> SyncOrAsync[DirectoryGroup]:
+        """Gets details for a single provisioned Directory Group.
+
+        Args:
+            group_id (str): Directory Group unique identifier.
+
+        Returns:
+            DirectoryGroup: Directory Group response from WorkOS.
+        """
+        ...
+
+    def get_directory(self, directory_id: str) -> SyncOrAsync[Directory]:
+        """Gets details for a single Directory
+
+        Args:
+            directory_id (str): Directory unique identifier.
+
+        Returns:
+            Directory: Directory response from WorkOS
+        """
+        ...
+
+    def delete_directory(self, directory_id: str) -> SyncOrAsync[None]:
+        """Delete one existing Directory.
+
+        Args:
+            directory_id (str): Directory unique identifier.
+
+        Returns:
+            None
+        """
+        ...
+
+
+class DirectorySync(DirectorySyncModule):
+    _http_client: SyncHTTPClient
+
+    def __init__(self, http_client: SyncHTTPClient) -> None:
+        self._http_client = http_client
+
+    def list_users(
+        self,
+        *,
+        directory_id: Optional[str] = None,
+        group_id: Optional[str] = None,
+        limit: int = DEFAULT_LIST_RESPONSE_LIMIT,
+        before: Optional[str] = None,
+        after: Optional[str] = None,
+        order: PaginationOrder = "desc",
+    ) -> DirectoryUsersListResource:
+
+        list_params: DirectoryUserListFilters = {
             "limit": limit,
             "before": before,
             "after": after,
             "order": order,
         }
-        if group is not None:
-            params["group"] = group
-        if directory is not None:
-            params["directory"] = directory
-        if order is not None:
-            if not isinstance(order, Order):
-                raise ValueError("'order' must be of asc or desc order")
-            params["order"] = str(order.value)
-        return self.request_helper.request(
+
+        if group_id is not None:
+            list_params["group"] = group_id
+        if directory_id is not None:
+            list_params["directory"] = directory_id
+
+        response = self._http_client.request(
             "directory_users",
             method=REQUEST_METHOD_GET,
-            params=params,
-            token=workos.api_key,
+            params=list_params,
+        )
+
+        return WorkOSListResource(
+            list_method=self.list_users,
+            list_args=list_params,
+            **ListPage[DirectoryUserWithGroups](**response).model_dump(),
         )
 
     def list_groups(
         self,
-        directory=None,
-        user=None,
-        limit=RESPONSE_LIMIT,
-        before=None,
-        after=None,
-        order=None,
-    ):
-        """Gets a list of provisioned Groups for a Directory .
-
-        Note, either 'directory' or 'user' must be provided.
+        *,
+        directory_id: Optional[str] = None,
+        user_id: Optional[str] = None,
+        limit: int = DEFAULT_LIST_RESPONSE_LIMIT,
+        before: Optional[str] = None,
+        after: Optional[str] = None,
+        order: PaginationOrder = "desc",
+    ) -> DirectoryGroupsListResource:
+        list_params: DirectoryGroupListFilters = {
+            "limit": limit,
+            "before": before,
+            "after": after,
+            "order": order,
+        }
 
-        Args:
-            directory (str): Directory unique identifier.
-            user (str): Directory User unique identifier.
-            limit (int): Maximum number of records to return.
-            before (str): Pagination cursor to receive records before a provided Directory ID.
-            after (str): Pagination cursor to receive records after a provided Directory ID.
-            order (Order): Sort records in either ascending or descending order by created_at timestamp.
+        if user_id is not None:
+            list_params["user"] = user_id
+        if directory_id is not None:
+            list_params["directory"] = directory_id
 
-        Returns:
-            dict: Directory Groups response from WorkOS.
-        """
-        params = {"limit": limit, "before": before, "after": after, "order": order}
-        if user is not None:
-            params["user"] = user
-        if directory is not None:
-            params["directory"] = directory
-        if order is not None:
-            if not isinstance(order, Order):
-                raise ValueError("'order' must be of asc or desc order")
-            params["order"] = str(order.value)
-        return self.request_helper.request(
+        response = self._http_client.request(
             "directory_groups",
             method=REQUEST_METHOD_GET,
-            params=params,
-            token=workos.api_key,
+            params=list_params,
         )
 
-    def get_user(self, user):
-        """Gets details for a single provisioned Directory User.
-
-        Args:
-            user (str): Directory User unique identifier.
+        return WorkOSListResource[
+            DirectoryGroup, DirectoryGroupListFilters, ListMetadata
+        ](
+            list_method=self.list_groups,
+            list_args=list_params,
+            **ListPage[DirectoryGroup](**response).model_dump(),
+        )
 
-        Returns:
-            dict: Directory User response from WorkOS.
-        """
-        return self.request_helper.request(
-            "directory_users/{user}".format(user=user),
+    def get_user(self, user_id: str) -> DirectoryUserWithGroups:
+        response = self._http_client.request(
+            "directory_users/{user}".format(user=user_id),
             method=REQUEST_METHOD_GET,
-            token=workos.api_key,
         )
 
-    def get_group(self, group):
-        """Gets details for a single provisioned Directory Group.
+        return DirectoryUserWithGroups.model_validate(response)
 
-        Args:
-            group (str): Directory Group unique identifier.
+    def get_group(self, group_id: str) -> DirectoryGroup:
+        response = self._http_client.request(
+            "directory_groups/{group}".format(group=group_id),
+            method=REQUEST_METHOD_GET,
+        )
+        return DirectoryGroup.model_validate(response)
 
-        Returns:
-            dict: Directory Group response from WorkOS.
-        """
-        response = self.request_helper.request(
-            "directory_groups/{group}".format(group=group),
+    def get_directory(self, directory_id: str) -> Directory:
+        response = self._http_client.request(
+            "directories/{directory}".format(directory=directory_id),
             method=REQUEST_METHOD_GET,
-            token=workos.api_key,
         )
 
-        return WorkOSDirectoryGroup.construct_from_response(response).to_dict()
+        return Directory.model_validate(response)
 
     def list_directories(
         self,
-        domain=None,
-        search=None,
-        limit=RESPONSE_LIMIT,
-        before=None,
-        after=None,
-        organization=None,
-        order=None,
-    ):
-        """Gets details for existing Directories.
+        *,
+        search: Optional[str] = None,
+        limit: int = DEFAULT_LIST_RESPONSE_LIMIT,
+        before: Optional[str] = None,
+        after: Optional[str] = None,
+        organization_id: Optional[str] = None,
+        order: PaginationOrder = "desc",
+    ) -> DirectoriesListResource:
+        list_params: DirectoryListFilters = {
+            "limit": limit,
+            "before": before,
+            "after": after,
+            "order": order,
+            "organization_id": organization_id,
+            "search": search,
+        }
 
-        Args:
-            domain (str): Domain of a Directory. (Optional)
-            organization: ID of an Organization (Optional)
-            search (str): Searchable text for a Directory. (Optional)
-            limit (int): Maximum number of records to return. (Optional)
-            before (str): Pagination cursor to receive records before a provided Directory ID. (Optional)
-            after (str): Pagination cursor to receive records after a provided Directory ID. (Optional)
-            order (Order): Sort records in either ascending or descending order by created_at timestamp.
+        response = self._http_client.request(
+            "directories",
+            method=REQUEST_METHOD_GET,
+            params=list_params,
+        )
+        return WorkOSListResource[Directory, DirectoryListFilters, ListMetadata](
+            list_method=self.list_directories,
+            list_args=list_params,
+            **ListPage[Directory](**response).model_dump(),
+        )
 
-        Returns:
-            dict: Directories response from WorkOS.
-        """
-        params = {
-            "domain": domain,
-            "organization_id": organization,
-            "search": search,
+    def delete_directory(self, directory_id: str) -> None:
+        self._http_client.request(
+            "directories/{directory}".format(directory=directory_id),
+            method=REQUEST_METHOD_DELETE,
+        )
+
+
+class AsyncDirectorySync(DirectorySyncModule):
+    """Offers methods through the WorkOS Directory Sync service."""
+
+    _http_client: AsyncHTTPClient
+
+    def __init__(self, http_client: AsyncHTTPClient):
+        self._http_client = http_client
+
+    async def list_users(
+        self,
+        *,
+        directory_id: Optional[str] = None,
+        group_id: Optional[str] = None,
+        limit: int = DEFAULT_LIST_RESPONSE_LIMIT,
+        before: Optional[str] = None,
+        after: Optional[str] = None,
+        order: PaginationOrder = "desc",
+    ) -> DirectoryUsersListResource:
+
+        list_params: DirectoryUserListFilters = {
             "limit": limit,
             "before": before,
             "after": after,
             "order": order,
         }
-        return self.request_helper.request(
-            "directories",
+
+        if group_id is not None:
+            list_params["group"] = group_id
+        if directory_id is not None:
+            list_params["directory"] = directory_id
+
+        response = await self._http_client.request(
+            "directory_users",
             method=REQUEST_METHOD_GET,
-            params=params,
-            token=workos.api_key,
+            params=list_params,
         )
 
-    def get_directory(self, directory_id):
-        """Gets details for a single Directory.
+        return WorkOSListResource(
+            list_method=self.list_users,
+            list_args=list_params,
+            **ListPage[DirectoryUserWithGroups](**response).model_dump(),
+        )
 
-        Args:
-            directory_id (str): Directory unique identifier.
+    async def list_groups(
+        self,
+        *,
+        directory_id: Optional[str] = None,
+        user_id: Optional[str] = None,
+        limit: int = DEFAULT_LIST_RESPONSE_LIMIT,
+        before: Optional[str] = None,
+        after: Optional[str] = None,
+        order: PaginationOrder = "desc",
+    ) -> DirectoryGroupsListResource:
+        list_params: DirectoryGroupListFilters = {
+            "limit": limit,
+            "before": before,
+            "after": after,
+            "order": order,
+        }
+        if user_id is not None:
+            list_params["user"] = user_id
+        if directory_id is not None:
+            list_params["directory"] = directory_id
 
-        Returns:
-            dict: Directory response from WorkOS.
-        """
-        return self.request_helper.request(
-            "directories/{directory_id}".format(directory_id=directory_id),
+        response = await self._http_client.request(
+            "directory_groups",
             method=REQUEST_METHOD_GET,
-            token=workos.api_key,
+            params=list_params,
         )
 
-    def delete_directory(self, directory):
-        """Delete one existing Directory.
+        return WorkOSListResource[
+            DirectoryGroup, DirectoryGroupListFilters, ListMetadata
+        ](
+            list_method=self.list_groups,
+            list_args=list_params,
+            **ListPage[DirectoryGroup](**response).model_dump(),
+        )
 
-        Args:
-            directory (str): The ID of the directory to be deleted. (Required)
+    async def get_user(self, user_id: str) -> DirectoryUserWithGroups:
+        response = await self._http_client.request(
+            "directory_users/{user}".format(user=user_id),
+            method=REQUEST_METHOD_GET,
+        )
 
-        Returns:
-            dict: Directories response from WorkOS.
-        """
-        return self.request_helper.request(
-            "directories/{directory}".format(directory=directory),
+        return DirectoryUserWithGroups.model_validate(response)
+
+    async def get_group(self, group_id: str) -> DirectoryGroup:
+        response = await self._http_client.request(
+            "directory_groups/{group}".format(group=group_id),
+            method=REQUEST_METHOD_GET,
+        )
+        return DirectoryGroup.model_validate(response)
+
+    async def get_directory(self, directory_id: str) -> Directory:
+        response = await self._http_client.request(
+            "directories/{directory}".format(directory=directory_id),
+            method=REQUEST_METHOD_GET,
+        )
+
+        return Directory.model_validate(response)
+
+    async def list_directories(
+        self,
+        *,
+        search: Optional[str] = None,
+        limit: int = DEFAULT_LIST_RESPONSE_LIMIT,
+        before: Optional[str] = None,
+        after: Optional[str] = None,
+        organization_id: Optional[str] = None,
+        order: PaginationOrder = "desc",
+    ) -> DirectoriesListResource:
+        list_params: DirectoryListFilters = {
+            "organization_id": organization_id,
+            "search": search,
+            "limit": limit,
+            "before": before,
+            "after": after,
+            "order": order,
+        }
+
+        response = await self._http_client.request(
+            "directories",
+            method=REQUEST_METHOD_GET,
+            params=list_params,
+        )
+        return WorkOSListResource[Directory, DirectoryListFilters, ListMetadata](
+            list_method=self.list_directories,
+            list_args=list_params,
+            **ListPage[Directory](**response).model_dump(),
+        )
+
+    async def delete_directory(self, directory_id: str) -> None:
+        await self._http_client.request(
+            "directories/{directory}".format(directory=directory_id),
             method=REQUEST_METHOD_DELETE,
-            token=workos.api_key,
         )