187 lines
8.0 KiB
Python
187 lines
8.0 KiB
Python
from typing import Any
|
|
|
|
from Initializer.event_clusters import Event
|
|
from Validations.response import (
|
|
PaginateOnly,
|
|
Pagination,
|
|
PaginationResult,
|
|
PostgresResponseSingle,
|
|
PostgresResponse,
|
|
EndpointResponse
|
|
)
|
|
|
|
from Schemas import (
|
|
Events,
|
|
Event2Employee,
|
|
Event2Occupant,
|
|
Event2EmployeeExtra,
|
|
Event2OccupantExtra,
|
|
Service2Events,
|
|
Services,
|
|
)
|
|
|
|
# List available events endpoint
|
|
EventsListAvailableEvent = Event(
|
|
name="event_endpoint_list_available",
|
|
key="d39af512-ec71-4c0f-9b35-e53b0d06d3a4",
|
|
request_validator=None, # TODO: Add request validator
|
|
response_validator=None, # TODO: Add response validator
|
|
description="Super Users List available events endpoint",
|
|
)
|
|
|
|
# List appended events endpoint
|
|
EventsListAppendedEvent = Event(
|
|
name="event_endpoint_list_appended",
|
|
key="bea77d6a-d99f-468b-9002-b3bda6bb6ad0",
|
|
request_validator=None, # TODO: Add request validator
|
|
response_validator=None, # TODO: Add response validator
|
|
description="Super Users List appended events endpoint",
|
|
)
|
|
|
|
# Event Register endpoint
|
|
EventRegisterServiceEvent = Event(
|
|
name="event_endpoint_register_service",
|
|
key="e18e7f89-5708-4a15-9258-99b0903ed43d",
|
|
request_validator=None, # TODO: Add request validator
|
|
response_validator=None, # TODO: Add response validator
|
|
description="Super Users Register service endpoint",
|
|
)
|
|
|
|
# Event Unregister endpoint
|
|
EventUnRegisterServiceEvent = Event(
|
|
name="service_endpoint_unregister_service",
|
|
key="4d693774-4857-435b-a63c-c39baebfe916",
|
|
request_validator=None, # TODO: Add request validator
|
|
response_validator=None, # TODO: Add response validator
|
|
description="Super Users Unregister service endpoint",
|
|
)
|
|
|
|
# Bind employee extra endpoint
|
|
EventBindEmployeeExtraEvent = Event(
|
|
name="service_endpoint_bind_employee_extra",
|
|
key="cd452928-4256-4fb4-b81e-0ca41d723616",
|
|
request_validator=None, # TODO: Add request validator
|
|
response_validator=None, # TODO: Add response validator
|
|
description="Super Users Bind service to employee extra endpoint",
|
|
)
|
|
|
|
# Bind occupant extra endpoint
|
|
EventBindOccupantExtraEvent = Event(
|
|
name="service_endpoint_bind_occupant_extra",
|
|
key="cb11a150-8049-45c9-8cf3-d5290ffd2e4a",
|
|
request_validator=None, # TODO: Add request validator
|
|
response_validator=None, # TODO: Add response validator
|
|
description="Super Users Bind service to occupant extra endpoint",
|
|
)
|
|
|
|
|
|
def events_list_available_callable(list_options: PaginateOnly):
|
|
"""List available events with pagination and filtering options"""
|
|
list_options = PaginateOnly(**list_options.model_dump())
|
|
service_uu_id = list_options.query.get("service_uu_id__ilike", None)
|
|
if not service_uu_id:
|
|
return {"message": "MSG0003-PARAM-MISSING", "data": list_options.query, "completed": False}
|
|
list_options.query.pop("service_uu_id__ilike", None)
|
|
list_options.query.pop("service_uu_id", None)
|
|
with Events.new_session() as db_session:
|
|
service2events = Service2Events.query.filter(Service2Events.service_uu_id.ilike(f'%{service_uu_id}%'),)
|
|
already_events = [service_to_event.event_id for service_to_event in service2events.all()]
|
|
if list_options.query:
|
|
events_list = Events.query.filter(*Events.convert(list_options.query), Events.id.not_in(already_events))
|
|
else:
|
|
events_list = Events.query.filter(Events.id.not_in(already_events))
|
|
events_response = PostgresResponse(data=events_list)
|
|
pagination = Pagination(data=events_response)
|
|
pagination.change(**list_options.model_dump())
|
|
pagination_result = PaginationResult(data=events_response, pagination=pagination)
|
|
return EndpointResponse(message="MSG0003-LIST", pagination_result=pagination_result).response
|
|
|
|
|
|
EventsListAvailableEvent.event_callable = events_list_available_callable
|
|
|
|
|
|
def events_list_appended_callable(list_options: PaginateOnly):
|
|
"""List appended events with pagination and filtering options"""
|
|
list_options = PaginateOnly(**list_options.model_dump())
|
|
service_uu_id = list_options.query.get("service_uu_id__ilike", None)
|
|
if not service_uu_id:
|
|
return {"message": "MSG0003-PARAM-MISSING", "data": list_options.query, "completed": False}
|
|
list_options.query.pop("service_uu_id__ilike", None)
|
|
list_options.query.pop("service_uu_id", None)
|
|
with Events.new_session() as db_session:
|
|
service2events = Service2Events.query.filter(*Service2Events.convert({"service_uu_id__ilike": service_uu_id}))
|
|
already_events = [service_to_event.event_id for service_to_event in service2events.all()]
|
|
if list_options.query:
|
|
events_list = Events.filter_all(*Events.convert(list_options.query), Events.id.in_(already_events))
|
|
else:
|
|
events_list = Events.filter_all(Events.id.in_(already_events))
|
|
events_response = PostgresResponse(data=events_list)
|
|
pagination = Pagination(data=events_response)
|
|
pagination.change(**list_options.model_dump())
|
|
pagination_result = PaginationResult(data=events_response, pagination=pagination)
|
|
return EndpointResponse(message="MSG0003-LIST", pagination_result=pagination_result).response
|
|
|
|
|
|
EventsListAppendedEvent.event_callable = events_list_appended_callable
|
|
|
|
|
|
def event_register_service_callable(data: Any):
|
|
"""Register event to service"""
|
|
with Events.new_session() as db_session:
|
|
event = Events.query.filter(Events.uu_id == data.event_uu_id).first()
|
|
if not event:
|
|
return EndpointResponse(message="MSG0003-NOT-FOUND", data=data.model_dump(), completed=False).response
|
|
service = Services.query.filter(Services.uu_id == data.service_uu_id).first()
|
|
if not service:
|
|
return {"message": "MSG0003-NOT-FOUND", "data": data.model_dump(), "completed": False,}
|
|
service_to_event = Service2Events.query.filter_by(service_id=service.data.id, event_id=event.data.id).first()
|
|
Service2Events.set_session(db_session)
|
|
if not service_to_event:
|
|
service_to_event = Service2Events.create(
|
|
service_id=service.data.id, service_uu_id=str(service.data.uu_id), event_id=event.data.id,
|
|
event_uu_id=str(event.data.uu_id), is_confirmed=True,
|
|
)
|
|
service_to_event.save()
|
|
return {"message": "MSG0003-REGISTERED", "data": data.model_dump(), "completed": True}
|
|
return {"message": "MSG0003-REGISTER-ERROR", "data": data.model_dump(), "completed": True}
|
|
|
|
|
|
EventRegisterServiceEvent.event_callable = event_register_service_callable
|
|
|
|
|
|
def event_unregister_service_callable(data: Any):
|
|
"""Unregister event from service"""
|
|
with Events.new_session() as db_session:
|
|
event = Events.query.filter(Events.uu_id == data.event_uu_id).first()
|
|
if not event:
|
|
return EndpointResponse(message="MSG0003-NOT-FOUND", data=data.model_dump(), completed=False).response
|
|
service = Services.query.filter(Services.uu_id == data.service_uu_id).first()
|
|
if not service:
|
|
return {"message": "MSG0003-NOT-FOUND", "data": data.model_dump(), "completed": False}
|
|
service_to_event = Service2Events.query.filter(
|
|
Service2Events.service_id == service.data.id, Service2Events.event_id == event.data.id,
|
|
).first()
|
|
if not service_to_event:
|
|
return {"message": "MSG0003-NOT-FOUND", "data": data.model_dump(), "completed": False}
|
|
service_to_event.query.delete()
|
|
return {"message": "MSG0003-UNREGISTER", "data": data.model_dump(), "completed": True}
|
|
|
|
|
|
EventUnRegisterServiceEvent.event_callable = event_unregister_service_callable
|
|
|
|
|
|
def event_bind_employee_extra_callable(data: Any):
|
|
"""Bind event to employee extra"""
|
|
return EndpointResponse(message="MSG0003-BIND-EMP").response
|
|
|
|
|
|
EventBindEmployeeExtraEvent.event_callable = event_bind_employee_extra_callable
|
|
|
|
|
|
def event_bind_occupant_extra_callable(data: Any):
|
|
"""Bind event to occupant extra"""
|
|
return EndpointResponse(message="MSG0003-BIND-OCUP").response
|
|
|
|
|
|
EventBindOccupantExtraEvent.event_callable = event_bind_occupant_extra_callable
|