maleo-identity 0.0.55__py3-none-any.whl → 0.0.58__py3-none-any.whl

maleo_identity/client/controllers/http/organization.py

@@ -1,7 +1,10 @@
+from typing import Dict, Optional
+from maleo_foundation.authorization import Authorization
 from maleo_foundation.managers.client.base import BearerAuth
 from maleo_foundation.managers.client.maleo import MaleoClientHTTPController
 from maleo_foundation.models.transfers.results.client.controllers.http \
     import BaseClientHTTPControllerResults
+from maleo_foundation.utils.merger import deep_merge
 from maleo_identity.models.transfers.parameters.general.organization_role \
     import MaleoIdentityOrganizationRoleGeneralParametersTransfers
 from maleo_identity.models.transfers.parameters.general.organization \
@@ -22,12 +25,14 @@ from maleo_identity.models.transfers.parameters.client.user_organization_role \
 class MaleoIdentityOrganizationHTTPController(MaleoClientHTTPController):
     async def get_organizations(
         self,
-        parameters:MaleoIdentityOrganizationClientParametersTransfers.GetMultiple
+        parameters:MaleoIdentityOrganizationClientParametersTransfers.GetMultiple,
+        authorization:Optional[Authorization] = None,
+        headers:Optional[Dict[str, str]] = None
     ) -> BaseClientHTTPControllerResults:
         """Fetch organizations from MaleoIdentity"""
         async with self._manager.get_client() as client:
             #* Define URL
-            url = f"{self._manager.url.api}/v1/organizations/"
+            url = f"{self._manager.url}/v1/organizations/"
 
             #* Parse parameters to query params
             params = (
@@ -42,22 +47,40 @@ class MaleoIdentityOrganizationHTTPController(MaleoClientHTTPController):
                 )
             )
 
+            #* Create headers
+            base_headers = {
+                "Content-Type": "application/json"
+            }
+            if headers is not None:
+                headers = deep_merge(
+                    base_headers,
+                    headers
+                )
+            else:
+                headers = base_headers
+
             #* Create auth
-            token = self._service_manager.token
-            auth = BearerAuth(token=token) if token is not None else None
+            token = None
+            if authorization and authorization.scheme == "Bearer":
+                token = authorization.credentials
+            elif self._service_manager.token:
+                token = self._service_manager.token
+            auth = BearerAuth(token) if token else None
 
             #* Send request and wait for response
-            response = await client.get(url=url, params=params, auth=auth)
+            response = await client.get(url=url, params=params, headers=headers, auth=auth)
             return BaseClientHTTPControllerResults(response=response)
 
     async def get_structured_organizations(
         self,
-        parameters:MaleoIdentityOrganizationClientParametersTransfers.GetMultipleStructured
+        parameters:MaleoIdentityOrganizationClientParametersTransfers.GetMultipleStructured,
+        authorization:Optional[Authorization] = None,
+        headers:Optional[Dict[str, str]] = None
     ) -> BaseClientHTTPControllerResults:
         """Fetch structured organizations from MaleoIdentity"""
         async with self._manager.get_client() as client:
             #* Define URL
-            url = f"{self._manager.url.api}/v1/organizations/structured"
+            url = f"{self._manager.url}/v1/organizations/structured"
 
             #* Parse parameters to query params
             params = (
@@ -72,22 +95,40 @@ class MaleoIdentityOrganizationHTTPController(MaleoClientHTTPController):
                 )
             )
 
+            #* Create headers
+            base_headers = {
+                "Content-Type": "application/json"
+            }
+            if headers is not None:
+                headers = deep_merge(
+                    base_headers,
+                    headers
+                )
+            else:
+                headers = base_headers
+
             #* Create auth
-            token = self._service_manager.token
-            auth = BearerAuth(token=token) if token is not None else None
+            token = None
+            if authorization and authorization.scheme == "Bearer":
+                token = authorization.credentials
+            elif self._service_manager.token:
+                token = self._service_manager.token
+            auth = BearerAuth(token) if token else None
 
             #* Send request and wait for response
-            response = await client.get(url=url, params=params, auth=auth)
+            response = await client.get(url=url, params=params, headers=headers, auth=auth)
             return BaseClientHTTPControllerResults(response=response)
 
     async def get_organization(
         self,
-        parameters:MaleoIdentityOrganizationGeneralParametersTransfers.GetSingle
+        parameters:MaleoIdentityOrganizationGeneralParametersTransfers.GetSingle,
+        authorization:Optional[Authorization] = None,
+        headers:Optional[Dict[str, str]] = None
     ) -> BaseClientHTTPControllerResults:
         """Fetch organization from MaleoIdentity"""
         async with self._manager.get_client() as client:
             #* Define URL
-            url = f"{self._manager.url.api}/v1/organizations/{parameters.identifier}/{parameters.value}"
+            url = f"{self._manager.url}/v1/organizations/{parameters.identifier}/{parameters.value}"
 
             #* Parse parameters to query params
             params = (
@@ -99,22 +140,40 @@ class MaleoIdentityOrganizationHTTPController(MaleoClientHTTPController):
                 .model_dump(exclude_none=True)
             )
 
+            #* Create headers
+            base_headers = {
+                "Content-Type": "application/json"
+            }
+            if headers is not None:
+                headers = deep_merge(
+                    base_headers,
+                    headers
+                )
+            else:
+                headers = base_headers
+
             #* Create auth
-            token = self._service_manager.token
-            auth = BearerAuth(token=token) if token is not None else None
+            token = None
+            if authorization and authorization.scheme == "Bearer":
+                token = authorization.credentials
+            elif self._service_manager.token:
+                token = self._service_manager.token
+            auth = BearerAuth(token) if token else None
 
             #* Send request and wait for response
-            response = await client.get(url=url, params=params, auth=auth)
+            response = await client.get(url=url, params=params, headers=headers, auth=auth)
             return BaseClientHTTPControllerResults(response=response)
 
     async def get_structured_organization(
         self,
-        parameters:MaleoIdentityOrganizationGeneralParametersTransfers.GetSingle
+        parameters:MaleoIdentityOrganizationGeneralParametersTransfers.GetSingle,
+        authorization:Optional[Authorization] = None,
+        headers:Optional[Dict[str, str]] = None
     ) -> BaseClientHTTPControllerResults:
         """Fetch structured organization from MaleoIdentity"""
         async with self._manager.get_client() as client:
             #* Define URL
-            url = f"{self._manager.url.api}/v1/organizations/{parameters.identifier}/{parameters.value}/structured"
+            url = f"{self._manager.url}/v1/organizations/{parameters.identifier}/{parameters.value}/structured"
 
             #* Parse parameters to query params
             params = (
@@ -126,22 +185,40 @@ class MaleoIdentityOrganizationHTTPController(MaleoClientHTTPController):
                 .model_dump(exclude_none=True)
             )
 
+            #* Create headers
+            base_headers = {
+                "Content-Type": "application/json"
+            }
+            if headers is not None:
+                headers = deep_merge(
+                    base_headers,
+                    headers
+                )
+            else:
+                headers = base_headers
+
             #* Create auth
-            token = self._service_manager.token
-            auth = BearerAuth(token=token) if token is not None else None
+            token = None
+            if authorization and authorization.scheme == "Bearer":
+                token = authorization.credentials
+            elif self._service_manager.token:
+                token = self._service_manager.token
+            auth = BearerAuth(token) if token else None
 
             #* Send request and wait for response
-            response = await client.get(url=url, params=params, auth=auth)
+            response = await client.get(url=url, params=params, headers=headers, auth=auth)
             return BaseClientHTTPControllerResults(response=response)
 
     async def get_organization_children(
         self,
-        parameters:MaleoIdentityOrganizationClientParametersTransfers.GetMultipleChildren
+        parameters:MaleoIdentityOrganizationClientParametersTransfers.GetMultipleChildren,
+        authorization:Optional[Authorization] = None,
+        headers:Optional[Dict[str, str]] = None
     ) -> BaseClientHTTPControllerResults:
         """Fetch organization's children from MaleoIdentity"""
         async with self._manager.get_client() as client:
             #* Define URL
-            url = f"{self._manager.url.api}/v1/organizations/{parameters.organization_id}/children"
+            url = f"{self._manager.url}/v1/organizations/{parameters.organization_id}/children"
 
             #* Parse parameters to query params
             params = (
@@ -156,22 +233,40 @@ class MaleoIdentityOrganizationHTTPController(MaleoClientHTTPController):
                 )
             )
 
+            #* Create headers
+            base_headers = {
+                "Content-Type": "application/json"
+            }
+            if headers is not None:
+                headers = deep_merge(
+                    base_headers,
+                    headers
+                )
+            else:
+                headers = base_headers
+
             #* Create auth
-            token = self._service_manager.token
-            auth = BearerAuth(token=token) if token is not None else None
+            token = None
+            if authorization and authorization.scheme == "Bearer":
+                token = authorization.credentials
+            elif self._service_manager.token:
+                token = self._service_manager.token
+            auth = BearerAuth(token) if token else None
 
             #* Send request and wait for response
-            response = await client.get(url=url, params=params, auth=auth)
+            response = await client.get(url=url, params=params, headers=headers, auth=auth)
             return BaseClientHTTPControllerResults(response=response)
 
     async def create(
         self,
-        parameters:MaleoIdentityOrganizationGeneralParametersTransfers.Create
+        parameters:MaleoIdentityOrganizationGeneralParametersTransfers.Create,
+        authorization:Optional[Authorization] = None,
+        headers:Optional[Dict[str, str]] = None
     ) -> BaseClientHTTPControllerResults:
         """Create a new organization in MaleoIdentity"""
         async with self._manager.get_client() as client:
             #* Define URL
-            url = f"{self._manager.url.api}/v1/organizations/"
+            url = f"{self._manager.url}/v1/organizations/"
 
             #* Declare body
             json = (
@@ -193,9 +288,25 @@ class MaleoIdentityOrganizationHTTPController(MaleoClientHTTPController):
                 .model_dump(exclude_none=True)
             )
 
+            #* Create headers
+            base_headers = {
+                "Content-Type": "application/json"
+            }
+            if headers is not None:
+                headers = deep_merge(
+                    base_headers,
+                    headers
+                )
+            else:
+                headers = base_headers
+
             #* Create auth
-            token = self._service_manager.token
-            auth = BearerAuth(token=token) if token is not None else None
+            token = None
+            if authorization and authorization.scheme == "Bearer":
+                token = authorization.credentials
+            elif self._service_manager.token:
+                token = self._service_manager.token
+            auth = BearerAuth(token) if token else None
 
             #* Send request and wait for response
             response = await client.post(url=url, json=json, params=params, auth=auth)
@@ -203,12 +314,14 @@ class MaleoIdentityOrganizationHTTPController(MaleoClientHTTPController):
 
     async def update(
         self,
-        parameters:MaleoIdentityOrganizationGeneralParametersTransfers.Update
+        parameters:MaleoIdentityOrganizationGeneralParametersTransfers.Update,
+        authorization:Optional[Authorization] = None,
+        headers:Optional[Dict[str, str]] = None
     ) -> BaseClientHTTPControllerResults:
         """Update organization's data in MaleoIdentity"""
         async with self._manager.get_client() as client:
             #* Define URL
-            url = f"{self._manager.url.api}/v1/organizations/{parameters.identifier}/{parameters.value}"
+            url = f"{self._manager.url}/v1/organizations/{parameters.identifier}/{parameters.value}"
 
             #* Declare body
             json = (
@@ -230,9 +343,25 @@ class MaleoIdentityOrganizationHTTPController(MaleoClientHTTPController):
                 .model_dump(exclude_none=True)
             )
 
+            #* Create headers
+            base_headers = {
+                "Content-Type": "application/json"
+            }
+            if headers is not None:
+                headers = deep_merge(
+                    base_headers,
+                    headers
+                )
+            else:
+                headers = base_headers
+
             #* Create auth
-            token = self._service_manager.token
-            auth = BearerAuth(token=token) if token is not None else None
+            token = None
+            if authorization and authorization.scheme == "Bearer":
+                token = authorization.credentials
+            elif self._service_manager.token:
+                token = self._service_manager.token
+            auth = BearerAuth(token) if token else None
 
             #* Send request and wait for response
             response = await client.put(url=url, json=json, params=params, auth=auth)
@@ -240,12 +369,14 @@ class MaleoIdentityOrganizationHTTPController(MaleoClientHTTPController):
 
     async def get_organization_roles(
         self,
-        parameters:MaleoIdentityOrganizationRoleClientParametersTransfers.GetMultipleFromOrganization
+        parameters:MaleoIdentityOrganizationRoleClientParametersTransfers.GetMultipleFromOrganization,
+        authorization:Optional[Authorization] = None,
+        headers:Optional[Dict[str, str]] = None
     ) -> BaseClientHTTPControllerResults:
         """Get organization's roles from MaleoIdentity"""
         async with self._manager.get_client() as client:
             #* Define URL
-            url = f"{self._manager.url.api}/v1/organizations/{parameters.organization_id}/roles/"
+            url = f"{self._manager.url}/v1/organizations/{parameters.organization_id}/roles/"
 
             #* Parse parameters to query params
             params = (
@@ -260,22 +391,40 @@ class MaleoIdentityOrganizationHTTPController(MaleoClientHTTPController):
                 )
             )
 
+            #* Create headers
+            base_headers = {
+                "Content-Type": "application/json"
+            }
+            if headers is not None:
+                headers = deep_merge(
+                    base_headers,
+                    headers
+                )
+            else:
+                headers = base_headers
+
             #* Create auth
-            token = self._service_manager.token
-            auth = BearerAuth(token=token) if token is not None else None
+            token = None
+            if authorization and authorization.scheme == "Bearer":
+                token = authorization.credentials
+            elif self._service_manager.token:
+                token = self._service_manager.token
+            auth = BearerAuth(token) if token else None
 
             #* Send request and wait for response
-            response = await client.get(url=url, params=params, auth=auth)
+            response = await client.get(url=url, params=params, headers=headers, auth=auth)
             return BaseClientHTTPControllerResults(response=response)
 
     async def get_organization_role(
         self,
-        parameters:MaleoIdentityOrganizationRoleGeneralParametersTransfers.GetSingle
+        parameters:MaleoIdentityOrganizationRoleGeneralParametersTransfers.GetSingle,
+        authorization:Optional[Authorization] = None,
+        headers:Optional[Dict[str, str]] = None
     ) -> BaseClientHTTPControllerResults:
         """Get organization's role from MaleoIdentity"""
         async with self._manager.get_client() as client:
             #* Define URL
-            url = f"{self._manager.url.api}/v1/organizations/{parameters.organization_id}/roles/{parameters.key}"
+            url = f"{self._manager.url}/v1/organizations/{parameters.organization_id}/roles/{parameters.key}"
 
             #* Parse parameters to query params
             params = (
@@ -287,22 +436,40 @@ class MaleoIdentityOrganizationHTTPController(MaleoClientHTTPController):
                 .model_dump(exclude_none=True)
             )
 
+            #* Create headers
+            base_headers = {
+                "Content-Type": "application/json"
+            }
+            if headers is not None:
+                headers = deep_merge(
+                    base_headers,
+                    headers
+                )
+            else:
+                headers = base_headers
+
             #* Create auth
-            token = self._service_manager.token
-            auth = BearerAuth(token=token) if token is not None else None
+            token = None
+            if authorization and authorization.scheme == "Bearer":
+                token = authorization.credentials
+            elif self._service_manager.token:
+                token = self._service_manager.token
+            auth = BearerAuth(token) if token else None
 
             #* Send request and wait for response
-            response = await client.get(url=url, params=params, auth=auth)
+            response = await client.get(url=url, params=params, headers=headers, auth=auth)
             return BaseClientHTTPControllerResults(response=response)
 
     async def get_organization_users(
         self,
-        parameters:MaleoIdentityUserOrganizationClientParametersTransfers.GetMultipleFromOrganization
+        parameters:MaleoIdentityUserOrganizationClientParametersTransfers.GetMultipleFromOrganization,
+        authorization:Optional[Authorization] = None,
+        headers:Optional[Dict[str, str]] = None
     ) -> BaseClientHTTPControllerResults:
         """Get organization's users from MaleoIdentity"""
         async with self._manager.get_client() as client:
             #* Define URL
-            url = f"{self._manager.url.api}/v1/organizations/{parameters.organization_id}/users/"
+            url = f"{self._manager.url}/v1/organizations/{parameters.organization_id}/users/"
 
             #* Parse parameters to query params
             params = (
@@ -317,22 +484,40 @@ class MaleoIdentityOrganizationHTTPController(MaleoClientHTTPController):
                 )
             )
 
+            #* Create headers
+            base_headers = {
+                "Content-Type": "application/json"
+            }
+            if headers is not None:
+                headers = deep_merge(
+                    base_headers,
+                    headers
+                )
+            else:
+                headers = base_headers
+
             #* Create auth
-            token = self._service_manager.token
-            auth = BearerAuth(token=token) if token is not None else None
+            token = None
+            if authorization and authorization.scheme == "Bearer":
+                token = authorization.credentials
+            elif self._service_manager.token:
+                token = self._service_manager.token
+            auth = BearerAuth(token) if token else None
 
             #* Send request and wait for response
-            response = await client.get(url=url, params=params, auth=auth)
+            response = await client.get(url=url, params=params, headers=headers, auth=auth)
             return BaseClientHTTPControllerResults(response=response)
 
     async def get_organization_user(
         self,
-        parameters:MaleoIdentityUserOrganizationGeneralParametersTransfers.GetSingle
+        parameters:MaleoIdentityUserOrganizationGeneralParametersTransfers.GetSingle,
+        authorization:Optional[Authorization] = None,
+        headers:Optional[Dict[str, str]] = None
     ) -> BaseClientHTTPControllerResults:
         """Get organization's user from MaleoIdentity"""
         async with self._manager.get_client() as client:
             #* Define URL
-            url = f"{self._manager.url.api}/v1/organizations/{parameters.organization_id}/users/{parameters.user_id}"
+            url = f"{self._manager.url}/v1/organizations/{parameters.organization_id}/users/{parameters.user_id}"
 
             #* Parse parameters to query params
             params = (
@@ -344,22 +529,40 @@ class MaleoIdentityOrganizationHTTPController(MaleoClientHTTPController):
                 .model_dump(exclude_none=True)
             )
 
+            #* Create headers
+            base_headers = {
+                "Content-Type": "application/json"
+            }
+            if headers is not None:
+                headers = deep_merge(
+                    base_headers,
+                    headers
+                )
+            else:
+                headers = base_headers
+
             #* Create auth
-            token = self._service_manager.token
-            auth = BearerAuth(token=token) if token is not None else None
+            token = None
+            if authorization and authorization.scheme == "Bearer":
+                token = authorization.credentials
+            elif self._service_manager.token:
+                token = self._service_manager.token
+            auth = BearerAuth(token) if token else None
 
             #* Send request and wait for response
-            response = await client.get(url=url, params=params, auth=auth)
+            response = await client.get(url=url, params=params, headers=headers, auth=auth)
             return BaseClientHTTPControllerResults(response=response)
 
     async def get_organization_user_roles(
         self,
-        parameters:MaleoIdentityUserOrganizationRoleClientParametersTransfers.GetMultipleFromUserOrOrganization
+        parameters:MaleoIdentityUserOrganizationRoleClientParametersTransfers.GetMultipleFromUserOrOrganization,
+        authorization:Optional[Authorization] = None,
+        headers:Optional[Dict[str, str]] = None
     ) -> BaseClientHTTPControllerResults:
         """Get organization's user roles from MaleoIdentity"""
         async with self._manager.get_client() as client:
             #* Define URL
-            url = f"{self._manager.url.api}/v1/organizations/{parameters.organization_id}/users/{parameters.user_id}/roles/"
+            url = f"{self._manager.url}/v1/organizations/{parameters.organization_id}/users/{parameters.user_id}/roles/"
 
             #* Parse parameters to query params
             params = (
@@ -374,22 +577,40 @@ class MaleoIdentityOrganizationHTTPController(MaleoClientHTTPController):
                 )
             )
 
+            #* Create headers
+            base_headers = {
+                "Content-Type": "application/json"
+            }
+            if headers is not None:
+                headers = deep_merge(
+                    base_headers,
+                    headers
+                )
+            else:
+                headers = base_headers
+
             #* Create auth
-            token = self._service_manager.token
-            auth = BearerAuth(token=token) if token is not None else None
+            token = None
+            if authorization and authorization.scheme == "Bearer":
+                token = authorization.credentials
+            elif self._service_manager.token:
+                token = self._service_manager.token
+            auth = BearerAuth(token) if token else None
 
             #* Send request and wait for response
-            response = await client.get(url=url, params=params, auth=auth)
+            response = await client.get(url=url, params=params, headers=headers, auth=auth)
             return BaseClientHTTPControllerResults(response=response)
 
     async def get_organization_user_role(
         self,
-        parameters:MaleoIdentityUserOrganizationRoleGeneralParametersTransfers.GetSingle
+        parameters:MaleoIdentityUserOrganizationRoleGeneralParametersTransfers.GetSingle,
+        authorization:Optional[Authorization] = None,
+        headers:Optional[Dict[str, str]] = None
     ) -> BaseClientHTTPControllerResults:
         """Get organization's user role from MaleoIdentity"""
         async with self._manager.get_client() as client:
             #* Define URL
-            url = f"{self._manager.url.api}/v1/organizations/{parameters.organization_id}/users/{parameters.user_id}/roles/{parameters.key}"
+            url = f"{self._manager.url}/v1/organizations/{parameters.organization_id}/users/{parameters.user_id}/roles/{parameters.key}"
 
             #* Parse parameters to query params
             params = (
@@ -401,10 +622,26 @@ class MaleoIdentityOrganizationHTTPController(MaleoClientHTTPController):
                 .model_dump(exclude_none=True)
             )
 
+            #* Create headers
+            base_headers = {
+                "Content-Type": "application/json"
+            }
+            if headers is not None:
+                headers = deep_merge(
+                    base_headers,
+                    headers
+                )
+            else:
+                headers = base_headers
+
             #* Create auth
-            token = self._service_manager.token
-            auth = BearerAuth(token=token) if token is not None else None
+            token = None
+            if authorization and authorization.scheme == "Bearer":
+                token = authorization.credentials
+            elif self._service_manager.token:
+                token = self._service_manager.token
+            auth = BearerAuth(token) if token else None
 
             #* Send request and wait for response
-            response = await client.get(url=url, params=params, auth=auth)
+            response = await client.get(url=url, params=params, headers=headers, auth=auth)
             return BaseClientHTTPControllerResults(response=response)