maleo-identity 0.1.31__py3-none-any.whl → 0.1.59__py3-none-any.whl

maleo/identity/schemas/user_medical_role.py

@@ -1,19 +1,19 @@
 from pydantic import BaseModel, Field
 from typing import Annotated, Generic, Literal, TypeVar, overload
 from uuid import UUID
-from maleo.enums.medical import (
+from nexo.enums.medical import (
     MedicalRole,
     OptMedicalRole,
     FullMedicalRoleMixin,
     OptListOfMedicalRoles,
     FullMedicalRolesMixin,
 )
-from maleo.enums.status import (
+from nexo.enums.status import (
     ListOfDataStatuses,
     FULL_DATA_STATUSES,
 )
-from maleo.schemas.mixins.filter import convert as convert_filter
-from maleo.schemas.mixins.identity import (
+from nexo.schemas.mixins.filter import convert as convert_filter
+from nexo.schemas.mixins.identity import (
     IdentifierMixin,
     Ids,
     UUIDs,
@@ -22,23 +22,41 @@ from maleo.schemas.mixins.identity import (
     IntOrganizationId,
     IntOrganizationIds,
 )
-from maleo.schemas.mixins.sort import convert as convert_sort
-from maleo.schemas.operation.enums import ResourceOperationStatusUpdateType
-from maleo.schemas.parameter import (
+from nexo.schemas.mixins.sort import convert as convert_sort
+from nexo.schemas.operation.enums import ResourceOperationStatusUpdateType
+from nexo.schemas.parameter import (
     ReadSingleParameter as BaseReadSingleParameter,
     ReadPaginatedMultipleParameter,
     StatusUpdateParameter as BaseStatusUpdateParameter,
     DeleteSingleParameter as BaseDeleteSingleParameter,
 )
-from maleo.types.dict import StrToAnyDict
-from maleo.types.integer import OptInt, OptListOfInts
-from maleo.types.uuid import OptListOfUUIDs
-from ..enums.user_medical_role import IdentifierType
-from ..mixins.user_medical_role import UserMedicalRoleIdentifier
+from nexo.types.dict import StrToAnyDict
+from nexo.types.integer import OptInt, OptListOfInts
+from nexo.types.uuid import OptListOfUUIDs
+from ..enums.user_medical_role import IdentifierType, OptListOfExpandableFields
+from ..mixins.user_medical_role import Expand, UserMedicalRoleIdentifier
 from ..types.user_medical_role import CompositeIdentifierType, IdentifierValueType
 
 
-class CreateParameter(
+class BaseCreateData(FullMedicalRoleMixin[MedicalRole]):
+    pass
+
+
+class CreateFromOrganizationData(
+    FullMedicalRoleMixin[MedicalRole],
+    IntUserId[int],
+):
+    pass
+
+
+class CreateFromUserData(
+    FullMedicalRoleMixin[MedicalRole],
+    IntOrganizationId[int],
+):
+    pass
+
+
+class FullCreateData(
     FullMedicalRoleMixin[MedicalRole],
     IntOrganizationId[int],
     IntUserId[int],
@@ -46,7 +64,15 @@ class CreateParameter(
     pass
 
 
+class CreateParameter(
+    Expand,
+    FullCreateData,
+):
+    pass
+
+
 class ReadMultipleParameter(
+    Expand,
     ReadPaginatedMultipleParameter,
     FullMedicalRolesMixin[OptListOfMedicalRoles],
     IntOrganizationIds[OptListOfInts],
@@ -77,6 +103,7 @@ class ReadMultipleParameter(
             "page",
             "limit",
             "use_cache",
+            "expand",
         }
 
     def to_query_params(self) -> StrToAnyDict:
@@ -89,15 +116,18 @@ class ReadMultipleParameter(
         return params
 
 
-class ReadSingleParameter(BaseReadSingleParameter[UserMedicalRoleIdentifier]):
+class ReadSingleParameter(Expand, BaseReadSingleParameter[UserMedicalRoleIdentifier]):
     @classmethod
     def from_identifier(
         cls,
         identifier: UserMedicalRoleIdentifier,
         statuses: ListOfDataStatuses = FULL_DATA_STATUSES,
         use_cache: bool = True,
+        expand: OptListOfExpandableFields = None,
     ) -> "ReadSingleParameter":
-        return cls(identifier=identifier, statuses=statuses, use_cache=use_cache)
+        return cls(
+            identifier=identifier, statuses=statuses, use_cache=use_cache, expand=expand
+        )
 
     @overload
     @classmethod
@@ -107,6 +137,7 @@ class ReadSingleParameter(BaseReadSingleParameter[UserMedicalRoleIdentifier]):
         identifier_value: int,
         statuses: ListOfDataStatuses = list(FULL_DATA_STATUSES),
         use_cache: bool = True,
+        expand: OptListOfExpandableFields = None,
     ) -> "ReadSingleParameter": ...
     @overload
     @classmethod
@@ -116,6 +147,7 @@ class ReadSingleParameter(BaseReadSingleParameter[UserMedicalRoleIdentifier]):
         identifier_value: UUID,
         statuses: ListOfDataStatuses = list(FULL_DATA_STATUSES),
         use_cache: bool = True,
+        expand: OptListOfExpandableFields = None,
     ) -> "ReadSingleParameter": ...
     @overload
     @classmethod
@@ -125,6 +157,7 @@ class ReadSingleParameter(BaseReadSingleParameter[UserMedicalRoleIdentifier]):
         identifier_value: CompositeIdentifierType,
         statuses: ListOfDataStatuses = list(FULL_DATA_STATUSES),
         use_cache: bool = True,
+        expand: OptListOfExpandableFields = None,
     ) -> "ReadSingleParameter": ...
     @overload
     @classmethod
@@ -134,6 +167,7 @@ class ReadSingleParameter(BaseReadSingleParameter[UserMedicalRoleIdentifier]):
         identifier_value: IdentifierValueType,
         statuses: ListOfDataStatuses = list(FULL_DATA_STATUSES),
         use_cache: bool = True,
+        expand: OptListOfExpandableFields = None,
     ) -> "ReadSingleParameter": ...
     @classmethod
     def new(
@@ -142,6 +176,7 @@ class ReadSingleParameter(BaseReadSingleParameter[UserMedicalRoleIdentifier]):
         identifier_value: IdentifierValueType,
         statuses: ListOfDataStatuses = list(FULL_DATA_STATUSES),
         use_cache: bool = True,
+        expand: OptListOfExpandableFields = None,
     ) -> "ReadSingleParameter":
         return cls(
             identifier=UserMedicalRoleIdentifier(
@@ -149,11 +184,12 @@ class ReadSingleParameter(BaseReadSingleParameter[UserMedicalRoleIdentifier]):
             ),
             statuses=statuses,
             use_cache=use_cache,
+            expand=expand,
         )
 
     def to_query_params(self) -> StrToAnyDict:
         return self.model_dump(
-            mode="json", include={"statuses", "use_cache"}, exclude_none=True
+            mode="json", include={"statuses", "use_cache", "expand"}, exclude_none=True
         )
 
 
@@ -181,6 +217,7 @@ class UpdateDataMixin(BaseModel, Generic[UpdateDataT]):
 
 
 class UpdateParameter(
+    Expand,
     UpdateDataMixin[UpdateDataT],
     IdentifierMixin[UserMedicalRoleIdentifier],
     Generic[UpdateDataT],
@@ -192,6 +229,7 @@ class UpdateParameter(
         identifier_type: Literal[IdentifierType.ID],
         identifier_value: int,
         data: UpdateDataT,
+        expand: OptListOfExpandableFields = None,
     ) -> "UpdateParameter": ...
     @overload
     @classmethod
@@ -200,6 +238,7 @@ class UpdateParameter(
         identifier_type: Literal[IdentifierType.UUID],
         identifier_value: UUID,
         data: UpdateDataT,
+        expand: OptListOfExpandableFields = None,
     ) -> "UpdateParameter": ...
     @overload
     @classmethod
@@ -208,6 +247,7 @@ class UpdateParameter(
         identifier_type: Literal[IdentifierType.COMPOSITE],
         identifier_value: CompositeIdentifierType,
         data: UpdateDataT,
+        expand: OptListOfExpandableFields = None,
     ) -> "UpdateParameter": ...
     @overload
     @classmethod
@@ -216,6 +256,7 @@ class UpdateParameter(
         identifier_type: IdentifierType,
         identifier_value: IdentifierValueType,
         data: UpdateDataT,
+        expand: OptListOfExpandableFields = None,
     ) -> "UpdateParameter": ...
     @classmethod
     def new(
@@ -223,16 +264,19 @@ class UpdateParameter(
         identifier_type: IdentifierType,
         identifier_value: IdentifierValueType,
         data: UpdateDataT,
+        expand: OptListOfExpandableFields = None,
     ) -> "UpdateParameter":
         return cls(
             identifier=UserMedicalRoleIdentifier(
                 type=identifier_type, value=identifier_value
             ),
             data=data,
+            expand=expand,
         )
 
 
 class StatusUpdateParameter(
+    Expand,
     BaseStatusUpdateParameter[UserMedicalRoleIdentifier],
 ):
     @overload
@@ -242,6 +286,7 @@ class StatusUpdateParameter(
         identifier_type: Literal[IdentifierType.ID],
         identifier_value: int,
         type: ResourceOperationStatusUpdateType,
+        expand: OptListOfExpandableFields = None,
     ) -> "StatusUpdateParameter": ...
     @overload
     @classmethod
@@ -250,6 +295,7 @@ class StatusUpdateParameter(
         identifier_type: Literal[IdentifierType.UUID],
         identifier_value: UUID,
         type: ResourceOperationStatusUpdateType,
+        expand: OptListOfExpandableFields = None,
     ) -> "StatusUpdateParameter": ...
     @overload
     @classmethod
@@ -258,6 +304,7 @@ class StatusUpdateParameter(
         identifier_type: Literal[IdentifierType.COMPOSITE],
         identifier_value: CompositeIdentifierType,
         type: ResourceOperationStatusUpdateType,
+        expand: OptListOfExpandableFields = None,
     ) -> "StatusUpdateParameter": ...
     @overload
     @classmethod
@@ -266,6 +313,7 @@ class StatusUpdateParameter(
         identifier_type: IdentifierType,
         identifier_value: IdentifierValueType,
         type: ResourceOperationStatusUpdateType,
+        expand: OptListOfExpandableFields = None,
     ) -> "StatusUpdateParameter": ...
     @classmethod
     def new(
@@ -273,12 +321,14 @@ class StatusUpdateParameter(
         identifier_type: IdentifierType,
         identifier_value: IdentifierValueType,
         type: ResourceOperationStatusUpdateType,
+        expand: OptListOfExpandableFields = None,
     ) -> "StatusUpdateParameter":
         return cls(
             identifier=UserMedicalRoleIdentifier(
                 type=identifier_type, value=identifier_value
             ),
             type=type,
+            expand=expand,
         )
 
 

maleo/identity/schemas/user_organization.py

@@ -0,0 +1,334 @@
+from pydantic import BaseModel, Field
+from typing import Annotated, Generic, Literal, TypeVar, overload
+from uuid import UUID
+from nexo.enums.status import (
+    ListOfDataStatuses,
+    FULL_DATA_STATUSES,
+)
+from nexo.schemas.mixins.filter import convert as convert_filter
+from nexo.schemas.mixins.identity import (
+    IdentifierMixin,
+    Ids,
+    UUIDs,
+    IntUserId,
+    IntUserIds,
+    IntOrganizationId,
+    IntOrganizationIds,
+)
+from nexo.schemas.mixins.sort import convert as convert_sort
+from nexo.schemas.operation.enums import ResourceOperationStatusUpdateType
+from nexo.schemas.parameter import (
+    ReadSingleParameter as BaseReadSingleParameter,
+    ReadPaginatedMultipleParameter,
+    StatusUpdateParameter as BaseStatusUpdateParameter,
+    DeleteSingleParameter as BaseDeleteSingleParameter,
+)
+from nexo.types.dict import StrToAnyDict
+from nexo.types.integer import OptInt, OptListOfInts
+from nexo.types.uuid import OptListOfUUIDs
+from ..enums.user_organization import IdentifierType, OptListOfExpandableFields
+from ..mixins.user_organization import Expand, UserOrganizationIdentifier
+from ..types.user_organization import CompositeIdentifierType, IdentifierValueType
+
+
+class CreateData(
+    IntOrganizationId[int],
+    IntUserId[int],
+):
+    pass
+
+
+class CreateParameter(
+    Expand,
+    CreateData,
+):
+    pass
+
+
+class ReadMultipleParameter(
+    Expand,
+    ReadPaginatedMultipleParameter,
+    IntOrganizationIds[OptListOfInts],
+    IntUserIds[OptListOfInts],
+    UUIDs[OptListOfUUIDs],
+    Ids[OptListOfInts],
+):
+    ids: Annotated[OptListOfInts, Field(None, description="Ids")] = None
+    uuids: Annotated[OptListOfUUIDs, Field(None, description="UUIDs")] = None
+    user_ids: Annotated[OptListOfInts, Field(None, description="User's IDs")] = None
+    organization_ids: Annotated[
+        OptListOfInts, Field(None, description="Organization's IDs")
+    ] = None
+
+    @property
+    def _query_param_fields(self) -> set[str]:
+        return {
+            "ids",
+            "uuids",
+            "statuses",
+            "user_ids",
+            "organization_ids",
+            "search",
+            "page",
+            "limit",
+            "use_cache",
+            "expand",
+        }
+
+    def to_query_params(self) -> StrToAnyDict:
+        params = self.model_dump(
+            mode="json", include=self._query_param_fields, exclude_none=True
+        )
+        params["filters"] = convert_filter(self.date_filters)
+        params["sorts"] = convert_sort(self.sort_columns)
+        params = {k: v for k, v in params.items()}
+        return params
+
+
+class ReadSingleParameter(Expand, BaseReadSingleParameter[UserOrganizationIdentifier]):
+    @classmethod
+    def from_identifier(
+        cls,
+        identifier: UserOrganizationIdentifier,
+        statuses: ListOfDataStatuses = FULL_DATA_STATUSES,
+        use_cache: bool = True,
+        expand: OptListOfExpandableFields = None,
+    ) -> "ReadSingleParameter":
+        return cls(
+            identifier=identifier, statuses=statuses, use_cache=use_cache, expand=expand
+        )
+
+    @overload
+    @classmethod
+    def new(
+        cls,
+        identifier_type: Literal[IdentifierType.ID],
+        identifier_value: int,
+        statuses: ListOfDataStatuses = list(FULL_DATA_STATUSES),
+        use_cache: bool = True,
+        expand: OptListOfExpandableFields = None,
+    ) -> "ReadSingleParameter": ...
+    @overload
+    @classmethod
+    def new(
+        cls,
+        identifier_type: Literal[IdentifierType.UUID],
+        identifier_value: UUID,
+        statuses: ListOfDataStatuses = list(FULL_DATA_STATUSES),
+        use_cache: bool = True,
+        expand: OptListOfExpandableFields = None,
+    ) -> "ReadSingleParameter": ...
+    @overload
+    @classmethod
+    def new(
+        cls,
+        identifier_type: Literal[IdentifierType.COMPOSITE],
+        identifier_value: CompositeIdentifierType,
+        statuses: ListOfDataStatuses = list(FULL_DATA_STATUSES),
+        use_cache: bool = True,
+        expand: OptListOfExpandableFields = None,
+    ) -> "ReadSingleParameter": ...
+    @overload
+    @classmethod
+    def new(
+        cls,
+        identifier_type: IdentifierType,
+        identifier_value: IdentifierValueType,
+        statuses: ListOfDataStatuses = list(FULL_DATA_STATUSES),
+        use_cache: bool = True,
+        expand: OptListOfExpandableFields = None,
+    ) -> "ReadSingleParameter": ...
+    @classmethod
+    def new(
+        cls,
+        identifier_type: IdentifierType,
+        identifier_value: IdentifierValueType,
+        statuses: ListOfDataStatuses = list(FULL_DATA_STATUSES),
+        use_cache: bool = True,
+        expand: OptListOfExpandableFields = None,
+    ) -> "ReadSingleParameter":
+        return cls(
+            identifier=UserOrganizationIdentifier(
+                type=identifier_type,
+                value=identifier_value,
+            ),
+            statuses=statuses,
+            use_cache=use_cache,
+            expand=expand,
+        )
+
+    def to_query_params(self) -> StrToAnyDict:
+        return self.model_dump(
+            mode="json", include={"statuses", "use_cache", "expand"}, exclude_none=True
+        )
+
+
+class FullUpdateData(
+    IntOrganizationId[int],
+    IntUserId[int],
+):
+    pass
+
+
+class PartialUpdateData(
+    IntOrganizationId[OptInt],
+    IntUserId[OptInt],
+):
+    pass
+
+
+UpdateDataT = TypeVar("UpdateDataT", FullUpdateData, PartialUpdateData)
+
+
+class UpdateDataMixin(BaseModel, Generic[UpdateDataT]):
+    data: UpdateDataT = Field(..., description="Update data")
+
+
+class UpdateParameter(
+    Expand,
+    UpdateDataMixin[UpdateDataT],
+    IdentifierMixin[UserOrganizationIdentifier],
+    Generic[UpdateDataT],
+):
+    @overload
+    @classmethod
+    def new(
+        cls,
+        identifier_type: Literal[IdentifierType.ID],
+        identifier_value: int,
+        data: UpdateDataT,
+        expand: OptListOfExpandableFields = None,
+    ) -> "UpdateParameter": ...
+    @overload
+    @classmethod
+    def new(
+        cls,
+        identifier_type: Literal[IdentifierType.UUID],
+        identifier_value: UUID,
+        data: UpdateDataT,
+        expand: OptListOfExpandableFields = None,
+    ) -> "UpdateParameter": ...
+    @overload
+    @classmethod
+    def new(
+        cls,
+        identifier_type: Literal[IdentifierType.COMPOSITE],
+        identifier_value: CompositeIdentifierType,
+        data: UpdateDataT,
+        expand: OptListOfExpandableFields = None,
+    ) -> "UpdateParameter": ...
+    @overload
+    @classmethod
+    def new(
+        cls,
+        identifier_type: IdentifierType,
+        identifier_value: IdentifierValueType,
+        data: UpdateDataT,
+        expand: OptListOfExpandableFields = None,
+    ) -> "UpdateParameter": ...
+    @classmethod
+    def new(
+        cls,
+        identifier_type: IdentifierType,
+        identifier_value: IdentifierValueType,
+        data: UpdateDataT,
+        expand: OptListOfExpandableFields = None,
+    ) -> "UpdateParameter":
+        return cls(
+            identifier=UserOrganizationIdentifier(
+                type=identifier_type, value=identifier_value
+            ),
+            data=data,
+            expand=expand,
+        )
+
+
+class StatusUpdateParameter(
+    Expand,
+    BaseStatusUpdateParameter[UserOrganizationIdentifier],
+):
+    @overload
+    @classmethod
+    def new(
+        cls,
+        identifier_type: Literal[IdentifierType.ID],
+        identifier_value: int,
+        type: ResourceOperationStatusUpdateType,
+        expand: OptListOfExpandableFields = None,
+    ) -> "StatusUpdateParameter": ...
+    @overload
+    @classmethod
+    def new(
+        cls,
+        identifier_type: Literal[IdentifierType.UUID],
+        identifier_value: UUID,
+        type: ResourceOperationStatusUpdateType,
+        expand: OptListOfExpandableFields = None,
+    ) -> "StatusUpdateParameter": ...
+    @overload
+    @classmethod
+    def new(
+        cls,
+        identifier_type: Literal[IdentifierType.COMPOSITE],
+        identifier_value: CompositeIdentifierType,
+        type: ResourceOperationStatusUpdateType,
+        expand: OptListOfExpandableFields = None,
+    ) -> "StatusUpdateParameter": ...
+    @overload
+    @classmethod
+    def new(
+        cls,
+        identifier_type: IdentifierType,
+        identifier_value: IdentifierValueType,
+        type: ResourceOperationStatusUpdateType,
+        expand: OptListOfExpandableFields = None,
+    ) -> "StatusUpdateParameter": ...
+    @classmethod
+    def new(
+        cls,
+        identifier_type: IdentifierType,
+        identifier_value: IdentifierValueType,
+        type: ResourceOperationStatusUpdateType,
+        expand: OptListOfExpandableFields = None,
+    ) -> "StatusUpdateParameter":
+        return cls(
+            identifier=UserOrganizationIdentifier(
+                type=identifier_type, value=identifier_value
+            ),
+            type=type,
+            expand=expand,
+        )
+
+
+class DeleteSingleParameter(BaseDeleteSingleParameter[UserOrganizationIdentifier]):
+    @overload
+    @classmethod
+    def new(
+        cls, identifier_type: Literal[IdentifierType.ID], identifier_value: int
+    ) -> "DeleteSingleParameter": ...
+    @overload
+    @classmethod
+    def new(
+        cls, identifier_type: Literal[IdentifierType.UUID], identifier_value: UUID
+    ) -> "DeleteSingleParameter": ...
+    @overload
+    @classmethod
+    def new(
+        cls,
+        identifier_type: Literal[IdentifierType.COMPOSITE],
+        identifier_value: CompositeIdentifierType,
+    ) -> "DeleteSingleParameter": ...
+    @overload
+    @classmethod
+    def new(
+        cls, identifier_type: IdentifierType, identifier_value: IdentifierValueType
+    ) -> "DeleteSingleParameter": ...
+    @classmethod
+    def new(
+        cls, identifier_type: IdentifierType, identifier_value: IdentifierValueType
+    ) -> "DeleteSingleParameter":
+        return cls(
+            identifier=UserOrganizationIdentifier(
+                type=identifier_type, value=identifier_value
+            )
+        )