production-evyos-systems-an.../ServicesApi/Schemas/others/enums.py

112 lines
3.9 KiB
Python

from Schemas.base_imports import (
CrudCollection,
UUID,
String,
text,
mapped_column,
Mapped,
)
class ApiEnumDropdown(CrudCollection):
__tablename__ = "api_enum_dropdown"
__exclude__fields__ = ["enum_class"]
__language_model__ = None
id: Mapped[int] = mapped_column(primary_key=True)
uu_id: Mapped[str] = mapped_column(
UUID, server_default=text("gen_random_uuid()"), index=True, unique=True
)
lang: Mapped[str] = mapped_column(String, nullable=False, server_default="tr", comment="Language")
enum_class: Mapped[str] = mapped_column(
String, nullable=False, comment="Enum Constant Name"
)
key: Mapped[str] = mapped_column(String, nullable=False, comment="Enum Key")
value: Mapped[str] = mapped_column(String, nullable=False, comment="Enum Value")
description: Mapped[str] = mapped_column(String, nullable=True)
__table_args__ = ({"comment": "Enum objets that are linked to tables"},)
@classmethod
def get_by_uuid(cls, uuid: str):
with cls.new_session() as db_session:
return cls.filter_by_one(uu_id=str(uuid), db=db_session).data
@classmethod
def get_debit_search(cls, search_debit: str = None, search_uu_id: str = None):
with cls.new_session() as db_session:
if search_uu_id:
if search := cls.filter_one_system(
cls.enum_class.in_(["DebitTypes"]),
cls.uu_id == search_uu_id,
db=db_session,
).data:
return search
elif search_debit:
if search := cls.filter_one(
cls.enum_class.in_(["DebitTypes"]),
cls.key == search_debit,
db=db_session,
).data:
return search
return cls.filter_all_system(
cls.enum_class.in_(["DebitTypes"]), db=db_session
).data
@classmethod
def get_due_types(cls):
with cls.new_session() as db_session:
if due_list := cls.filter_all_system(
cls.enum_class == "BuildDuesTypes",
cls.key.in_(["BDT-A", "BDT-D"]),
db=db_session,
).data:
return [due.uu_id.__str__() for due in due_list]
# raise HTTPException(
# status_code=404,
# detail="No dues types found",
# )
@classmethod
def due_type_search(cls, search_management: str = None, search_uu_id: str = None):
with cls.new_session() as db_session:
if search_uu_id:
if search := cls.filter_one_system(
cls.enum_class.in_(["BuildDuesTypes"]),
cls.uu_id == search_uu_id,
db=db_session,
).data:
return search
elif search_management:
if search := cls.filter_one_system(
cls.enum_class.in_(["BuildDuesTypes"]),
cls.key == search_management,
db=db_session,
).data:
return search
return cls.filter_all_system(
cls.enum_class.in_(["BuildDuesTypes"]), db=db_session
).data
def get_enum_dict(self):
return {
"uu_id": str(self.uu_id),
"enum_class": self.enum_class,
"key": self.key,
"value": self.value,
"description": self.description,
}
@classmethod
def uuid_of_enum(cls, enum_class: str, key: str):
with cls.new_session() as db_session:
return str(
getattr(
cls.filter_one_system(
cls.enum_class == enum_class, cls.key == key, db=db_session
).data,
"uu_id",
None,
)
)