mirror of
https://github.com/Balshgit/gpt_chat_bot.git
synced 2025-09-11 22:30:41 +03:00
23031b0777
* update chat_microservice * reformat logger_conf * add database * add service and repository logic * fix constants gpt base url * add models endpoints
35 lines
857 B
Python
35 lines
857 B
Python
from sqlalchemy import Table, inspect
|
|
from sqlalchemy.orm import as_declarative, declared_attr
|
|
|
|
from infra.database.meta import meta
|
|
|
|
|
|
@as_declarative(metadata=meta)
|
|
class Base:
|
|
"""
|
|
Base for all models.
|
|
|
|
It has some type definitions to
|
|
enhance autocompletion.
|
|
"""
|
|
|
|
# Generate __tablename__ automatically
|
|
@declared_attr
|
|
def __tablename__(self) -> str:
|
|
return self.__name__.lower()
|
|
|
|
__table__: Table
|
|
|
|
@classmethod
|
|
def get_real_column_name(cls, attr_name: str) -> str:
|
|
return getattr(inspect(cls).c, attr_name).name # type: ignore
|
|
|
|
def __str__(self) -> str:
|
|
return self.__repr__()
|
|
|
|
def __repr__(self) -> str:
|
|
try:
|
|
return f"{self.__class__.__name__}(id={self.id})" # type: ignore[attr-defined]
|
|
except AttributeError:
|
|
return super().__repr__()
|