目的:解决手动插入用户麻烦问题
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()