数据库插入大量数据

目的:解决手动插入用户麻烦问题

from natsort import natsorted
import pymysql
import re
import time

"""
    1、根据sql找到数据库最大Id数,每插一次Id加1
    2、根据sql找到数据库目前最大默认用户值,每插一次默认用户值加1
    注:插入sql语句中其他不关键字段需要调整的话可直接改插入sql内容
"""

# 配置要插多少个默认用户
min_nmber = 10

#配置数据库信息
ip='127.0.0.1'
user='root'
psw='root'
port=3306

try:
    db = pymysql.connect(host=ip, port=port,
                         user=user, passwd=psw, db='management', charset='utf8', connect_timeout=10)
    cursor = db.cursor()
    print("数据库连接成功")
except:
    print("数据库连接失败")

cursor.execute("""select max(id) from user;""")      #查询user表最大ID
max_id = cursor.fetchone()[0]
print("目前user表最大Id:{}".format(max_id))
cursor.execute("""SELECT username from `user` WHERE username LIKE 'compile_user%'""")      #查询user表默认用户
manage_user = cursor.fetchall()                   #拿到所有用户
cm_mr = str(manage_user).split('compile_user')    #拿到所有用户的数字值
number_user = re.findall(r'\d+', str(cm_mr))      #正则获取具体数字以列表保存
max_user = natsorted(number_user, reverse=True)[0]      #拿到最大值
print("目前最大默认用户:compile_user{}".format(max_user))

for i in range(min_nmber):       #要插多少用户就执行几次该循环下语句
    max_id = max_id + 1           #每次执行user表最大ID+1
    max_user = int(max_user) + 1    #每次执行默认用户数字值+1
    load_time = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time()))  #获取系统当前时间
    # 插入sql
    sql_cr = """INSERT INTO `user` VALUES ('{0}', 'compile_user{1}', 'compile_user{2}', null, null, null, '0', '0', null, '2147483647', '{3}', null, '0', null, '0');""".format(
        max_id, max_user, max_user,load_time)
    try:
        cursor.execute(sql_cr)
        db.commit()
        print("Id:{0}  username:compile_user{1}".format(max_id, max_user) + "  插入成功")
    except:
        db.rollback()
        print("Id:{0}  username:compile_user{1}".format(max_id, max_user) + "  插入失败")

db.close()

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值