Python SqlAlchemy动态添加数据表字段

痛定思痛。 2022-06-05 09:38 452阅读 0赞

--------------------

SqlAlchemy版本：1.2  
操作系统：Windows  
Python：3.5  
**欢迎加入学习交流QQ群：657341423**

--------------------

我们知道使用SqlAlchemy创建类来映射数据表，类属性等于数据库字段，但有时候要在我们创建表的时候，不确定数据表字段数量，遇到这种情况，应如何解决？

先看常规用法

from sqlalchemy import create_engine,Column,String,Integer
    class Mybase(Base):
        #表名
        __tablename__ ='mycars'
        #字段，属性
        myid=Column(String(50), primary_key=True)
        price=Column(String(50))

上述代码是创建mycars数据表，字段分别为myid和price，字段数量是固定，那么要实现动态创建数量不明确的字段，代码如下：

from sqlalchemy import *
    from sqlalchemy.orm import sessionmaker
    from sqlalchemy.ext.declarative import declarative_base
    engine = create_engine('mysql+pymysql://root:1234@localhost/test?charset=utf8', echo=False)
    # engine = create_engine('sqlite:MyDB.sqlite3', echo=False)
    DBSession = sessionmaker(bind=engine)
    session = DBSession()
    Base = declarative_base()
    #定义类
    class table_class(Base):
        __tablename__ = 'aaa'
        id=Column(Integer,primary_key=True)
        
    # 动态添加字段
    for i in range(3):
        setattr(table_class,'Col'+str(i),(Column('Col'+str(i), String(50),comment='Col'+str(i))))
    Base.metadata.create_all(engine)
    
    # 添加数据
    dt=table_class(Col1='aaa',Col2="aaa")
    session.add(dt)
    session.commit()

上述代码可看出，通过使用setattr（）来实现动态添加字段，而且字段名可根据实际命名，comment是字段注释，这个只有SqlAlchemy 1.2版本才有的功能  
**（SqlAlchemy 1.2 安装 ：pip install --pre sqlalchemy）**  
运行结果：  
![这里写图片描述][SouthEast]

当然，在插入数据和查询数据的时候，会出现无法确定代码中对应字段的属性。可以使用SqlAlchemy执行sql实现插入

[SouthEast]: /images/20220605/2d553edeb50b4a3080a3d5bbb617ab4c.png