first commit

api_events/events/company/company_duties.py

@@ -0,0 +1,199 @@
+from fastapi import status
+from fastapi.responses import JSONResponse
+
+from validations import InsertDuties, UpdateDuties, SelectDuties
+from validations.root_validates import PatchRecord, ListOptions
+
+from databases import Departments, Duty, Duties
+
+from api_events.events.abstract_class import MethodToEvent, ActionsSchema
+from api_objects.auth.token_objects import EmployeeTokenObject, OccupantTokenObject
+
+
+
+class DutiesListEventMethods(MethodToEvent):
+
+    event_type = "SELECT"
+    __event_keys__ = {
+        "44b72beb-53a8-407b-a12a-76e74b65794d": "duties_list",
+    }
+
+    @classmethod
+    def duties_list(cls, list_options: ListOptions, token_dict):
+        Duties.filter_attr = list_options
+        records = Duties.filter_active(
+            *Duties.get_smart_query(smart_query=list_options.query),
+            Duties.company_id == token_dict.selected_company.company_id
+        )
+        return {
+            "completed": True if records.count else False,
+            "status": "success",
+            "data": records.get_dict(),
+        }
+
+
+class DutiesGetByUUIDEventMethods(MethodToEvent):
+
+    event_type = "GET"
+    __event_keys__ = {
+        "30c54cce-3303-4d36-959a-b64e383ae177": "duties_get_by_uuid",
+    }
+
+    @classmethod
+    def duties_get_by_uuid(cls, data: SelectDuties, token_dict):
+
+        duty = Duty.find_one(uu_id=data.duty_uu_id)
+        if not duty:
+            return JSONResponse(
+                content={
+                    "completed": False,
+                    "message": "Duty record is not found",
+                    "data": {},
+                },
+                status_code=status.HTTP_404_NOT_FOUND,
+            )
+
+        records = Duties.filter_active(
+            Duties.duties_id == duty.id,
+            Duties.company_id == token_dict.selected_company.company_id,
+        )
+        if not records.data:
+            return JSONResponse(
+                content={
+                    "completed": False,
+                    "message": "Duties record is not found",
+                    "data": {},
+                },
+                status_code=status.HTTP_404_NOT_FOUND,
+            )
+        return {
+            "completed": True,
+            "status": "success",
+            "data": [record.get_dict() for record in records.data],
+        }
+
+
+class DutiesCreateEventMethods(MethodToEvent):
+
+    event_type = "CREATE"
+    __event_keys__ = {
+        "3524ae42-0825-4af7-be85-7c890a4f65d3": "duties_create",
+    }
+
+    @classmethod
+    def duties_create(cls, data: InsertDuties, token_dict):
+        duty = Duty.find_one(uu_id=data.duties_uu_id)
+        department = Departments.find_one(uu_id=data.department_uu_id)
+
+        created_duties = Duties.find_or_create(
+            company_id=token_dict.selected_company.company_id,
+            company_uu_id=token_dict.selected_company.company_uu_id,
+            duties_id=duty.id,
+            duties_uu_id=str(duty.uu_id),
+            department_id=department.id,
+            department_uu_id=str(department.uu_id),
+            is_confirmed=data.is_confirmed,
+        )
+        if data.is_default_duty:
+            created_duties.update(users_default_duty=created_duties.id)
+
+        if not created_duties:
+            return JSONResponse(
+                content={
+                    "completed": False,
+                    "message": "Failed to create Duties record",
+                    "data": {},
+                },
+                status_code=status.HTTP_400_BAD_REQUEST,
+            )
+        return {
+            "completed": created_duties.is_found,
+            "message": (
+                "Create Duties record"
+                if created_duties.is_found
+                else "This record is already created"
+            ),
+            "data": created_duties.get_dict(),
+        }
+
+
+class DutiesUpdateEventMethods(MethodToEvent):
+
+    event_type = "UPDATE"
+    __event_keys__ = {
+        "3fc77829-f1ee-4511-a2ca-582daa03125b": "duties_update",
+    }
+
+    @classmethod
+    def duties_update(cls, duties_uu_id: str, data: UpdateDuties, token_dict):
+        find_one_duties = Duties.find_one_or_abort(uu_id=duties_uu_id)
+        access_authorized_duties = Duties.select_action(
+            duty_id=getattr(token_dict, "duty_id", 5),
+            filter_expr=[Duties.id == find_one_duties.id],
+        )
+        if access_authorized_duties.count:
+            data_dict = data.excluded_dump()
+            updated_duties = find_one_duties.update(**data_dict)
+            return {
+                "completed": True,
+                "message": "Update Duties record",
+                "data": updated_duties,
+            }
+        return {
+            "completed": False,
+            "message": "Update Duties record failed",
+            "data": {},
+        }
+
+
+class DutiesPatchEventMethods(MethodToEvent):
+
+    event_type = "PATCH"
+    __event_keys__ = {
+        "ca81c6d1-975a-4288-a27b-1069aea84afe": "duties_patch",
+    }
+
+    @classmethod
+    def duties_patch(cls, duties_uu_id: str, data: PatchRecord, token_dict):
+        find_one_duties = Duties.find_one_or_abort(uu_id=duties_uu_id)
+        access_authorized_duties = Duties.select_action(
+            duty_id=getattr(token_dict, "duty_id", 5),
+            filter_expr=[Duties.id == find_one_duties.id],
+        )
+        if access_authorized_duties.count:
+            action = data.excluded_dump()
+            find_one_duties.active = bool(action.get("active", find_one_duties.active))
+            find_one_duties.is_confirmed = bool(
+                action.get("confirm", find_one_duties.is_confirmed)
+            )
+            find_one_duties.deleted = bool(
+                action.get("delete", find_one_duties.deleted)
+            )
+            find_one_duties.save()
+            return {
+                "completed": True,
+                "message": "Patch Duties record completed",
+                "data": find_one_duties.get_dict(),
+            }
+        return {
+            "completed": False,
+            "message": "Patch Duties record failed",
+            "data": {},
+        }
+
+
+DutiesListEventMethod = DutiesListEventMethods(
+    action=ActionsSchema(endpoint="/duties/list")
+)
+DutiesGetByUUIDEventMethod = DutiesGetByUUIDEventMethods(
+    action=ActionsSchema(endpoint="/duties/get_by_duty_uuid")
+)
+DutiesCreateEventMethod = DutiesCreateEventMethods(
+    action=ActionsSchema(endpoint="/duties/create")
+)
+DutiesUpdateEventMethod = DutiesUpdateEventMethods(
+    action=ActionsSchema(endpoint="/duties/update")
+)
+DutiesPatchEventMethod = DutiesPatchEventMethods(
+    action=ActionsSchema(endpoint="/duties/patch")
+)