宜配屋

Python SqlAlchemy动态添加数据表字段实例解析

yipeiwu_com6年前 (2020-03-06)Python基础

本文研究的主要是Python SqlAlchemy动态添加数据表字段，具体如下。

我们知道使用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）

运行结果：