wag-managment-api-service-version-2/api_events/events/company/company_company.py

import typing

from fastapi import status
from fastapi.responses import JSONResponse

from databases import Companies

from api_validations import (
    InsertCompany,
    UpdateCompany,
    ListOptions,
    PatchRecord,
)

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


class CompanyListEventMethods(MethodToEvent):

    event_type = "SELECT"
    __event_keys__ = {
        "f6900cb5-ac5b-478e-8e7c-fa87e65cd2e5": "company_list",
    }

    @classmethod
    def company_list(
        cls,
        list_options: ListOptions,
        token_dict: typing.Union[EmployeeTokenObject, OccupantTokenObject],
    ):
        if isinstance(token_dict, EmployeeTokenObject):
            Companies.pre_query = Companies.select_action(
                duty_id_list=[
                    token_dict.selected_company.duty_id,
                    token_dict.selected_company.bulk_duties_id,
                ]
            )
        elif isinstance(token_dict, OccupantTokenObject):
            Companies.pre_query = Companies.filter_active(
                Companies.id == token_dict.selected_occupant.responsible_company_id
            ).query
        Companies.filter_attr = list_options
        records = Companies.filter_active(
            *Companies.get_smart_query(list_options.query)
        )
        return return_json_response_from_alchemy(
            response=records, pagination=list_options
        )


class CompanyCreateEventMethods(MethodToEvent):

    event_type = "CREATE"
    __event_keys__ = {
        "76f11a08-5f4a-4e1f-961f-aaef21699acd": "company_create",
    }

    @classmethod
    def company_create(cls, data: InsertCompany, token_dict):
        created_company = Companies.create_action(
            data=data, token=token_dict.companies_list
        )
        created_company.related_company = token_dict.get("company_uu_id")
        return JSONResponse(
            content={
                "completed": True,
                "message": "Create Company record",
                "data": created_company.get_dict(),
                "password_token": {
                    "password_token": created_company.password_token,
                    "password_expires_day": str(created_company.password_expires_day),
                    "password_expiry_begins": str(
                        created_company.password_expiry_begins
                    ),
                    "hash_password": created_company.hash_password,
                    "related_company": created_company.related_company,
                },
            },
            status_code=status.HTTP_200_OK,
        )


class CompanyUpdateEventMethods(MethodToEvent):

    event_type = "UPDATE"
    __event_keys__ = {
        "41ea7f29-006a-4310-b5c4-b2a0e1a504bd": "company_update",
    }

    @classmethod
    def company_update(cls, company_uu_id: str, data: UpdateCompany, token_dict):
        find_one_company = Companies.find_one_or_abort(uu_id=company_uu_id)
        access_authorized_company = Companies.select_action(
            duty_id=getattr(token_dict, "duty_id", 5),
            filter_expr=[Companies.id == token_dict.get("")],
        )
        if access_authorized_company.count:
            data_dict = data.excluded_dump()
            updated_company = find_one_company.update(**data_dict)
            return JSONResponse(
                content={
                    "completed": True,
                    "message": "Update Company record",
                    "data": updated_company,
                },
                status_code=status.HTTP_200_OK,
            )
        return JSONResponse(
            content={"completed": True, "message": "Update Company record", "data": {}},
            status_code=status.HTTP_200_OK,
        )


class CompanyPatchEventMethods(MethodToEvent):

    event_type = "PATCH"
    __event_keys__ = {
        "6320d696-1fd1-49f9-860a-8f22e5b8a68d": "company_patch",
    }

    @classmethod
    def company_patch(cls, company_uu_id: str, data: PatchRecord, token_dict):
        find_one_company = Companies.find_one_or_abort(uu_id=company_uu_id)
        access_authorized_company = Companies.select_action(
            duty_id=getattr(token_dict, "duty_id", 5),
            filter_expr=[Companies.id == find_one_company.id],
        )
        if access_authorized_company.count:
            action = data.excluded_dump()
            find_one_company.active = bool(
                action.get("active", find_one_company.active)
            )
            find_one_company.is_confirmed = bool(
                action.get("confirm", find_one_company.is_confirmed)
            )
            find_one_company.deleted = bool(
                action.get("delete", find_one_company.deleted)
            )
            find_one_company.save()
            return JSONResponse(
                content={
                    "completed": True,
                    "message": "Patch Company record completed",
                    "data": find_one_company.get_dict(),
                },
                status_code=status.HTTP_200_OK,
            )
        return JSONResponse(
            content={
                "completed": False,
                "message": "Patch Company record failed",
                "data": {},
            },
            status_code=status.HTTP_200_OK,
        )


CompanyPatchEventMethod = CompanyListEventMethods(
    action=ActionsSchema(endpoint="/company/list")
)
CompanyCreateEventMethod = CompanyCreateEventMethods(
    action=ActionsSchema(endpoint="/company/create")
)
CompanyUpdateEventMethod = CompanyUpdateEventMethods(
    action=ActionsSchema(endpoint="/company/update")
)
CompanyListEventMethod = CompanyPatchEventMethods(
    action=ActionsSchema(endpoint="/company/patch")
)