maleo-identity-client 0.6.2__py3-none-any.whl

maleo/identity/client/config.py

@@ -0,0 +1,13 @@
+from pydantic import Field
+from typing import Annotated
+from nexo.client.config import ClientConfig as BaseClientConfig
+from maleo.enums.service import ServiceKey, ServiceName
+
+
+class ClientConfig(BaseClientConfig):
+    key: Annotated[
+        str, Field(ServiceKey.IDENTITY.value, description="Client's key")
+    ] = ServiceKey.IDENTITY.value
+    name: Annotated[
+        str, Field(ServiceName.IDENTITY.value, description="Client's name")
+    ] = ServiceName.IDENTITY.value

maleo/identity/client/manager.py

@@ -0,0 +1,98 @@
+from nexo.client.manager import ClientManager
+from .config import ClientConfig
+from .services.organization_registration_code import (
+    OrganizationRegistrationCodeClientService,
+)
+from .services.organization_relation import OrganizationRelationClientService
+from .services.organization import OrganizationClientService
+from .services.patient import PatientClientService
+from .services.user_medical_role import UserMedicalRoleClientService
+from .services.user_organization_role import UserOrganizationRoleClientService
+from .services.user_organization import UserOrganizationClientService
+from .services.user_profile import UserProfileClientService
+from .services.user_system_role import UserSystemRoleClientService
+from .services.user import UserClientService
+
+
+class MaleoIdentityClientManager(ClientManager[ClientConfig]):
+    def initalize_services(self):
+        self.organization_registration_code = OrganizationRegistrationCodeClientService(
+            application_context=self._application_context,
+            config=self._config,
+            logger=self._logger,
+            http_client_manager=self._http_client_manager,
+            private_key=self._private_key,
+            redis=self._redis,
+        )
+        self.organization_relation = OrganizationRelationClientService(
+            application_context=self._application_context,
+            config=self._config,
+            logger=self._logger,
+            http_client_manager=self._http_client_manager,
+            private_key=self._private_key,
+            redis=self._redis,
+        )
+        self.organization = OrganizationClientService(
+            application_context=self._application_context,
+            config=self._config,
+            logger=self._logger,
+            http_client_manager=self._http_client_manager,
+            private_key=self._private_key,
+            redis=self._redis,
+        )
+        self.patient = PatientClientService(
+            application_context=self._application_context,
+            config=self._config,
+            logger=self._logger,
+            http_client_manager=self._http_client_manager,
+            private_key=self._private_key,
+            redis=self._redis,
+        )
+        self.user_medical_role = UserMedicalRoleClientService(
+            application_context=self._application_context,
+            config=self._config,
+            logger=self._logger,
+            http_client_manager=self._http_client_manager,
+            private_key=self._private_key,
+            redis=self._redis,
+        )
+        self.user_organization_role = UserOrganizationRoleClientService(
+            application_context=self._application_context,
+            config=self._config,
+            logger=self._logger,
+            http_client_manager=self._http_client_manager,
+            private_key=self._private_key,
+            redis=self._redis,
+        )
+        self.user_organization = UserOrganizationClientService(
+            application_context=self._application_context,
+            config=self._config,
+            logger=self._logger,
+            http_client_manager=self._http_client_manager,
+            private_key=self._private_key,
+            redis=self._redis,
+        )
+        self.user_profile = UserProfileClientService(
+            application_context=self._application_context,
+            config=self._config,
+            logger=self._logger,
+            http_client_manager=self._http_client_manager,
+            private_key=self._private_key,
+            redis=self._redis,
+        )
+        self.user_system_role = UserSystemRoleClientService(
+            application_context=self._application_context,
+            config=self._config,
+            logger=self._logger,
+            http_client_manager=self._http_client_manager,
+            private_key=self._private_key,
+            redis=self._redis,
+        )
+        self.user = UserClientService(
+            application_context=self._application_context,
+            config=self._config,
+            logger=self._logger,
+            http_client_manager=self._http_client_manager,
+            private_key=self._private_key,
+            redis=self._redis,
+        )

maleo/identity/client/services/organization.py

@@ -0,0 +1,272 @@
+import json
+from copy import deepcopy
+from datetime import datetime, timezone
+from typing import ClassVar, Literal, overload
+from uuid import UUID
+from nexo.client.service import ClientService
+from nexo.database.enums import Connection
+from nexo.database.utils import build_cache_key
+from nexo.enums.cardinality import Cardinality
+from nexo.enums.connection import Header
+from nexo.logging.enums import LogLevel
+from nexo.schemas.connection import ConnectionContext
+from nexo.schemas.exception.factory import MaleoExceptionFactory
+from nexo.schemas.operation.action.resource import ReadResourceOperationAction
+from nexo.schemas.operation.enums import OperationType, Target
+from nexo.schemas.operation.mixins import Timestamp
+from nexo.schemas.operation.resource import (
+    ReadMultipleResourceOperation,
+    ReadSingleResourceOperation,
+)
+from nexo.schemas.pagination import StrictPagination
+from nexo.schemas.resource import Resource, AggregateField
+from nexo.schemas.response import (
+    MultipleDataResponse,
+    ReadMultipleDataResponse,
+    SingleDataResponse,
+    ReadSingleDataResponse,
+)
+from nexo.schemas.security.authorization import (
+    AnyAuthorization,
+    AuthorizationFactory,
+)
+from nexo.schemas.security.impersonation import OptImpersonation
+from nexo.types.dict import OptStrToStrDict
+from nexo.utils.merger import merge_dicts
+from maleo.identity.constants.organization import ORGANIZATION_RESOURCE
+from maleo.identity.mixins.organization import is_id_identifier
+from maleo.identity.schemas.common import OrganizationSchema
+from maleo.identity.schemas.organization import (
+    ReadMultipleParameter,
+    ReadSingleParameter,
+)
+from ..config import ClientConfig
+
+
+class OrganizationClientService(ClientService[ClientConfig]):
+    _resource: ClassVar[Resource] = ORGANIZATION_RESOURCE
+
+    @overload
+    async def read(
+        self,
+        cardinality: Literal[Cardinality.MULTIPLE],
+        *,
+        operation_id: UUID,
+        connection_context: ConnectionContext,
+        authorization: AnyAuthorization,
+        impersonation: OptImpersonation = None,
+        parameters: ReadMultipleParameter,
+        headers: OptStrToStrDict = None,
+    ) -> ReadMultipleDataResponse[OrganizationSchema, StrictPagination, None]: ...
+    @overload
+    async def read(
+        self,
+        cardinality: Literal[Cardinality.SINGLE],
+        *,
+        operation_id: UUID,
+        connection_context: ConnectionContext,
+        authorization: AnyAuthorization,
+        impersonation: OptImpersonation = None,
+        parameters: ReadSingleParameter,
+        headers: OptStrToStrDict = None,
+    ) -> ReadSingleDataResponse[OrganizationSchema, None]: ...
+    async def read(
+        self,
+        cardinality: Cardinality,
+        *,
+        operation_id: UUID,
+        connection_context: ConnectionContext,
+        authorization: AnyAuthorization,
+        impersonation: OptImpersonation = None,
+        parameters: ReadMultipleParameter | ReadSingleParameter,
+        headers: OptStrToStrDict = None,
+    ) -> (
+        ReadMultipleDataResponse[OrganizationSchema, StrictPagination, None]
+        | ReadSingleDataResponse[OrganizationSchema, None]
+    ):
+        redis_client = self._redis.manager.client.get(Connection.ASYNC)
+
+        executed_at = datetime.now(tz=timezone.utc)
+
+        # Define arguments being used in this function
+        positional_arguments = [cardinality]
+        keyword_arguments = {
+            "authorization": (
+                authorization.model_dump(mode="json")
+                if authorization is not None
+                else None
+            ),
+            "parameters": parameters.model_dump(mode="json"),
+        }
+
+        # Define full function string
+        ext = f"({json.dumps(positional_arguments)}|{json.dumps(keyword_arguments)})"
+
+        # Define full cache_key
+        cache_key = build_cache_key(ext, namespace=self._namespace)
+
+        if parameters.use_cache:
+            # Initialize cache operation context
+            operation_context = deepcopy(self._operation_context)
+            operation_context.target.type = Target.CACHE
+
+            redis_response_str = await redis_client.get(cache_key)
+
+            if redis_response_str is not None:
+                operation_timestamp = Timestamp.completed_now(executed_at)
+                if cardinality is Cardinality.MULTIPLE:
+                    response = ReadMultipleDataResponse[
+                        OrganizationSchema, StrictPagination, None
+                    ].model_validate_json(redis_response_str)
+                    operation = ReadMultipleResourceOperation[
+                        OrganizationSchema, StrictPagination, None
+                    ](
+                        application_context=self._application_context,
+                        id=operation_id,
+                        context=operation_context,
+                        resource=self._resource,
+                        timestamp=operation_timestamp,
+                        summary=f"Successfully read multiple {self._resource.aggregate(AggregateField.NAME, sep=" ").lower()} from cache",
+                        connection_context=connection_context,
+                        authentication=None,
+                        authorization=authorization,
+                        impersonation=impersonation,
+                        response=response,
+                    )
+                    operation.log(self._logger, LogLevel.INFO)
+                    operation.publish(self._logger, self._publishers)
+                elif cardinality is Cardinality.SINGLE:
+                    response = ReadSingleDataResponse[
+                        OrganizationSchema, None
+                    ].model_validate_json(redis_response_str)
+                    operation = ReadSingleResourceOperation[OrganizationSchema, None](
+                        application_context=self._application_context,
+                        id=operation_id,
+                        context=operation_context,
+                        resource=self._resource,
+                        timestamp=operation_timestamp,
+                        summary=f"Successfully read single {self._resource.aggregate(AggregateField.NAME, sep=" ").lower()} from cache",
+                        connection_context=connection_context,
+                        authentication=None,
+                        authorization=authorization,
+                        impersonation=impersonation,
+                        response=response,
+                    )
+                    operation.log(self._logger, LogLevel.INFO)
+                    operation.publish(self._logger, self._publishers)
+
+                return response  # type: ignore
+
+        operation_context = deepcopy(self._operation_context)
+        operation_context.target.type = Target.MICROSERVICE
+
+        async with self._http_client_manager.get() as http_client:
+            base_headers = {
+                Header.CONTENT_TYPE.value: "application/json",
+                Header.X_OPERATION_ID.value: str(operation_id),
+            }
+            if impersonation is not None:
+                base_headers[Header.X_USER_ID.value] = str(impersonation.user_id)
+                if impersonation.organization_id is not None:
+                    base_headers[Header.X_ORGANIZATION_ID.value] = str(
+                        impersonation.organization_id
+                    )
+
+            if headers is not None:
+                headers = merge_dicts(base_headers, headers)
+            else:
+                headers = base_headers
+
+            auth = AuthorizationFactory.httpx_auth(
+                scheme=authorization.scheme, authorization=authorization.credentials
+            )
+
+            base_url = f"{self._config.url}/v1/{self._resource.identifiers[-1].slug}"
+            if isinstance(parameters, ReadMultipleParameter):
+                url = base_url
+            elif isinstance(parameters, ReadSingleParameter):
+                if is_id_identifier(parameters.identifier):
+                    url = base_url + f"/{parameters.identifier.value}"
+                else:
+                    url = (
+                        base_url
+                        + f"/{parameters.identifier.type}/{parameters.identifier.value}"
+                    )
+
+            params = parameters.to_query_params()
+
+            response = await http_client.get(
+                url, params=params, headers=headers, auth=auth
+            )
+
+            operation_timestamp = Timestamp.completed_now(executed_at)
+
+            if response.is_error:
+                exc = MaleoExceptionFactory.from_httpx(
+                    response,
+                    operation_type=OperationType.REQUEST,
+                    application_context=self._application_context,
+                    operation_id=operation_id,
+                    operation_context=operation_context,
+                    operation_action=ReadResourceOperationAction(),
+                    operation_timestamp=operation_timestamp,
+                    connection_context=connection_context,
+                    authentication=None,
+                    authorization=authorization,
+                    impersonation=impersonation,
+                    logger=self._logger,
+                )
+                exc.log_and_publish_operation(self._logger, self._publishers)
+                raise exc
+
+            if isinstance(parameters, ReadMultipleParameter):
+                validated_response = MultipleDataResponse[
+                    OrganizationSchema, StrictPagination, None
+                ].model_validate(response.json())
+                service_response = ReadMultipleDataResponse[
+                    OrganizationSchema, StrictPagination, None
+                ].new(
+                    data=validated_response.data,
+                    pagination=validated_response.pagination,
+                )
+                operation = ReadMultipleResourceOperation[
+                    OrganizationSchema, StrictPagination, None
+                ](
+                    application_context=self._application_context,
+                    id=operation_id,
+                    context=operation_context,
+                    resource=self._resource,
+                    timestamp=operation_timestamp,
+                    summary=f"Successfully read multiple {self._resource.aggregate(AggregateField.NAME, sep=" ").lower()} from microservice",
+                    connection_context=connection_context,
+                    authentication=None,
+                    authorization=authorization,
+                    impersonation=impersonation,
+                    response=service_response,
+                )
+                operation.log(self._logger, LogLevel.INFO)
+                operation.publish(self._logger, self._publishers)
+            elif isinstance(parameters, ReadSingleParameter):
+                validated_response = SingleDataResponse[
+                    OrganizationSchema, None
+                ].model_validate(response.json())
+                service_response = ReadSingleDataResponse[OrganizationSchema, None].new(
+                    data=validated_response.data,
+                )
+                operation = ReadSingleResourceOperation[OrganizationSchema, None](
+                    application_context=self._application_context,
+                    id=operation_id,
+                    context=operation_context,
+                    resource=self._resource,
+                    timestamp=operation_timestamp,
+                    summary=f"Successfully read single {self._resource.aggregate(AggregateField.NAME, sep=" ").lower()} from microservice",
+                    connection_context=connection_context,
+                    authentication=None,
+                    authorization=authorization,
+                    impersonation=impersonation,
+                    response=service_response,
+                )
+                operation.log(self._logger, LogLevel.INFO)
+                operation.publish(self._logger, self._publishers)
+
+            return service_response  # type: ignore

maleo/identity/client/services/organization_registration_code.py

@@ -0,0 +1,284 @@
+import json
+from copy import deepcopy
+from datetime import datetime, timezone
+from typing import ClassVar, Literal, overload
+from uuid import UUID
+from nexo.client.service import ClientService
+from nexo.database.enums import Connection
+from nexo.database.utils import build_cache_key
+from nexo.enums.cardinality import Cardinality
+from nexo.enums.connection import Header
+from nexo.logging.enums import LogLevel
+from nexo.schemas.connection import ConnectionContext
+from nexo.schemas.exception.factory import MaleoExceptionFactory
+from nexo.schemas.operation.action.resource import ReadResourceOperationAction
+from nexo.schemas.operation.enums import OperationType, Target
+from nexo.schemas.operation.mixins import Timestamp
+from nexo.schemas.operation.resource import (
+    ReadMultipleResourceOperation,
+    ReadSingleResourceOperation,
+)
+from nexo.schemas.pagination import StrictPagination
+from nexo.schemas.resource import Resource, AggregateField
+from nexo.schemas.response import (
+    MultipleDataResponse,
+    ReadMultipleDataResponse,
+    SingleDataResponse,
+    ReadSingleDataResponse,
+)
+from nexo.schemas.security.authorization import (
+    AnyAuthorization,
+    AuthorizationFactory,
+)
+from nexo.schemas.security.impersonation import OptImpersonation
+from nexo.types.dict import OptStrToStrDict
+from nexo.utils.merger import merge_dicts
+from maleo.identity.constants.organization_registration_code import (
+    ORGANIZATION_REGISTRATION_CODE_RESOURCE,
+)
+from maleo.identity.mixins.organization_registration_code import is_id_identifier
+from maleo.identity.schemas.common import OrganizationRegistrationCodeSchema
+from maleo.identity.schemas.organization_registration_code import (
+    ReadMultipleParameter,
+    ReadSingleParameter,
+)
+from ..config import ClientConfig
+
+
+class OrganizationRegistrationCodeClientService(ClientService[ClientConfig]):
+    _resource: ClassVar[Resource] = ORGANIZATION_REGISTRATION_CODE_RESOURCE
+
+    @overload
+    async def read(
+        self,
+        cardinality: Literal[Cardinality.MULTIPLE],
+        *,
+        operation_id: UUID,
+        connection_context: ConnectionContext,
+        authorization: AnyAuthorization,
+        impersonation: OptImpersonation = None,
+        parameters: ReadMultipleParameter,
+        headers: OptStrToStrDict = None,
+    ) -> ReadMultipleDataResponse[
+        OrganizationRegistrationCodeSchema, StrictPagination, None
+    ]: ...
+    @overload
+    async def read(
+        self,
+        cardinality: Literal[Cardinality.SINGLE],
+        *,
+        operation_id: UUID,
+        connection_context: ConnectionContext,
+        authorization: AnyAuthorization,
+        impersonation: OptImpersonation = None,
+        parameters: ReadSingleParameter,
+        headers: OptStrToStrDict = None,
+    ) -> ReadSingleDataResponse[OrganizationRegistrationCodeSchema, None]: ...
+    async def read(
+        self,
+        cardinality: Cardinality,
+        *,
+        operation_id: UUID,
+        connection_context: ConnectionContext,
+        authorization: AnyAuthorization,
+        impersonation: OptImpersonation = None,
+        parameters: ReadMultipleParameter | ReadSingleParameter,
+        headers: OptStrToStrDict = None,
+    ) -> (
+        ReadMultipleDataResponse[
+            OrganizationRegistrationCodeSchema, StrictPagination, None
+        ]
+        | ReadSingleDataResponse[OrganizationRegistrationCodeSchema, None]
+    ):
+        redis_client = self._redis.manager.client.get(Connection.ASYNC)
+
+        executed_at = datetime.now(tz=timezone.utc)
+
+        # Define arguments being used in this function
+        positional_arguments = [cardinality]
+        keyword_arguments = {
+            "authorization": (
+                authorization.model_dump(mode="json")
+                if authorization is not None
+                else None
+            ),
+            "parameters": parameters.model_dump(mode="json"),
+        }
+
+        # Define full function string
+        ext = f"({json.dumps(positional_arguments)}|{json.dumps(keyword_arguments)})"
+
+        # Define full cache_key
+        cache_key = build_cache_key(ext, namespace=self._namespace)
+
+        if parameters.use_cache:
+            # Initialize cache operation context
+            operation_context = deepcopy(self._operation_context)
+            operation_context.target.type = Target.CACHE
+
+            redis_response_str = await redis_client.get(cache_key)
+
+            if redis_response_str is not None:
+                operation_timestamp = Timestamp.completed_now(executed_at)
+                if cardinality is Cardinality.MULTIPLE:
+                    response = ReadMultipleDataResponse[
+                        OrganizationRegistrationCodeSchema, StrictPagination, None
+                    ].model_validate_json(redis_response_str)
+                    operation = ReadMultipleResourceOperation[
+                        OrganizationRegistrationCodeSchema, StrictPagination, None
+                    ](
+                        application_context=self._application_context,
+                        id=operation_id,
+                        context=operation_context,
+                        resource=self._resource,
+                        timestamp=operation_timestamp,
+                        summary=f"Successfully read multiple {self._resource.aggregate(AggregateField.NAME, sep=" ").lower()} from cache",
+                        connection_context=connection_context,
+                        authentication=None,
+                        authorization=authorization,
+                        impersonation=impersonation,
+                        response=response,
+                    )
+                    operation.log(self._logger, LogLevel.INFO)
+                    operation.publish(self._logger, self._publishers)
+                elif cardinality is Cardinality.SINGLE:
+                    response = ReadSingleDataResponse[
+                        OrganizationRegistrationCodeSchema, None
+                    ].model_validate_json(redis_response_str)
+                    operation = ReadSingleResourceOperation[
+                        OrganizationRegistrationCodeSchema, None
+                    ](
+                        application_context=self._application_context,
+                        id=operation_id,
+                        context=operation_context,
+                        resource=self._resource,
+                        timestamp=operation_timestamp,
+                        summary=f"Successfully read single {self._resource.aggregate(AggregateField.NAME, sep=" ").lower()} from cache",
+                        connection_context=connection_context,
+                        authentication=None,
+                        authorization=authorization,
+                        impersonation=impersonation,
+                        response=response,
+                    )
+                    operation.log(self._logger, LogLevel.INFO)
+                    operation.publish(self._logger, self._publishers)
+
+                return response  # type: ignore
+
+        operation_context = deepcopy(self._operation_context)
+        operation_context.target.type = Target.MICROSERVICE
+
+        async with self._http_client_manager.get() as http_client:
+            base_headers = {
+                Header.CONTENT_TYPE.value: "application/json",
+                Header.X_OPERATION_ID.value: str(operation_id),
+            }
+            if impersonation is not None:
+                base_headers[Header.X_USER_ID.value] = str(impersonation.user_id)
+                if impersonation.organization_id is not None:
+                    base_headers[Header.X_ORGANIZATION_ID.value] = str(
+                        impersonation.organization_id
+                    )
+
+            if headers is not None:
+                headers = merge_dicts(base_headers, headers)
+            else:
+                headers = base_headers
+
+            auth = AuthorizationFactory.httpx_auth(
+                scheme=authorization.scheme, authorization=authorization.credentials
+            )
+
+            base_url = f"{self._config.url}/v1/{self._resource.identifiers[-1].slug}"
+            if isinstance(parameters, ReadMultipleParameter):
+                url = base_url
+            elif isinstance(parameters, ReadSingleParameter):
+                if is_id_identifier(parameters.identifier):
+                    url = base_url + f"/{parameters.identifier.value}"
+                else:
+                    url = (
+                        base_url
+                        + f"/{parameters.identifier.type}/{parameters.identifier.value}"
+                    )
+
+            params = parameters.to_query_params()
+
+            response = await http_client.get(
+                url, params=params, headers=headers, auth=auth
+            )
+
+            operation_timestamp = Timestamp.completed_now(executed_at)
+
+            if response.is_error:
+                exc = MaleoExceptionFactory.from_httpx(
+                    response,
+                    operation_type=OperationType.REQUEST,
+                    application_context=self._application_context,
+                    operation_id=operation_id,
+                    operation_context=operation_context,
+                    operation_action=ReadResourceOperationAction(),
+                    operation_timestamp=operation_timestamp,
+                    connection_context=connection_context,
+                    authentication=None,
+                    authorization=authorization,
+                    impersonation=impersonation,
+                    logger=self._logger,
+                )
+                exc.log_and_publish_operation(self._logger, self._publishers)
+                raise exc
+
+            if isinstance(parameters, ReadMultipleParameter):
+                validated_response = MultipleDataResponse[
+                    OrganizationRegistrationCodeSchema, StrictPagination, None
+                ].model_validate(response.json())
+                service_response = ReadMultipleDataResponse[
+                    OrganizationRegistrationCodeSchema, StrictPagination, None
+                ].new(
+                    data=validated_response.data,
+                    pagination=validated_response.pagination,
+                )
+                operation = ReadMultipleResourceOperation[
+                    OrganizationRegistrationCodeSchema, StrictPagination, None
+                ](
+                    application_context=self._application_context,
+                    id=operation_id,
+                    context=operation_context,
+                    resource=self._resource,
+                    timestamp=operation_timestamp,
+                    summary=f"Successfully read multiple {self._resource.aggregate(AggregateField.NAME, sep=" ").lower()} from microservice",
+                    connection_context=connection_context,
+                    authentication=None,
+                    authorization=authorization,
+                    impersonation=impersonation,
+                    response=service_response,
+                )
+                operation.log(self._logger, LogLevel.INFO)
+                operation.publish(self._logger, self._publishers)
+            elif isinstance(parameters, ReadSingleParameter):
+                validated_response = SingleDataResponse[
+                    OrganizationRegistrationCodeSchema, None
+                ].model_validate(response.json())
+                service_response = ReadSingleDataResponse[
+                    OrganizationRegistrationCodeSchema, None
+                ].new(
+                    data=validated_response.data,
+                )
+                operation = ReadSingleResourceOperation[
+                    OrganizationRegistrationCodeSchema, None
+                ](
+                    application_context=self._application_context,
+                    id=operation_id,
+                    context=operation_context,
+                    resource=self._resource,
+                    timestamp=operation_timestamp,
+                    summary=f"Successfully read single {self._resource.aggregate(AggregateField.NAME, sep=" ").lower()} from microservice",
+                    connection_context=connection_context,
+                    authentication=None,
+                    authorization=authorization,
+                    impersonation=impersonation,
+                    response=service_response,
+                )
+                operation.log(self._logger, LogLevel.INFO)
+                operation.publish(self._logger, self._publishers)
+
+            return service_response  # type: ignore