events imports are checked

api_events/events/identity/users.py

@@ -3,15 +3,20 @@ import typing
 from fastapi import status
 from fastapi.responses import JSONResponse
 
-
 from databases import MongoQueryIdentity, Users, Companies, People
 from databases.no_sql_models.validations import DomainViaUser
 
 from api_events.events.abstract_class import MethodToEvent, ActionsSchema
+
 from api_objects.auth.token_objects import EmployeeTokenObject, OccupantTokenObject
 from api_validations.core_response import return_json_response_from_alchemy
 
-from validations import InsertUsers, UpdateUsers, PatchRecord, ListOptions
+from api_validations.validations_request import (
+    InsertUsers,
+    UpdateUsers,
+    PatchRecord,
+    ListOptions, RegisterServices2Occupant,
+)
 
 
 class UserListEventMethods(MethodToEvent):
@@ -55,7 +60,9 @@ class UserCreateEventMethods(MethodToEvent):
     }
 
     @classmethod
-    def user_create(cls, data: InsertUsers, token_dict):
+    def user_create(
+            cls, data: InsertUsers, token_dict: typing.Union[EmployeeTokenObject, OccupantTokenObject]
+    ):
         created_user = Users.create_action(create_user=data)
         created_user.related_company = token_dict.selected_company.company_uu_id
         domain_via_user = DomainViaUser(
@@ -66,7 +73,6 @@ class UserCreateEventMethods(MethodToEvent):
             company_uuid=created_user.related_company,
         )
         mongo_query_identity.create_domain_via_user(payload=domain_via_user)
-
         return JSONResponse(
             content={
                 "completed": True,
@@ -92,7 +98,9 @@ class UserUpdateEventMethods(MethodToEvent):
     }
 
     @classmethod
-    def user_update(cls, data: UpdateUsers, user_uu_id: str, token_dict):
+    def user_update(
+            cls, data: UpdateUsers, user_uu_id: str, token_dict: typing.Union[EmployeeTokenObject, OccupantTokenObject]
+    ):
         find_one_user = Users.find_one_or_abort(uu_id=user_uu_id)
         access_authorized_company = Companies.select_action(
             duty_id=getattr(token_dict, "duty_id", 5),
@@ -101,6 +109,7 @@ class UserUpdateEventMethods(MethodToEvent):
         if access_authorized_company.count:
             data_dict = data.excluded_dump()
             updated_user = find_one_user.update(**data_dict)
+            Users.save()
             return JSONResponse(
                 content={
                     "completed": True,