workos 5.35.0__tar.gz → 5.37.0__tar.gz

{workos-5.35.0 → workos-5.37.0}/PKG-INFO

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

{workos-5.35.0 → workos-5.37.0}/tests/test_user_management.py

@@ -6,6 +6,7 @@ import pytest
 
 from tests.utils.fixtures.mock_auth_factor_totp import MockAuthenticationFactorTotp
 from tests.utils.fixtures.mock_email_verification import MockEmailVerification
+from tests.utils.fixtures.mock_feature_flag import MockFeatureFlag
 from tests.utils.fixtures.mock_invitation import MockInvitation
 from tests.utils.fixtures.mock_magic_auth import MockMagicAuth
 from tests.utils.fixtures.mock_organization_membership import MockOrganizationMembership
@@ -146,6 +147,14 @@ class UserManagementFixtures:
         invitations_list = [MockInvitation(id=str(i)).dict() for i in range(40)]
         return list_response_of(data=invitations_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},
+        }
+
 
 class TestUserManagementBase(UserManagementFixtures):
     @pytest.fixture(autouse=True)
@@ -1250,3 +1259,28 @@ class TestUserManagement(UserManagementFixtures):
 
         with pytest.raises(Exception):
             syncify(self.user_management.resend_invitation("invitation_accepted"))
+
+    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.user_management.list_feature_flags(
+                user_id="user_01H7ZGXFP5C6BBQY6Z7277ZCT0"
+            )
+        )
+
+        def to_dict(x):
+            return x.dict()
+
+        assert request_kwargs["method"] == "get"
+        assert request_kwargs["url"].endswith(
+            "/user_management/users/user_01H7ZGXFP5C6BBQY6Z7277ZCT0/feature-flags"
+        )
+        assert (
+            list(map(to_dict, feature_flags_response.data))
+            == mock_feature_flags["data"]
+        )

{workos-5.35.0 → workos-5.37.0}/workos/__about__.py

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

{workos-5.35.0 → workos-5.37.0}/workos/types/events/__init__.py

@@ -10,4 +10,4 @@ from .event_type import *
 from .event import *
 from .organization_domain_verification_failed_payload import *
 from .previous_attributes import *
-from .session_created_payload import *
+from .session_payload import *

{workos-5.35.0 → workos-5.37.0}/workos/types/events/event.py

@@ -36,7 +36,10 @@ from workos.types.events.event_model import EventModel
 from workos.types.events.organization_domain_verification_failed_payload import (
     OrganizationDomainVerificationFailedPayload,
 )
-from workos.types.events.session_created_payload import SessionCreatedPayload
+from workos.types.events.session_payload import (
+    SessionCreatedPayload,
+    SessionRevokedPayload,
+)
 from workos.types.organizations.organization_common import OrganizationCommon
 from workos.types.organization_domains import OrganizationDomain
 from workos.types.roles.role import EventRole
@@ -249,6 +252,10 @@ class SessionCreatedEvent(EventModel[SessionCreatedPayload]):
     event: Literal["session.created"]
 
 
+class SessionRevokedEvent(EventModel[SessionRevokedPayload]):
+    event: Literal["session.revoked"]
+
+
 class UserCreatedEvent(EventModel[User]):
     event: Literal["user.created"]
 
@@ -308,6 +315,7 @@ Event = Annotated[
         RoleDeletedEvent,
         RoleUpdatedEvent,
         SessionCreatedEvent,
+        SessionRevokedEvent,
         UserCreatedEvent,
         UserDeletedEvent,
         UserUpdatedEvent,

{workos-5.35.0 → workos-5.37.0}/workos/types/events/event_model.py

@@ -35,7 +35,11 @@ from workos.types.events.directory_user_with_previous_attributes import (
 from workos.types.events.organization_domain_verification_failed_payload import (
     OrganizationDomainVerificationFailedPayload,
 )
-from workos.types.events.session_created_payload import SessionCreatedPayload
+
+from workos.types.events.session_payload import (
+    SessionCreatedPayload,
+    SessionRevokedPayload,
+)
 from workos.types.organizations.organization_common import OrganizationCommon
 from workos.types.organization_domains import OrganizationDomain
 from workos.types.roles.role import EventRole
@@ -81,6 +85,7 @@ EventPayload = TypeVar(
     OrganizationMembership,
     PasswordResetCommon,
     SessionCreatedPayload,
+    SessionRevokedPayload,
     User,
 )
 

{workos-5.35.0 → workos-5.37.0}/workos/types/events/event_type.py

@@ -50,6 +50,7 @@ EventType = Literal[
     "role.deleted",
     "role.updated",
     "session.created",
+    "session.revoked",
     "user.created",
     "user.deleted",
     "user.updated",

workos-5.35.0/workos/types/events/session_created_payload.py → workos-5.37.0/workos/types/events/session_payload.py

@@ -13,3 +13,15 @@ class SessionCreatedPayload(WorkOSModel):
     user_id: str
     created_at: str
     updated_at: str
+
+
+class SessionRevokedPayload(WorkOSModel):
+    object: Literal["session"]
+    id: str
+    impersonator: Optional[Impersonator] = None
+    ip_address: Optional[str] = None
+    organization_id: Optional[str] = None
+    user_agent: Optional[str] = None
+    user_id: str
+    created_at: str
+    updated_at: str

{workos-5.35.0 → workos-5.37.0}/workos/types/webhooks/webhook.py

@@ -36,7 +36,10 @@ from workos.types.events.directory_user_with_previous_attributes import (
 from workos.types.events.organization_domain_verification_failed_payload import (
     OrganizationDomainVerificationFailedPayload,
 )
-from workos.types.events.session_created_payload import SessionCreatedPayload
+from workos.types.events.session_payload import (
+    SessionCreatedPayload,
+    SessionRevokedPayload,
+)
 from workos.types.organization_domains import OrganizationDomain
 from workos.types.organizations.organization_common import OrganizationCommon
 from workos.types.roles.role import EventRole
@@ -255,6 +258,10 @@ class SessionCreatedWebhook(WebhookModel[SessionCreatedPayload]):
     event: Literal["session.created"]
 
 
+class SessionRevokedWebhook(WebhookModel[SessionRevokedPayload]):
+    event: Literal["session.revoked"]
+
+
 class UserCreatedWebhook(WebhookModel[User]):
     event: Literal["user.created"]
 
@@ -314,6 +321,7 @@ Webhook = Annotated[
         RoleDeletedWebhook,
         RoleUpdatedWebhook,
         SessionCreatedWebhook,
+        SessionRevokedWebhook,
         UserCreatedWebhook,
         UserDeletedWebhook,
         UserUpdatedWebhook,

{workos-5.35.0 → workos-5.37.0}/workos/user_management.py

@@ -2,6 +2,8 @@ from typing import Awaitable, Optional, Protocol, Sequence, Type, Union, cast
 from urllib.parse import urlencode
 from workos._client_configuration import ClientConfiguration
 from workos.session import AsyncSession, Session
+from workos.types.feature_flags import FeatureFlag
+from workos.types.feature_flags.list_filters import FeatureFlagListFilters
 from workos.types.list_resource import (
     ListArgs,
     ListMetadata,
@@ -97,6 +99,7 @@ INVITATION_REVOKE_PATH = "user_management/invitations/{0}/revoke"
 INVITATION_RESEND_PATH = "user_management/invitations/{0}/resend"
 PASSWORD_RESET_PATH = "user_management/password_reset"
 PASSWORD_RESET_DETAIL_PATH = "user_management/password_reset/{0}"
+USER_FEATURE_FLAGS_PATH = "user_management/users/{0}/feature-flags"
 
 
 UsersListResource = WorkOSListResource[User, UsersListFilters, ListMetadata]
@@ -113,6 +116,10 @@ InvitationsListResource = WorkOSListResource[
     Invitation, InvitationsListFilters, ListMetadata
 ]
 
+FeatureFlagsListResource = WorkOSListResource[
+    FeatureFlag, FeatureFlagListFilters, ListMetadata
+]
+
 from workos.types.user_management.list_filters import SessionsListFilters
 
 SessionsListResource = WorkOSListResource[
@@ -908,6 +915,29 @@ class UserManagementModule(Protocol):
         """
         ...
 
+    def list_feature_flags(
+        self,
+        user_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 a user
+
+        Args:
+            user_id (str): User'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 UserManagement(UserManagementModule):
     _http_client: SyncHTTPClient
@@ -1603,6 +1633,34 @@ class UserManagement(UserManagementModule):
 
         return Invitation.model_validate(response)
 
+    def list_feature_flags(
+        self,
+        user_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(
+            USER_FEATURE_FLAGS_PATH.format(user_id),
+            method=REQUEST_METHOD_GET,
+            params=list_params,
+        )
+
+        return FeatureFlagsListResource(
+            list_method=self.list_feature_flags,
+            list_args=list_params,
+            **ListPage[FeatureFlag](**response).model_dump(),
+        )
+
 
 class AsyncUserManagement(UserManagementModule):
     _http_client: AsyncHTTPClient
@@ -2312,3 +2370,31 @@ class AsyncUserManagement(UserManagementModule):
         )
 
         return Invitation.model_validate(response)
+
+    async def list_feature_flags(
+        self,
+        user_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(
+            USER_FEATURE_FLAGS_PATH.format(user_id),
+            method=REQUEST_METHOD_GET,
+            params=list_params,
+        )
+
+        return FeatureFlagsListResource(
+            list_method=self.list_feature_flags,
+            list_args=list_params,
+            **ListPage[FeatureFlag](**response).model_dump(),
+        )

{workos-5.35.0 → workos-5.37.0}/workos.egg-info/PKG-INFO

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

{workos-5.35.0 → workos-5.37.0}/workos.egg-info/SOURCES.txt

@@ -86,7 +86,7 @@ workos/types/events/event_type.py
 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/events/session_payload.py
 workos/types/feature_flags/__init__.py
 workos/types/feature_flags/feature_flag.py
 workos/types/feature_flags/list_filters.py