events updated

api_events/events/identity/people.py

@@ -4,13 +4,13 @@ from fastapi import status
 from fastapi.responses import JSONResponse
 
 from databases import (
+    Build,
     People,
     Users,
     Companies,
 )
 
 from api_validations.validations_request import InsertPerson, UpdateUsers
-
 from api_events.events.abstract_class import MethodToEvent, ActionsSchema
 from api_objects.auth.token_objects import EmployeeTokenObject, OccupantTokenObject
 from api_validations.core_response import AlchemyJsonResponse
@@ -22,18 +22,40 @@ class PeopleListEventMethods(MethodToEvent):
     __event_keys__ = {
         "0a05f03c-6ed8-4230-a4ff-6e7cf886909b": "super_users_people_list",
         "b5612538-0445-4a4a-ab13-d2a06037f7a5": "sales_users_people_list",
-        "c81c2cec-d32c-4cf2-9727-d4493e11ee1f": "human_resources_users_people_list",
-        "d1b1b1b1-1b1b-1b1b-1b1b-1b1b1b1b1b1b": "people_list_only_occupant_tenant_or_owner",
+        "25cbbaf8-117a-470f-a844-2cfc70f71dde": "human_resources_users_people_list",
+        "cdf62f06-ec50-40de-b19e-adb3dd34bb95": "people_list_only_occupant_tenant_or_owner",
     }
 
     @classmethod
     def super_users_people_list(
-        cls, list_options, token_dict: Union[EmployeeTokenObject, OccupantTokenObject]
+            cls,
+            list_options,
+            token_dict: Union[EmployeeTokenObject, OccupantTokenObject]
     ):
-        records = People.filter_all(
-            *People.get_smart_query(smart_query=list_options.query),
-            *People.valid_record_args(People),
-        ).data
+        records = []
+        if isinstance(token_dict, EmployeeTokenObject):
+            People.pre_query = People.select_action(
+                duty_id_list=[
+                    token_dict.selected_company.duty_id,
+                    token_dict.selected_company.bulk_duties_id,
+                ],
+            )
+            records = People.filter_all(
+                *People.get_smart_query(smart_query=list_options.query),
+                *People.valid_record_args(People),
+            ).data
+        elif isinstance(token_dict, OccupantTokenObject):
+            related_users =Users.filter_all(
+                Users.related_company == token_dict.selected_occupant.responsible_company_id,
+            ).data
+            People.pre_query = People.filter_all(
+                People.id.in_([user.person_id for user in related_users]),
+                *People.valid_record_args(People),
+            ).query
+            records = People.filter_all(
+                *People.get_smart_query(smart_query=list_options.query),
+                *People.valid_record_args(People),
+            ).data
         return AlchemyJsonResponse(
             completed=True,
             message="People are listed successfully",
@@ -42,7 +64,9 @@ class PeopleListEventMethods(MethodToEvent):
 
     @classmethod
     def sales_users_people_list(
-        cls, list_options, token_dict: Union[EmployeeTokenObject, OccupantTokenObject]
+            cls,
+            list_options,
+            token_dict: EmployeeTokenObject,
     ):
         records = People.filter_all(
             *People.get_smart_query(smart_query=list_options.query),
@@ -56,12 +80,15 @@ class PeopleListEventMethods(MethodToEvent):
 
     @classmethod
     def human_resources_users_people_list(
-        cls, list_options, token_dict: Union[EmployeeTokenObject, OccupantTokenObject]
+            cls,
+            list_options,
+            token_dict: EmployeeTokenObject,
     ):
-        records = People.filter_all(
-            *People.get_smart_query(smart_query=list_options.query),
-            *People.valid_record_args(People),
-        ).data
+        if isinstance(token_dict, EmployeeTokenObject):
+            records = People.filter_all(
+                *People.get_smart_query(smart_query=list_options.query),
+                *People.valid_record_args(People),
+            ).data
         return AlchemyJsonResponse(
             completed=True,
             message="People are listed successfully",
@@ -82,8 +109,12 @@ class PeopleCreateEventMethods(MethodToEvent):
         data: InsertPerson,
         token_dict: Union[EmployeeTokenObject, OccupantTokenObject],
     ):
-        created_user = People.create_action(data=data, token=token_dict)
-        People.save()
+        if isinstance(token_dict, EmployeeTokenObject):
+            created_user = People.create_action(data=data, token=token_dict)
+            People.save()
+        elif isinstance(token_dict, OccupantTokenObject):
+            created_user = People.create_action(data=data, token=token_dict)
+            People.save()
         return JSONResponse(
             content={
                 "completed": True,
@@ -108,26 +139,42 @@ class PeopleUpdateEventMethods(MethodToEvent):
         user_uu_id: str,
         token_dict: Union[EmployeeTokenObject, OccupantTokenObject],
     ):
-        find_one_user = Users.filter_one(
-            Users.uu_id == user_uu_id,
-            *Users.valid_record_args(Users),
-        ).data
-        access_authorized_company = Companies.select_action(
-            duty_id_list=[getattr(token_dict, "duty_id")],
-            filter_expr=[Companies.id == find_one_user.id],
-        )
-        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,
-                    "message": "Update User record",
-                    "data": updated_user,
-                },
-                status_code=status.HTTP_200_OK,
+        data_dict = data.excluded_dump()
+        if isinstance(token_dict, EmployeeTokenObject):
+            find_one_user = Users.filter_one(
+                Users.uu_id == user_uu_id,
+                *Users.valid_record_args(Users),
+            ).data
+            access_authorized_company = Companies.select_action(
+                duty_id_list=[
+                    token_dict.selected_company.duty_id,
+                    token_dict.selected_company.bulk_duties_id
+                ],
             )
+            if access_authorized_company.count:
+                updated_user = find_one_user.update(**data_dict)
+                Users.save()
+                return JSONResponse(
+                    content={
+                        "completed": True,
+                        "message": "Update User record",
+                        "data": updated_user,
+                    },
+                    status_code=status.HTTP_200_OK,
+                )
+        elif isinstance(token_dict, OccupantTokenObject):
+            find_one_user = People.filter_one(
+                People.uu_id == user_uu_id,
+                *People.valid_record_args(People),
+            ).data
+            access_authorized_company = Companies.select_action(
+                duty_id_list=[getattr(token_dict, "duty_id")],
+                filter_expr=[Companies.id == find_one_user.id],
+            )
+            if access_authorized_company.count:
+                data_dict = data.excluded_dump()
+                updated_user = find_one_user.update(**data_dict)
+                People.save()
         return JSONResponse(
             content={"completed": True, "message": "Update User record", "data": {}},
             status_code=status.HTTP_200_OK,