events imports are checked

api_events/events/company/company_employee.py

@@ -1,9 +1,10 @@
 from datetime import datetime
+from typing import Union
 
 from fastapi import status, HTTPException
 from fastapi.responses import JSONResponse
 
-from validations import (
+from api_validations.validations_request import (
     InsertEmployees,
     BindEmployees2People,
     PatchRecord,
@@ -25,7 +26,9 @@ class EmployeeListEventMethods(MethodToEvent):
     }
 
     @classmethod
-    def employee_list(cls, list_options: ListOptions, token_dict):
+    def employee_list(
+            cls, list_options: ListOptions, token_dict: Union[EmployeeTokenObject, OccupantTokenObject]
+    ):
         Employees.filter_attr = list_options
         records = Employees.filter_active(
             *Employees.get_smart_query(smart_query=list_options.query),
@@ -44,7 +47,9 @@ class EmployeeCreateEventMethods(MethodToEvent):
     }
 
     @classmethod
-    def employee_create(cls, data: InsertEmployees, token_dict):
+    def employee_create(
+            cls, data: InsertEmployees, token_dict: Union[EmployeeTokenObject, OccupantTokenObject]
+    ):
         person = People.find_one(uu_id=data.people_uu_id)
         staff = Staff.find_one(uu_id=data.staff_uu_id)
         if not staff:
@@ -63,6 +68,7 @@ class EmployeeCreateEventMethods(MethodToEvent):
             people_id=person.id if person else None,
             people_uu_id=str(person.uu_id) if person else None,
         )
+        Employees.save()
         return JSONResponse(
             content={
                 "completed": True if not created_employee.is_found else False,
@@ -90,6 +96,7 @@ class EmployeeUpdateEventMethods(MethodToEvent):
         if access_authorized_employee.count:
             data_dict = data.excluded_dump()
             updated_employee = find_one_employee.update(**data_dict)
+            Employees.save()
             return JSONResponse(
                 content={
                     "completed": True,
@@ -116,7 +123,9 @@ class EmployeePatchEventMethods(MethodToEvent):
     }
 
     @classmethod
-    def employee_patch(cls, employee_uu_id: str, data: PatchRecord, token_dict):
+    def employee_patch(
+            cls, employee_uu_id: str, data: PatchRecord, token_dict: Union[EmployeeTokenObject, OccupantTokenObject]
+    ):
         find_one_employee = Employees.find_one_or_abort(uu_id=employee_uu_id)
         access_authorized_employee = Employees.select_action(
             employee_id=getattr(token_dict, "employee_id", 5),
@@ -133,7 +142,6 @@ class EmployeePatchEventMethods(MethodToEvent):
             find_one_employee.deleted = bool(
                 action.get("delete", find_one_employee.deleted)
             )
-            find_one_employee.save()
             return JSONResponse(
                 content={
                     "completed": True,
@@ -197,7 +205,7 @@ class Employee2PeopleEmployEventMethods(MethodToEvent):
             expiry_starts=data.expiry_starts,
             **token_dict.update_creds,
         )
-
+        Employees.save()
         return JSONResponse(
             content={
                 "completed": True,
@@ -257,6 +265,7 @@ class Employee2PeopleFireEventMethods(MethodToEvent):
                 is_notification_send=find_one_employee.is_notification_send,
                 cryp_uu_id=find_one_employee.cryp_uu_id,
             )
+            Employees.save()
             return JSONResponse(
                 content={
                     "completed": True,