from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import DeclarativeMeta
class BaseController:
"""
Base BaseController, implement base CRUD sqlalchemy operations
"""
model_cls = None
base_filter = ()
def get_by_id(
cls,
model_id: Union[int, str],
query_field: Union[List, tuple] = None,
to_dict: bool = False,
id_field: str = None,
) -> Model or None:
session.query(cls.model_cls).filter
class CustomDeclarativeMeta(DeclarativeMeta):
def __init__(self, *args, **kwargs):
super(CustomDeclarativeMeta, self).__init__(*args, **kwargs)
if hasattr(self, 'objects'):
self.objects.model_cls = self
self.objects.base_filter = (self.objects.model_cls.is_delete == 0,)
Base = declarative_base(metaclass=CustomDeclarativeMeta)
class BaseModel(Base):
"""基类表模板"""
__abstract__ = True
objects = BaseController()
is_delete = Column(Boolean, nullable=False, default=False, comment="是否已删除")
class User(BaseModel):
nickname = Column("nickname", String(255), index=True, comment="昵称")
User.objects.get_by_id(1)