52 lines
1.2 KiB
Python
52 lines
1.2 KiB
Python
from typing import Any, Self, List
|
|
|
|
from sqlalchemy.orm import as_declarative, declared_attr, Session
|
|
|
|
from app.db import db_update, db_query
|
|
|
|
|
|
@as_declarative()
|
|
class Base:
|
|
id: Any
|
|
__name__: str
|
|
|
|
@db_update
|
|
def create(self, db: Session) -> Self:
|
|
db.add(self)
|
|
db.refresh(self)
|
|
return self
|
|
|
|
@classmethod
|
|
@db_query
|
|
def get(cls, db: Session, rid: int) -> Self:
|
|
return db.query(cls).filter(cls.id == rid).first()
|
|
|
|
@db_update
|
|
def update(self, db: Session, payload: dict):
|
|
payload = {k: v for k, v in payload.items() if v is not None}
|
|
for key, value in payload.items():
|
|
setattr(self, key, value)
|
|
|
|
@classmethod
|
|
@db_update
|
|
def delete(cls, db: Session, rid):
|
|
db.query(cls).filter(cls.id == rid).delete()
|
|
|
|
@classmethod
|
|
@db_update
|
|
def truncate(cls, db: Session):
|
|
db.query(cls).delete()
|
|
|
|
@classmethod
|
|
@db_query
|
|
def list(cls, db: Session) -> List[Self]:
|
|
result = db.query(cls).all()
|
|
return list(result)
|
|
|
|
def to_dict(self):
|
|
return {c.name: getattr(self, c.name, None) for c in self.__table__.columns}
|
|
|
|
@declared_attr
|
|
def __tablename__(self) -> str:
|
|
return self.__name__.lower()
|