42 lines
977 B
Python
42 lines
977 B
Python
from typing import Any
|
|
|
|
from sqlalchemy.orm import as_declarative, declared_attr
|
|
|
|
|
|
@as_declarative()
|
|
class Base:
|
|
id: Any
|
|
__name__: str
|
|
|
|
def create(self, db):
|
|
db.add(self)
|
|
db.commit()
|
|
db.refresh(self)
|
|
return self
|
|
|
|
@classmethod
|
|
def get(cls, db, rid: int):
|
|
return db.query(cls).filter(cls.id == rid).first()
|
|
|
|
def update(self, db, 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)
|
|
db.commit()
|
|
|
|
@classmethod
|
|
def delete(cls, db, rid):
|
|
db.query(cls).filter(cls.id == rid).delete()
|
|
db.commit()
|
|
|
|
@classmethod
|
|
def list(cls, db):
|
|
return db.query(cls).all()
|
|
|
|
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()
|