pymysql插入测试数据模板

import pymysql
import socket
import struct
import random

ip=""
user=""
pwd=""
database=""
index=5000
time=903139688000


def getData():
    data1=[]
    data2=[]
    data3=[]
    for i in range(index,index+5000):
        name="Test%s"%i
        data1.append((i,name,time,time))
        data2.append((name,i))
        data3.append((i,i,i,time,time))
    return tuple(data1),tuple(data2),tuple(data3)

def executeSql(data1,data2,data3):
    dsql1="delete from asset_group where id>=5000 and id<=10000"
    dsql2="delete from asset_group_relation where asset_group_id>=5000 and asset_group_id<=10000"
    sql1="insert into asset_group (id,name,gmt_create,gmt_modified,create_user) values (%s,%s,%s,%s,'31')"
    sql2="update asset set asset_group=%s,asset_status=5 where id=%s"
    sql3="""insert into asset_group_relation (id,asset_group_id,asset_id,gmt_create,gmt_modified,create_user,status)
            values (%s,%s,%s,%s,%s,'31','1')"""


    # 打开数据库连接
    db = pymysql.connect(host=ip,port=3306,user=user,password=pwd,db=database,charset='utf8')
    cursor = db.cursor()
    try:
        cursor.execute(dsql1)
        cursor.execute(dsql2)
        # cursor.executemany(sql1,data1)
        # cursor.executemany(sql2,data2)
        # cursor.executemany(sql3, data3)
        db.commit()
        print("id%s,执行成功"%id)
    except Exception as e:
        print("id%s,执行失败:"%id, e)
        db.rollback()
    db.close()

def main():
    data1,data2,data3=getData()
    executeSql(data1,data2,data3)


if __name__ == '__main__':
    main()
    # print(":".join(["%02x" % x for x in map(lambda x: random.randint(0, 255), range(6))]))