0
点赞
收藏
分享

微信扫一扫

python 利用pymysql 直接导入sql文件,生成数据库

晚熟的猫 2022-04-19 阅读 55

原理
根据导出的sql文件进行规律总结,发现一条语句会分成多行,结尾还会有;以结束符。我们可以进行遍历每一行代码,进行处理,如果以 / /n ! 开头。我们抛弃,其余的进行拼接字符串,直到最后形成sql语句如果在一行遍历到;判断最后一个;是否在这一行的最后,如果是,则执行sql

import pymysql
host = "127.0.0.1"
user = "root"
password = ""
db = "ceshi_install"
conn = pymysql.connect(host = host, user = user, password = password, database = db)# 连接数据库
cursor = conn.cursor()# 创建游标
a = 0
try:
  with open('xc_ghavu_com.sql', 'r', encoding = 'utf-8') as f:
      sql = ''
  for i in f:
      if i == '\n'
  or i[0] == '/'
  or i[0] == '-':
      pass
  else :
      a = a + 1# 处理空行
  i = i.strip()
  i = i.strip('\r')
  i = i.strip('\n')# 构造字符串
  sql = sql + i# 判断此行sql语句中是否只含有‘;’, 如果含有, 则进行判断是否在结尾, 反之, 继续拼接
  if ';' in i:
      pot = i.rfind(';')
  if pot + 1 == len(i): cursor.execute(sql) sql = ''
  conn.commit()
  cursor.close()# 关闭游标 conn.close()# 关闭连接
  print("创建成功")
except Exception as e: #失败后的回滚操作
  conn.rollback()
  cursor.close()
  conn.close()
  print("创建失败: ", e)
print('----------', a)
举报

相关推荐

0 条评论