sqlalchemy批量添加数据-数据源是json(小算法)

时间:2023-03-10 03:16:12
sqlalchemy批量添加数据-数据源是json(小算法)

需求:

想要写1个增加case的接口

问题:

sqlalchemy添加case的方式,只能是1条数据1条数据的插入,像这样:

   ro2 = Role(name='user')
db.session.add_all([ro1])
db.session.commit()

我希望的效果是这样的:

我传入一个字典{'name':'lisa'},然后自动转化成name=‘lisa’这样的格式

接口的参数是一组dict类型的数据,每一组数据即视为1条case,通过程序将json传过来的数据对号入座

 a=[{'name':'lisa'},{'name':'bob'}]

 case=[]
for i in a: #i={'name':'lisa'}
name=i['name'] #name='lisa' obj = Role(name=name)#obj=<Role 'lisa'>
case.append(obj)
print(case)#[<Role 'lisa'>, <Role 'bob'>]
db.session.add_all(case)
db.session.commit()

完整的代码如下:

 #!/usr/bin/python
# -*- coding: utf-8 -*- from flask import Flask
from flask_sqlalchemy import SQLAlchemy app = Flask(__name__) # 设置连接数据库的URL
app.config['SQLALCHEMY_DATABASE_URI'] ='mysql+mysqlconnector://root:admin123456@10.1.71.32:3306/test' # 设置每次请求结束后会自动提交数据库的改动
app.config['SQLALCHEMY_COMMIT_ON_TEARDOWN'] = True
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = True # 查询时显示原始SQL语句
app.config['SQLALCHEMY_ECHO'] = False
db = SQLAlchemy(app) class Role(db.Model):
__tablename__ = 'roles'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(64)) def __repr__(self):
return '<Role %r>' % self.name #数据源-要批量添加的case数据 list套dict
a=[{'name':'lisa'},{'name':'bob'}] case=[]
for i in a: #i={'name':'lisa'}
name=i['name'] #name='lisa' obj = Role(name=name)#obj=<Role 'lisa'>
case.append(obj)
print(case)#[<Role 'lisa'>, <Role 'bob'>]
db.session.add_all(case)
db.session.commit()

结果:可以循环插入多条数据

sqlalchemy批量添加数据-数据源是json(小算法)