用Postgres动态生成SqlAlchemy模型中的字段

from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker

# 创建数据库引擎和会话
engine = create_engine('postgresql://username:password@localhost:5432/database')
Session = sessionmaker(bind=engine)
session = Session()

# 创建基础模型
Base = declarative_base()

# 定义元类
class DynamicModelMeta(type):
    def __new__(cls, name, bases, attrs):
        # 获取数据库中的表结构信息
        table_info = session.execute("SELECT column_name, data_type FROM information_schema.columns WHERE table_name = 'your_table_name'").fetchall()
        
        # 动态生成字段
        for column_name, data_type in table_info:
            if data_type == 'integer':
                column = Column(Integer, primary_key=True)
            elif data_type == 'character varying':
                column = Column(String)
            else:
                # 其他数据类型的处理逻辑
                pass
            
            attrs[column_name] = column
        
        return super().__new__(cls, name, bases, attrs)

# 创建动态模型
class DynamicModel(Base, metaclass=DynamicModelMeta):
    __tablename__ = 'your_table_name'

# 使用动态模型
dynamic_instance = DynamicModel()