ORM 模型基类新增获取字段列表方法

kinit-api/db/db_base.py

@@ -8,7 +8,7 @@
 from datetime import datetime
 from sqlalchemy.orm import Mapped, mapped_column
 from core.database import Base
-from sqlalchemy import DateTime, Integer, func, Boolean
+from sqlalchemy import DateTime, Integer, func, Boolean, inspect
 
 
 # 使用命令：alembic init alembic 初始化迁移数据库环境
@@ -29,3 +29,40 @@ class BaseModel(Base):
     )
     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()