mtsd/backend/app/models/base.py

"""基础模型类"""
from datetime import datetime
from typing import Optional, Any
from sqlalchemy import Column, String, DateTime, Integer, func
from sqlalchemy.ext.declarative import declared_attr

from app.core.database import Base


class BaseModel(Base):
    """所有模型的基类，包含通用字段"""
    __abstract__ = True

    @declared_attr
    def __tablename__(cls) -> str:
        """自动生成表名"""
        import re
        name = cls.__name__
        # 驼峰转下划线
        return re.sub(r'(?<!^)(?=[A-Z])', '_', name).lower()

    id = Column(String(32), primary_key=True, comment="主键ID")
    create_at = Column(DateTime, default=datetime.now, comment="创建时间")
    create_by = Column(String(32), nullable=True, comment="创建人ID")
    create_name = Column(String(100), nullable=True, comment="创建人姓名")
    update_at = Column(DateTime, default=datetime.now, onupdate=datetime.now, comment="修改时间")
    update_by = Column(String(32), nullable=True, comment="修改人ID")

    def to_dict(self, exclude: list = None) -> dict:
        """转换为字典"""
        if exclude is None:
            exclude = []
        result = {}
        for column in self.__table__.columns:
            if column.name not in exclude:
                value = getattr(self, column.name)
                if isinstance(value, datetime):
                    result[column.name] = value.strftime("%Y-%m-%d %H:%M:%S")
                else:
                    result[column.name] = value
        return result


class SoftDeleteModel(BaseModel):
    """支持软删除的模型基类"""
    __abstract__ = True

    is_del = Column(Integer, default=0, comment="是否删除: 0-否, 1-是")

    @classmethod
    def get_active(cls, db, **kwargs):
        """获取未删除的记录"""
        return db.query(cls).filter(cls.is_del == 0, **kwargs)

    def soft_delete(self, db):
        """软删除"""
        self.is_del = 1
        db.commit()

    def restore(self, db):
        """恢复删除"""
        self.is_del = 0
        db.commit()