enum

api/application/contracts/user/roles.py

@@ -0,0 +1,15 @@
+from enum import Enum
+
+
+class UserRole(Enum):
+    # base roles
+    ADMIN = "Administrator"
+
+    # service provider roles
+    SUPPLIER_EMPLOYER = "Director"
+    SUPPLIER = "Supplie of service"
+    SUPPLIER_EMPLOYEE = "Supplie employee"
+
+    CLIENT = "Client"
+    EMPLOYER = "Client company employer"
+    EMPLOYEE = "Client company employee"

api/application/usecase/company/create_company.py

@@ -1,7 +1,6 @@
 from api.application.abstractions.transaction import TransactionContextManager
 from api.application.contracts.company.company_request import CreateNewCompany
-from api.application.contracts.company.company_response import \
-    CompanyBaseResponse
+from api.application.contracts.company.company_response import CompanyBaseResponse
 from api.domain.company.repository import CompanyRepository
 
 

api/application/usecase/company/get_users_company.py

@@ -1,8 +1,6 @@
 from api.application.abstractions.transaction import TransactionContextManager
-from api.application.contracts.company.company_request import \
-    CompanyByOwnerEmail
-from api.application.contracts.company.company_response import \
-    CompanyBaseResponse
+from api.application.contracts.company.company_request import CompanyByOwnerEmail
+from api.application.contracts.company.company_response import CompanyBaseResponse
 from api.domain.company.repository import CompanyRepository
 
 
@@ -17,10 +15,5 @@ class GetCompaniesByOwnerEmail:
 
     async def execute(self, request: CompanyByOwnerEmail) -> list[CompanyBaseResponse]:
         async with self.transaction:
-            companies = await self.company_repository.get_companies_by_owner_email(
-                filter={"email": request.email}
-            )
-            return [
-                CompanyBaseResponse(name=comp.name.value, email=comp.email.value)
-                for comp in companies
-            ]
+            companies = await self.company_repository.get_companies_by_owner_email(filter={"email": request.email})
+            return [CompanyBaseResponse(name=comp.name.value, email=comp.email.value) for comp in companies]

api/application/usecase/user/get_user_by_email.py

@@ -4,9 +4,7 @@ from api.domain.user.repository import UserRepository
 
 
 class GetUserByEmail:
-    def __init__(
-        self, transaction: TransactionContextManager, user_repository: UserRepository
-    ) -> None:
+    def __init__(self, transaction: TransactionContextManager, user_repository: UserRepository) -> None:
         self.transaction = transaction
         self.user_repository = user_repository