69 lines
2.2 KiB
Python
69 lines
2.2 KiB
Python
# -*- coding: utf-8 -*-
|
||
# @version : 1.0
|
||
# @Create Time : 2021/10/18 22:19
|
||
# @File : db_base.py
|
||
# @IDE : PyCharm
|
||
# @desc : 数据库公共 ORM 模型
|
||
|
||
from datetime import datetime
|
||
from sqlalchemy.orm import Mapped, mapped_column
|
||
from core.database import Base
|
||
from sqlalchemy import DateTime, Integer, func, Boolean, inspect
|
||
|
||
|
||
# 使用命令：alembic init alembic 初始化迁移数据库环境
|
||
# 这时会生成alembic文件夹 和 alembic.ini文件
|
||
class BaseModel(Base):
|
||
"""
|
||
公共 ORM 模型，基表
|
||
"""
|
||
__abstract__ = True
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, comment='主键ID')
|
||
create_datetime: Mapped[datetime] = mapped_column(DateTime, server_default=func.now(), comment='创建时间')
|
||
update_datetime: Mapped[datetime] = mapped_column(
|
||
DateTime,
|
||
server_default=func.now(),
|
||
onupdate=func.now(),
|
||
comment='更新时间'
|
||
)
|
||
delete_datetime: Mapped[datetime | None] = mapped_column(DateTime, nullable=True, comment='删除时间')
|
||
is_delete: Mapped[bool] = mapped_column(Boolean, default=False, comment="是否软删除")
|
||
|
||
@classmethod
|
||
def get_column_attrs(cls) -> list:
|
||
"""
|
||
获取模型中除 relationships 外的所有字段名称
|
||
:return:
|
||
"""
|
||
mapper = inspect(cls)
|
||
|
||
# for attr_name, column_property in mapper.column_attrs.items():
|
||
# # 假设它是单列属性
|
||
# column = column_property.columns[0]
|
||
# # 访问各种属性
|
||
# print(f"属性: {attr_name}")
|
||
# print(f"类型: {column.type}")
|
||
# print(f"默认值: {column.default}")
|
||
# print(f"服务器默认值: {column.server_default}")
|
||
|
||
return mapper.column_attrs.keys()
|
||
|
||
@classmethod
|
||
def get_attrs(cls) -> list:
|
||
"""
|
||
获取模型所有字段名称
|
||
:return:
|
||
"""
|
||
mapper = inspect(cls)
|
||
return mapper.attrs.keys()
|
||
|
||
@classmethod
|
||
def get_relationships_attrs(cls) -> list:
|
||
"""
|
||
获取模型中 relationships 所有字段名称
|
||
:return:
|
||
"""
|
||
mapper = inspect(cls)
|
||
return mapper.relationships.keys()
|