wag-managment-api-service-version-2/api_events/events/identity/users.py

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 api_validations.validations_request import (
    InsertUsers,
    UpdateUsers,
    PatchRecord,
    ListOptions,
    RegisterServices2Occupant,
)


class UserListEventMethods(MethodToEvent):

    event_type = "SELECT"
    __event_keys__ = {
        "1483a8a2-d244-4593-b9f8-f1b4bcbefcd5": "user_list",
    }

    @classmethod
    def user_list(
        cls,
        list_options: ListOptions,
        token_dict: typing.Union[EmployeeTokenObject, OccupantTokenObject],
    ):
        # Users.pre_query = Users.select_action(duty_id_list=[
        #     token_dict.selected_company.duty_id,
        #     token_dict.selected_company.bulk_duties_id
        # ])
        if "user_uu_id_list" in list_options.query:
            people_ids = list_options.query.pop("user_uu_id_list")
            people_id_list = [
                user.person_id
                for user in Users.filter_active(Users.uu_id.in_(people_ids)).data
            ]
            records = People.filter_active(People.id.in_(people_id_list))
            return return_json_response_from_alchemy(
                response=records, pagination=list_options
            )
        records = Users.filter_active(*Users.get_smart_query(list_options.query))
        return return_json_response_from_alchemy(
            response=records, pagination=list_options
        )


class UserCreateEventMethods(MethodToEvent):

    event_type = "CREATE"
    __event_keys__ = {
        "8eb50c24-4bdc-4309-9836-f7048daee409": "user_create",
    }

    @classmethod
    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(
            **{"user_uu_id": str(created_user.uu_id), "main_domain": "evyos.com.tr"}
        )
        created_user.save()
        mongo_query_identity = MongoQueryIdentity(
            company_uuid=created_user.related_company,
        )
        mongo_query_identity.create_domain_via_user(payload=domain_via_user)
        return JSONResponse(
            content={
                "completed": True,
                "message": "Create User record",
                "data": created_user.get_dict(),
                "password_token": {
                    "password_token": created_user.password_token,
                    "password_expires_day": str(created_user.password_expires_day),
                    "password_expiry_begins": str(created_user.password_expiry_begins),
                    "hash_password": created_user.hash_password,
                    "related_company": created_user.related_company,
                },
            },
            status_code=status.HTTP_200_OK,
        )


class UserUpdateEventMethods(MethodToEvent):

    event_type = "UPDATE"
    __event_keys__ = {
        "d08a9470-1eb0-4890-a9e8-b6686239d7e9": "user_update",
    }

    @classmethod
    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),
            filter_expr=[Companies.id == token_dict.get("")],
        )
        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,
            )
        return JSONResponse(
            content={"completed": True, "message": "Update User record", "data": {}},
            status_code=status.HTTP_200_OK,
        )


class UserPatchEventMethods(MethodToEvent):

    event_type = "PATCH"
    __event_keys__ = {
        "d26a1a3c-eaeb-4d01-b35b-a5ed714e29c0": "user_patch",
    }

    @classmethod
    def user_patch(cls, data: PatchRecord, user_uu_id: str, token_dict):
        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),
            filter_expr=[Companies.id == find_one_user.id],
        )
        if access_authorized_company.count:
            action = data.excluded_dump()
            find_one_user.active = bool(action.get("active", find_one_user.active))
            find_one_user.is_confirmed = bool(
                action.get("confirm", find_one_user.is_confirmed)
            )
            find_one_user.deleted = bool(action.get("delete", find_one_user.deleted))
            find_one_user.save()
            return JSONResponse(
                content={
                    "completed": True,
                    "message": "Patch User record completed",
                    "data": find_one_user.get_dict(),
                },
                status_code=status.HTTP_200_OK,
            )
        return JSONResponse(
            content={
                "completed": False,
                "message": "Patch User record failed",
                "data": {},
            },
            status_code=status.HTTP_200_OK,
        )


UserListEventMethod = UserListEventMethods(action=ActionsSchema(endpoint="/user/list"))
UserCreateEventMethod = UserCreateEventMethods(
    action=ActionsSchema(endpoint="/user/create")
)
UserUpdateEventMethod = UserUpdateEventMethods(
    action=ActionsSchema(endpoint="/user/update")
)
UserPatchEventMethod = UserPatchEventMethods(
    action=ActionsSchema(endpoint="/user/patch")
)