workos 5.28.0__tar.gz → 5.29.0__tar.gz

{workos-5.28.0 → workos-5.29.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: workos
-Version: 5.28.0
+Version: 5.29.0
 Summary: WorkOS Python Client
 Home-page: https://github.com/workos-inc/workos-python
 Author: WorkOS

{workos-5.28.0 → workos-5.29.0}/tests/test_organizations.py

@@ -2,6 +2,7 @@ import datetime
 from typing import Union
 import pytest
 from tests.types.test_auto_pagination_function import TestAutoPaginationFunction
+from tests.utils.fixtures.mock_feature_flag import MockFeatureFlag
 from tests.utils.fixtures.mock_organization import MockOrganization
 from tests.utils.fixtures.mock_role import MockRole
 from tests.utils.list_resource import list_response_of
@@ -77,6 +78,14 @@ class TestOrganizations:
             "object": "list",
         }
 
+    @pytest.fixture
+    def mock_feature_flags(self):
+        return {
+            "data": [MockFeatureFlag(id=f"flag_{str(i)}").dict() for i in range(2)],
+            "object": "list",
+            "list_metadata": {"before": None, "after": None},
+        }
+
     def test_list_organizations(
         self, mock_organizations, capture_and_mock_http_client_request
     ):
@@ -264,3 +273,28 @@ class TestOrganizations:
             list(map(to_dict, organization_roles_response.data))
             == mock_organization_roles["data"]
         )
+
+    def test_list_feature_flags(
+        self, mock_feature_flags, capture_and_mock_http_client_request
+    ):
+        request_kwargs = capture_and_mock_http_client_request(
+            self.http_client, mock_feature_flags, 200
+        )
+
+        feature_flags_response = syncify(
+            self.organizations.list_feature_flags(
+                organization_id="org_01EHT88Z8J8795GZNQ4ZP1J81T"
+            )
+        )
+
+        def to_dict(x):
+            return x.dict()
+
+        assert request_kwargs["method"] == "get"
+        assert request_kwargs["url"].endswith(
+            "/organizations/org_01EHT88Z8J8795GZNQ4ZP1J81T/feature-flags"
+        )
+        assert (
+            list(map(to_dict, feature_flags_response.data))
+            == mock_feature_flags["data"]
+        )

{workos-5.28.0 → workos-5.29.0}/workos/__about__.py

@@ -12,7 +12,7 @@ __package_name__ = "workos"
 
 __package_url__ = "https://github.com/workos-inc/workos-python"
 
-__version__ = "5.28.0"
+__version__ = "5.29.0"
 
 __author__ = "WorkOS"
 

{workos-5.28.0 → workos-5.29.0}/workos/organizations.py

@@ -1,5 +1,7 @@
 from typing import Optional, Protocol, Sequence
 
+from workos.types.feature_flags import FeatureFlag
+from workos.types.feature_flags.list_filters import FeatureFlagListFilters
 from workos.types.metadata import Metadata
 from workos.types.organizations.domain_data_input import DomainDataInput
 from workos.types.organizations.list_filters import OrganizationListFilters
@@ -24,6 +26,10 @@ OrganizationsListResource = WorkOSListResource[
     Organization, OrganizationListFilters, ListMetadata
 ]
 
+FeatureFlagsListResource = WorkOSListResource[
+    FeatureFlag, FeatureFlagListFilters, ListMetadata
+]
+
 
 class OrganizationsModule(Protocol):
     """Offers methods through the WorkOS Organizations service."""
@@ -128,6 +134,29 @@ class OrganizationsModule(Protocol):
         """
         ...
 
+    def list_feature_flags(
+        self,
+        organization_id: str,
+        *,
+        limit: int = DEFAULT_LIST_RESPONSE_LIMIT,
+        before: Optional[str] = None,
+        after: Optional[str] = None,
+        order: PaginationOrder = "desc",
+    ) -> SyncOrAsync[FeatureFlagsListResource]:
+        """Retrieve a list of feature flags for an organization
+
+        Args:
+            organization_id (str): Organization's unique identifier
+            limit (int): Maximum number of records to return. (Optional)
+            before (str): Pagination cursor to receive records before a provided Feature Flag ID. (Optional)
+            after (str): Pagination cursor to receive records after a provided Feature Flag ID. (Optional)
+            order (Literal["asc","desc"]): Sort records in either ascending or descending (default) order by created_at timestamp. (Optional)
+
+        Returns:
+            FeatureFlagsListResource: Feature flags list response from WorkOS.
+        """
+        ...
+
 
 class Organizations(OrganizationsModule):
     _http_client: SyncHTTPClient
@@ -247,6 +276,34 @@ class Organizations(OrganizationsModule):
 
         return RoleList.model_validate(response)
 
+    def list_feature_flags(
+        self,
+        organization_id: str,
+        *,
+        limit: int = DEFAULT_LIST_RESPONSE_LIMIT,
+        before: Optional[str] = None,
+        after: Optional[str] = None,
+        order: PaginationOrder = "desc",
+    ) -> FeatureFlagsListResource:
+        list_params: FeatureFlagListFilters = {
+            "limit": limit,
+            "before": before,
+            "after": after,
+            "order": order,
+        }
+
+        response = self._http_client.request(
+            f"organizations/{organization_id}/feature-flags",
+            method=REQUEST_METHOD_GET,
+            params=list_params,
+        )
+
+        return WorkOSListResource[FeatureFlag, FeatureFlagListFilters, ListMetadata](
+            list_method=self.list_feature_flags,
+            list_args=list_params,
+            **ListPage[FeatureFlag](**response).model_dump(),
+        )
+
 
 class AsyncOrganizations(OrganizationsModule):
     _http_client: AsyncHTTPClient
@@ -365,3 +422,31 @@ class AsyncOrganizations(OrganizationsModule):
         )
 
         return RoleList.model_validate(response)
+
+    async def list_feature_flags(
+        self,
+        organization_id: str,
+        *,
+        limit: int = DEFAULT_LIST_RESPONSE_LIMIT,
+        before: Optional[str] = None,
+        after: Optional[str] = None,
+        order: PaginationOrder = "desc",
+    ) -> FeatureFlagsListResource:
+        list_params: FeatureFlagListFilters = {
+            "limit": limit,
+            "before": before,
+            "after": after,
+            "order": order,
+        }
+
+        response = await self._http_client.request(
+            f"organizations/{organization_id}/feature-flags",
+            method=REQUEST_METHOD_GET,
+            params=list_params,
+        )
+
+        return WorkOSListResource[FeatureFlag, FeatureFlagListFilters, ListMetadata](
+            list_method=self.list_feature_flags,
+            list_args=list_params,
+            **ListPage[FeatureFlag](**response).model_dump(),
+        )

workos-5.29.0/workos/types/feature_flags/__init__.py

@@ -0,0 +1,3 @@
+from workos.types.feature_flags.feature_flag import FeatureFlag
+
+__all__ = ["FeatureFlag"]

workos-5.29.0/workos/types/feature_flags/feature_flag.py

@@ -0,0 +1,12 @@
+from typing import Literal, Optional
+from workos.types.workos_model import WorkOSModel
+
+
+class FeatureFlag(WorkOSModel):
+    id: str
+    object: Literal["feature_flag"]
+    slug: str
+    name: str
+    description: Optional[str]
+    created_at: str
+    updated_at: str

workos-5.29.0/workos/types/feature_flags/list_filters.py

@@ -0,0 +1,5 @@
+from workos.types.list_resource import ListArgs
+
+
+class FeatureFlagListFilters(ListArgs, total=False):
+    pass

{workos-5.28.0 → workos-5.29.0}/workos/types/list_resource.py

@@ -23,6 +23,7 @@ from workos.types.directory_sync import (
     DirectoryUserWithGroups,
 )
 from workos.types.events import Event
+from workos.types.feature_flags import FeatureFlag
 from workos.types.fga import (
     Warrant,
     AuthorizationResource,
@@ -46,6 +47,7 @@ ListableResource = TypeVar(
     DirectoryGroup,
     DirectoryUserWithGroups,
     Event,
+    FeatureFlag,
     Invitation,
     Organization,
     OrganizationMembership,

{workos-5.28.0 → workos-5.29.0}/workos/types/sso/connection.py

@@ -37,6 +37,7 @@ ConnectionType = Literal[
     "PingFederateSAML",
     "PingOneSAML",
     "RipplingSAML",
+    "SalesforceOAuth",
     "SalesforceSAML",
     "ShibbolethGenericSAML",
     "ShibbolethSAML",

{workos-5.28.0 → workos-5.29.0}/workos/types/sso/sso_provider_type.py

@@ -6,4 +6,5 @@ SsoProviderType = Literal[
     "GitHubOAuth",
     "GoogleOAuth",
     "MicrosoftOAuth",
+    "SalesforceOAuth",
 ]

{workos-5.28.0 → workos-5.29.0}/workos/types/user_management/authentication_response.py

@@ -13,6 +13,7 @@ AuthenticationMethod = Literal[
     "GitHubOAuth",
     "GoogleOAuth",
     "MicrosoftOAuth",
+    "SalesforceOAuth",
     "MagicAuth",
     "Impersonation",
 ]

{workos-5.28.0 → workos-5.29.0}/workos/types/user_management/oauth_tokens.py

@@ -7,6 +7,7 @@ OAuthTokensProvidersType = Literal[
     "GitHubOauth",
     "GoogleOauth",
     "MicrosoftOauth",
+    "SalesforceOauth",
 ]
 
 

workos-5.29.0/workos/types/user_management/user_management_provider_type.py

@@ -0,0 +1,11 @@
+from typing import Literal
+
+
+UserManagementProviderType = Literal[
+    "authkit",
+    "AppleOAuth",
+    "GitHubOAuth",
+    "GoogleOAuth",
+    "MicrosoftOAuth",
+    "SalesforceOAuth",
+]

{workos-5.28.0 → workos-5.29.0}/workos/user_management.py

@@ -396,7 +396,7 @@ class UserManagementModule(Protocol):
             organization_id (str): The organization_id connection selector is used to initiate SSO for an Organization.
                 The value of this parameter should be a WorkOS Organization ID. (Optional)
             provider (UserManagementProviderType): The provider connection selector is used to initiate SSO using an OAuth-compatible provider.
-                Currently, the supported values for provider are 'authkit', 'AppleOAuth', 'GitHubOAuth, 'GoogleOAuth', and 'MicrosoftOAuth'. (Optional)
+                Currently, the supported values for provider are 'authkit', 'AppleOAuth', 'GitHubOAuth, 'GoogleOAuth', 'MicrosoftOAuth', and 'SalesforceOAuth'. (Optional)
             provider_scopes (Sequence[str]): Can be used to specify additional scopes that will be requested when initiating SSO using an OAuth provider. (Optional)
             domain_hint (str): Can be used to pre-fill the domain field when initiating authentication with Microsoft OAuth,
                 or with a GoogleSAML connection type. (Optional)

{workos-5.28.0 → workos-5.29.0}/workos.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: workos
-Version: 5.28.0
+Version: 5.29.0
 Summary: WorkOS Python Client
 Home-page: https://github.com/workos-inc/workos-python
 Author: WorkOS

{workos-5.28.0 → workos-5.29.0}/workos.egg-info/SOURCES.txt

@@ -81,6 +81,9 @@ workos/types/events/list_filters.py
 workos/types/events/organization_domain_verification_failed_payload.py
 workos/types/events/previous_attributes.py
 workos/types/events/session_created_payload.py
+workos/types/feature_flags/__init__.py
+workos/types/feature_flags/feature_flag.py
+workos/types/feature_flags/list_filters.py
 workos/types/fga/__init__.py
 workos/types/fga/authorization_resource_types.py
 workos/types/fga/authorization_resources.py

workos-5.28.0/workos/types/user_management/user_management_provider_type.py

@@ -1,6 +0,0 @@
-from typing import Literal
-
-
-UserManagementProviderType = Literal[
-    "authkit", "AppleOAuth", "GitHubOAuth", "GoogleOAuth", "MicrosoftOAuth"
-]