496 lines
18 KiB
Python
496 lines
18 KiB
Python
import typing
|
|
|
|
from ApiEvents.abstract_class import (
|
|
MethodToEvent,
|
|
RouteFactoryConfig,
|
|
EndpointFactoryConfig,
|
|
endpoint_wrapper,
|
|
)
|
|
from ApiEvents.base_request_model import BaseRequestModel, DictRequestModel
|
|
|
|
from typing import TYPE_CHECKING, Dict, Any
|
|
from fastapi import Request, Path, Body, Depends, APIRouter
|
|
from pydantic import BaseModel, Field
|
|
|
|
if TYPE_CHECKING:
|
|
from fastapi import Request
|
|
|
|
from ApiValidations.Custom.token_objects import (
|
|
OccupantTokenObject,
|
|
EmployeeTokenObject,
|
|
)
|
|
from Schemas import (
|
|
BuildLivingSpace,
|
|
AccountRecords,
|
|
BuildIbans,
|
|
BuildDecisionBookPayments,
|
|
ApiEnumDropdown,
|
|
)
|
|
from ApiLibrary import system_arrow
|
|
from ApiValidations.Request import (
|
|
InsertAccountRecord,
|
|
UpdateAccountRecord,
|
|
ListOptions,
|
|
)
|
|
from Services.PostgresDb.Models.alchemy_response import (
|
|
AlchemyJsonResponse,
|
|
DictJsonResponse,
|
|
)
|
|
from ApiValidations.Response import AccountRecordResponse
|
|
|
|
|
|
class AddressUpdateRequest(BaseModel):
|
|
"""Request model for address update."""
|
|
|
|
data: Dict[str, Any] = Field(..., description="Updated address data")
|
|
|
|
|
|
class AddressUpdateResponse(BaseModel):
|
|
"""Response model for address update."""
|
|
|
|
address_uu_id: str = Field(..., description="UUID of the updated address")
|
|
data: Dict[str, Any] = Field(..., description="Updated address data")
|
|
function_code: str = Field(..., description="Function code for the endpoint")
|
|
|
|
|
|
class InsertAccountRecordRequestModel(BaseRequestModel[InsertAccountRecord]):
|
|
"""Request model for inserting account records."""
|
|
|
|
pass
|
|
|
|
|
|
class UpdateAccountRecordRequestModel(BaseRequestModel[UpdateAccountRecord]):
|
|
"""Request model for updating account records."""
|
|
|
|
pass
|
|
|
|
|
|
class ListOptionsRequestModel(BaseRequestModel[ListOptions]):
|
|
"""Request model for list options."""
|
|
|
|
pass
|
|
|
|
|
|
class AccountRecordsListEventMethods(MethodToEvent):
|
|
|
|
event_type = "SELECT"
|
|
event_description = ""
|
|
event_category = ""
|
|
|
|
__event_keys__ = {
|
|
"7192c2aa-5352-4e36-98b3-dafb7d036a3d": "account_records_list",
|
|
"208e6273-17ef-44f0-814a-8098f816b63a": "account_records_list_flt_res",
|
|
}
|
|
__event_validation__ = {
|
|
"7192c2aa-5352-4e36-98b3-dafb7d036a3d": AccountRecordResponse,
|
|
"208e6273-17ef-44f0-814a-8098f816b63a": AccountRecordResponse,
|
|
}
|
|
|
|
@classmethod
|
|
def account_records_list(
|
|
cls,
|
|
list_options: ListOptionsRequestModel,
|
|
token_dict: typing.Union[EmployeeTokenObject, OccupantTokenObject],
|
|
):
|
|
db_session = AccountRecords.new_session()
|
|
if isinstance(token_dict, OccupantTokenObject):
|
|
AccountRecords.pre_query = AccountRecords.filter_all(
|
|
AccountRecords.company_id
|
|
== token_dict.selected_occupant.responsible_company_id,
|
|
db=db_session,
|
|
).query
|
|
elif isinstance(token_dict, EmployeeTokenObject):
|
|
AccountRecords.pre_query = AccountRecords.filter_all(
|
|
AccountRecords.company_id == token_dict.selected_company.company_id,
|
|
db=db_session,
|
|
).query
|
|
AccountRecords.filter_attr = list_options
|
|
records = AccountRecords.filter_all(db=db_session)
|
|
return AlchemyJsonResponse(
|
|
completed=True,
|
|
message="Account records listed successfully",
|
|
result=records,
|
|
cls_object=AccountRecords,
|
|
filter_attributes=list_options,
|
|
response_model=AccountRecordResponse,
|
|
)
|
|
|
|
@classmethod
|
|
def account_records_list_flt_res(
|
|
cls,
|
|
list_options: ListOptionsRequestModel,
|
|
token_dict: typing.Union[EmployeeTokenObject, OccupantTokenObject],
|
|
):
|
|
db_session = AccountRecords.new_session()
|
|
if not isinstance(token_dict, OccupantTokenObject):
|
|
raise AccountRecords.raise_http_exception(
|
|
status_code="HTTP_404_NOT_FOUND",
|
|
error_case="UNAUTHORIZED",
|
|
message="Only Occupant can see this data",
|
|
data={},
|
|
)
|
|
|
|
return_list = []
|
|
living_space: BuildLivingSpace = BuildLivingSpace.filter_by_one(
|
|
id=token_dict.selected_occupant.living_space_id
|
|
).data
|
|
if not living_space:
|
|
raise AccountRecords.raise_http_exception(
|
|
status_code="HTTP_404_NOT_FOUND",
|
|
error_case="UNAUTHORIZED",
|
|
message="Living space not found",
|
|
data={},
|
|
)
|
|
|
|
if not list_options:
|
|
list_options = ListOptions()
|
|
|
|
main_filters = [
|
|
AccountRecords.living_space_id
|
|
== token_dict.selected_occupant.living_space_id,
|
|
BuildDecisionBookPayments.process_date
|
|
>= str(system_arrow.now().shift(months=-3).date()),
|
|
BuildDecisionBookPayments.process_date
|
|
< str(system_arrow.find_last_day_of_month(living_space.expiry_ends)),
|
|
BuildDecisionBookPayments.process_date
|
|
>= str(system_arrow.get(living_space.expiry_starts)),
|
|
BuildDecisionBookPayments.is_confirmed == True,
|
|
AccountRecords.active == True,
|
|
]
|
|
order_type = "desc"
|
|
if list_options.order_type:
|
|
order_type = "asc" if list_options.order_type[0] == "a" else "desc"
|
|
|
|
order_by_list = BuildDecisionBookPayments.process_date.desc()
|
|
if list_options.order_field:
|
|
if list_options.order_field == "process_date":
|
|
order_by_list = (
|
|
BuildDecisionBookPayments.process_date.asc()
|
|
if order_type == "asc"
|
|
else BuildDecisionBookPayments.process_date.desc()
|
|
)
|
|
if list_options.order_field == "bank_date":
|
|
order_by_list = (
|
|
AccountRecords.bank_date.desc()
|
|
if order_type == "asc"
|
|
else AccountRecords.bank_date.asc()
|
|
)
|
|
if list_options.order_field == "currency_value":
|
|
order_by_list = (
|
|
AccountRecords.currency_value.desc()
|
|
if order_type == "asc"
|
|
else AccountRecords.currency_value.asc()
|
|
)
|
|
if list_options.order_field == "process_comment":
|
|
order_by_list = (
|
|
AccountRecords.process_comment.desc()
|
|
if order_type == "asc"
|
|
else AccountRecords.process_comment.asc()
|
|
)
|
|
if list_options.order_field == "payment_amount":
|
|
order_by_list = (
|
|
BuildDecisionBookPayments.payment_amount.desc()
|
|
if order_type == "asc"
|
|
else BuildDecisionBookPayments.payment_amount.asc()
|
|
)
|
|
|
|
if list_options.query:
|
|
for key, value in list_options.query.items():
|
|
if key == "process_date":
|
|
main_filters.append(BuildDecisionBookPayments.process_date == value)
|
|
if key == "bank_date":
|
|
main_filters.append(AccountRecords.bank_date == value)
|
|
if key == "currency":
|
|
main_filters.append(BuildDecisionBookPayments.currency == value)
|
|
if key == "currency_value":
|
|
main_filters.append(AccountRecords.currency_value == value)
|
|
if key == "process_comment":
|
|
main_filters.append(AccountRecords.process_comment == value)
|
|
if key == "payment_amount":
|
|
main_filters.append(
|
|
BuildDecisionBookPayments.payment_amount == value
|
|
)
|
|
|
|
query = (
|
|
AccountRecords.session.query(
|
|
BuildDecisionBookPayments.process_date,
|
|
BuildDecisionBookPayments.payment_amount,
|
|
BuildDecisionBookPayments.currency,
|
|
AccountRecords.bank_date,
|
|
AccountRecords.currency_value,
|
|
AccountRecords.process_comment,
|
|
BuildDecisionBookPayments.uu_id,
|
|
)
|
|
.join(
|
|
AccountRecords,
|
|
AccountRecords.id == BuildDecisionBookPayments.account_records_id,
|
|
)
|
|
.filter(*main_filters)
|
|
).order_by(order_by_list)
|
|
|
|
query.limit(list_options.size or 5).offset(
|
|
(list_options.page or 1 - 1) * list_options.size or 5
|
|
)
|
|
for list_of_values in query.all() or []:
|
|
return_list.append(
|
|
{
|
|
"process_date": list_of_values[0],
|
|
"payment_amount": list_of_values[1],
|
|
"currency": list_of_values[2],
|
|
"bank_date": list_of_values[3],
|
|
"currency_value": list_of_values[4],
|
|
"process_comment": list_of_values[5],
|
|
}
|
|
)
|
|
return AlchemyJsonResponse(
|
|
completed=True,
|
|
message="Account records listed successfully",
|
|
result=return_list,
|
|
cls_object=AccountRecords,
|
|
filter_attributes=list_options,
|
|
response_model=AccountRecordResponse,
|
|
)
|
|
|
|
|
|
class AccountRecordsCreateEventMethods(MethodToEvent):
|
|
|
|
event_type = "CREATE"
|
|
event_description = ""
|
|
event_category = ""
|
|
|
|
__event_keys__ = {
|
|
"31f4f32f-0cd4-4995-8a6a-f9f56335848a": "account_records_create",
|
|
}
|
|
__event_validation__ = {
|
|
"31f4f32f-0cd4-4995-8a6a-f9f56335848a": InsertAccountRecord,
|
|
}
|
|
|
|
@classmethod
|
|
def account_records_create(
|
|
cls,
|
|
data: InsertAccountRecordRequestModel,
|
|
token_dict: typing.Union[EmployeeTokenObject, OccupantTokenObject],
|
|
):
|
|
data_dict = data.excluded_dump()
|
|
if isinstance(token_dict, OccupantTokenObject):
|
|
db_session = AccountRecords.new_session()
|
|
build_iban = BuildIbans.filter_one(
|
|
BuildIbans.iban == data.iban,
|
|
BuildIbans.build_id == token_dict.selected_occupant.build_id,
|
|
db=db_session,
|
|
).data
|
|
if not build_iban:
|
|
raise BuildIbans.raise_http_exception(
|
|
status_code="HTTP_404_NOT_FOUND",
|
|
error_case="UNAUTHORIZED",
|
|
message=f"{data.iban} is not found in company related to your organization",
|
|
data={
|
|
"iban": data.iban,
|
|
},
|
|
)
|
|
account_record = AccountRecords.find_or_create(**data.excluded_dump())
|
|
return AlchemyJsonResponse(
|
|
completed=True,
|
|
message="Account record created successfully",
|
|
result=account_record,
|
|
)
|
|
elif isinstance(token_dict, EmployeeTokenObject):
|
|
# Build.pre_query = Build.select_action(
|
|
# employee_id=token_dict.selected_employee.employee_id,
|
|
# )
|
|
# build_ids_list = Build.filter_all(
|
|
# )
|
|
# build_iban = BuildIbans.filter_one(
|
|
# BuildIbans.iban == data.iban,
|
|
# BuildIbans.build_id.in_([build.id for build in build_ids_list.data]),
|
|
# ).data
|
|
# if not build_iban:
|
|
# BuildIbans.raise_http_exception(
|
|
# status_code="HTTP_404_NOT_FOUND",
|
|
# error_case="UNAUTHORIZED",
|
|
# message=f"{data.iban} is not found in company related to your organization",
|
|
# data={
|
|
# "iban": data.iban,
|
|
# },
|
|
# )
|
|
bank_date = system_arrow.get(data.bank_date)
|
|
data_dict["bank_date_w"] = bank_date.weekday()
|
|
data_dict["bank_date_m"] = bank_date.month
|
|
data_dict["bank_date_d"] = bank_date.day
|
|
data_dict["bank_date_y"] = bank_date.year
|
|
|
|
if int(data.currency_value) < 0:
|
|
debit_type = ApiEnumDropdown.filter_by_one(
|
|
system=True, enum_class="DebitTypes", key="DT-D"
|
|
).data
|
|
data_dict["receive_debit"] = debit_type.id
|
|
data_dict["receive_debit_uu_id"] = str(debit_type.uu_id)
|
|
else:
|
|
debit_type = ApiEnumDropdown.filter_by_one(
|
|
system=True, enum_class="DebitTypes", key="DT-R"
|
|
).data
|
|
data_dict["receive_debit"] = debit_type.id
|
|
data_dict["receive_debit_uu_id"] = str(debit_type.uu_id)
|
|
|
|
account_record = AccountRecords.insert_one(data_dict).data
|
|
return AlchemyJsonResponse(
|
|
completed=True,
|
|
message="Account record created successfully",
|
|
result=account_record,
|
|
)
|
|
|
|
|
|
class AccountRecordsUpdateEventMethods(MethodToEvent):
|
|
|
|
event_type = "UPDATE"
|
|
event_description = ""
|
|
event_category = ""
|
|
|
|
__event_keys__ = {
|
|
"ec98ef2c-bcd0-432d-a8f4-1822a56c33b2": "account_records_update",
|
|
}
|
|
__event_validation__ = {
|
|
"ec98ef2c-bcd0-432d-a8f4-1822a56c33b2": UpdateAccountRecord,
|
|
}
|
|
|
|
@classmethod
|
|
def build_area_update(
|
|
cls,
|
|
build_uu_id: str,
|
|
data: UpdateAccountRecordRequestModel,
|
|
token_dict: typing.Union[EmployeeTokenObject, OccupantTokenObject],
|
|
):
|
|
if isinstance(token_dict, OccupantTokenObject):
|
|
pass
|
|
elif isinstance(token_dict, EmployeeTokenObject):
|
|
pass
|
|
AccountRecords.build_parts_id = token_dict.selected_occupant.build_part_id
|
|
account_record = AccountRecords.update_one(build_uu_id, data).data
|
|
return AlchemyJsonResponse(
|
|
completed=True,
|
|
message="Account record updated successfully",
|
|
result=account_record,
|
|
cls_object=AccountRecords,
|
|
response_model=UpdateAccountRecord,
|
|
)
|
|
|
|
|
|
@endpoint_wrapper("/account/records/address/list")
|
|
async def address_list(request: "Request", data: ListOptionsRequestModel):
|
|
"""Handle address list endpoint."""
|
|
return {
|
|
"data": data,
|
|
"request": str(request.headers),
|
|
"request_url": str(request.url),
|
|
"request_base_url": str(request.base_url),
|
|
}
|
|
|
|
|
|
@endpoint_wrapper("/account/records/address/create")
|
|
async def address_create(request: "Request", data: DictRequestModel):
|
|
"""Handle address creation endpoint."""
|
|
return {
|
|
"data": data,
|
|
"request": str(request.headers),
|
|
"request_url": str(request.url),
|
|
"request_base_url": str(request.base_url),
|
|
}
|
|
|
|
|
|
@endpoint_wrapper("/account/records/address/search")
|
|
async def address_search(request: "Request", data: DictRequestModel):
|
|
"""Handle address search endpoint."""
|
|
return {"data": data}
|
|
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@endpoint_wrapper("/account/records/address/{address_uu_id}")
|
|
async def address_update(
|
|
request: Request,
|
|
address_uu_id: str = Path(..., description="UUID of the address to update"),
|
|
request_data: DictRequestModel = Body(..., description="Request body"),
|
|
):
|
|
"""
|
|
Handle address update endpoint.
|
|
|
|
Args:
|
|
request: FastAPI request object
|
|
address_uu_id: UUID of the address to update
|
|
request_data: Request body containing updated address data
|
|
|
|
Returns:
|
|
DictJsonResponse: Response containing updated address info
|
|
"""
|
|
return DictJsonResponse(
|
|
data={
|
|
"address_uu_id": address_uu_id,
|
|
"data": request_data.root,
|
|
"request": str(request.headers),
|
|
"request_url": str(request.url),
|
|
"request_base_url": str(request.base_url),
|
|
}
|
|
)
|
|
|
|
|
|
prefix = "/account/records"
|
|
|
|
# Account Records Router Configuration
|
|
ACCOUNT_RECORDS_CONFIG = RouteFactoryConfig(
|
|
name="account_records",
|
|
prefix=prefix,
|
|
tags=["Account Records"],
|
|
include_in_schema=True,
|
|
endpoints=[
|
|
EndpointFactoryConfig(
|
|
url_prefix=prefix,
|
|
url_endpoint="/address/list",
|
|
url_of_endpoint="/account/records/address/list",
|
|
endpoint="/address/list",
|
|
method="POST",
|
|
summary="List Active/Delete/Confirm Address",
|
|
description="List Active/Delete/Confirm Address",
|
|
is_auth_required=True,
|
|
is_event_required=True,
|
|
endpoint_function=address_list,
|
|
),
|
|
EndpointFactoryConfig(
|
|
url_prefix=prefix,
|
|
url_endpoint="/address/create",
|
|
url_of_endpoint="/account/records/address/create",
|
|
endpoint="/address/create",
|
|
method="POST",
|
|
summary="Create Address with given auth levels",
|
|
description="Create Address with given auth levels",
|
|
is_auth_required=False,
|
|
is_event_required=False,
|
|
endpoint_function=address_create,
|
|
),
|
|
EndpointFactoryConfig(
|
|
url_prefix=prefix,
|
|
url_endpoint="/address/search",
|
|
url_of_endpoint="/account/records/address/search",
|
|
endpoint="/address/search",
|
|
method="POST",
|
|
summary="Search Address with given auth levels",
|
|
description="Search Address with given auth levels",
|
|
is_auth_required=True,
|
|
is_event_required=True,
|
|
endpoint_function=address_search,
|
|
),
|
|
EndpointFactoryConfig(
|
|
url_prefix=prefix,
|
|
url_endpoint="/address/{address_uu_id}",
|
|
url_of_endpoint="/account/records/address/{address_uu_id}",
|
|
endpoint="/address/{address_uu_id}",
|
|
method="PUT",
|
|
summary="Update Address with given auth levels",
|
|
description="Update Address with given auth levels",
|
|
is_auth_required=True,
|
|
is_event_required=True,
|
|
endpoint_function=address_update,
|
|
),
|
|
],
|
|
).as_dict()
|