89 lines
3.1 KiB
Python
89 lines
3.1 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
|
||
"""
|
||
基础模型模块
|
||
包含所有模型共用的字段和方法
|
||
"""
|
||
|
||
import datetime
|
||
from sqlalchemy import Column, Integer, Boolean, desc
|
||
from sqlalchemy.dialects.mysql import DATETIME
|
||
from config.database_config import Base
|
||
|
||
class BaseModel(Base):
|
||
"""
|
||
基础模型类
|
||
包含所有模型共用的字段和方法
|
||
"""
|
||
__abstract__ = True # 声明为抽象类，不会创建表
|
||
|
||
id = Column(Integer, primary_key=True, autoincrement=True, comment='主键ID')
|
||
created_at = Column(DATETIME(fsp=6), default=datetime.datetime.now, comment='创建时间')
|
||
updated_at = Column(DATETIME(fsp=6), default=datetime.datetime.now, onupdate=datetime.datetime.now, comment='更新时间')
|
||
is_deleted = Column(Boolean, default=False, comment='是否删除（软删除标记）')
|
||
|
||
def to_dict(self):
|
||
"""
|
||
将模型转换为字典
|
||
用于API响应
|
||
"""
|
||
result = {}
|
||
for column in self.__table__.columns:
|
||
value = getattr(self, column.name)
|
||
# 处理日期时间类型
|
||
if isinstance(value, datetime.datetime):
|
||
value = value.strftime('%Y-%m-%d %H:%M:%S')
|
||
result[column.name] = value
|
||
return result
|
||
|
||
def to_json(self, fields=None, exclude=None, timestamp_format='ms'):
|
||
"""
|
||
将模型转换为JSON友好的字典
|
||
|
||
Args:
|
||
fields (list, optional): 需要包含的字段列表，为None则包含所有字段
|
||
exclude (list, optional): 需要排除的字段列表
|
||
timestamp_format (str, optional): 时间戳格式，可选值：'ms'（毫秒时间戳）, 'iso'（ISO格式）, 'str'（字符串格式）
|
||
|
||
Returns:
|
||
dict: 包含指定字段的字典
|
||
"""
|
||
result = {}
|
||
|
||
# 获取所有列名
|
||
columns = [column.name for column in self.__table__.columns]
|
||
|
||
# 如果指定了fields，则只包含这些字段
|
||
if fields:
|
||
columns = [col for col in columns if col in fields]
|
||
|
||
# 排除指定的字段
|
||
if exclude:
|
||
columns = [col for col in columns if col not in exclude]
|
||
|
||
# 获取字段值
|
||
for column in columns:
|
||
value = getattr(self, column)
|
||
|
||
# 处理日期时间类型
|
||
if isinstance(value, datetime.datetime):
|
||
if timestamp_format == 'ms':
|
||
# 转换为毫秒时间戳
|
||
value = int(value.timestamp() * 1000)
|
||
elif timestamp_format == 'iso':
|
||
# 转换为ISO格式
|
||
value = value.isoformat()
|
||
else:
|
||
# 转换为字符串格式
|
||
value = value.strftime('%Y-%m-%d %H:%M:%S')
|
||
|
||
# 处理枚举类型
|
||
elif hasattr(value, 'name') and hasattr(value, 'value'):
|
||
# 如果是枚举类型，返回其值
|
||
value = value.value
|
||
|
||
result[column] = value
|
||
|
||
return result
|