利用Python如何批量修改数据库执行Sql文件


Posted in Python onJuly 29, 2018

前言

由于上篇文章中批量修改了文件,有的时候数据库也需要批量修改一下,之前的做法是使用宝塔的phpMyAdmin导出一个已经修改好了的sql文件,然后依次去其他数据库里导入,效率不说极低,也算低了,且都是些重复性的劳动,所以打算用Python来批量执行sql

环境

  • 版本:Python3.6
  • 系统:MacOS
  • IDE:PyCharm
  • 第三方库:pymysql

Show Code

import pymysql

host = 'xxx.65.9.191'
username = 'root'
password = 'root'

def connectMySQL():
 print('开始连接数据库')
 # 打开数据库连接
 db = pymysql.connect(host,username,password,charset='utf8')

 # 使用 cursor() 方法创建一个游标对象 cursor
 cursor = db.cursor()

 # 使用 execute() 显示所有数据库
 cursor.execute("SHOW DATABASES")
 print('开始查询所有数据库')

 # 获取所有数据库名称
 data = cursor.fetchall()

 # 开始操作
 for dbb in data:
  dbname = dbb[0]
  print('选中' + dbname + '数据库')
  # 选择数据库
  cursor.execute("use " + dbname)
  # 查看有哪些表
  cursor.execute("show tables")
  table = cursor.fetchall()
  # 如果不是3个表的就不管
  if len(table) != 3:
   continue
  for tb in table:
   tbname = tb[0]
   print('开始删除'+tbname+'表')
   # 删除所有的表
   cursor.execute("DROP TABLE " + tbname)
  executeScriptsFromFile('1.sql', cursor)
 db.close()


def executeScriptsFromFile(filename,cursor):
 fd = open(filename, 'r',encoding='utf-8')
 sqlFile = fd.read()
 fd.close()
 sqlCommands = sqlFile.split(';')

 for command in sqlCommands:
  try:
   cursor.execute(command)
  except Exception as msg:
   print(msg)

 print('sql执行完成')


if __name__ == "__main__":
 connectMySQL()

解释代码

这是用于执行sql文件,这里第一句就有个坑,最好设置encoding='utf-8'否则可能会报错UnicodeEncodeError: 'latin-1' codec can't encode characters in position 41-44: ordinal not in range(256),当读取了sql文件后用;分割语句然后用for循环依次执行sql语句

def executeScriptsFromFile(filename,cursor):
 fd = open(filename, 'r',encoding='utf-8')
 sqlFile = fd.read()
 fd.close()
 sqlCommands = sqlFile.split(';')

 for command in sqlCommands:
  try:
   cursor.execute(command)
  except Exception as msg:
   print(msg)
 print('sql执行完成')

这一段比较容易理解了,首先是连接数据库,注意还是最好设置一下charset='utf8' ,因为我要操作多个数据库执行sql文件,所以先把数据库名称全部获取出来,这里获取出来的结果是元组类型,即使for循环后出来的也是一个元组,所以用[0]取出元组中的值,然后选中数据库执行删表操作(当然不是每个人都要按我这些操作哈,需要执行啥操作就自己改SQL语句),表删完了,然后开始执行1.sql文件,执行完成后关闭数据库

def connectMySQL():
 print('开始连接数据库')
 # 打开数据库连接
 db = pymysql.connect(host,username,password,charset='utf8')

 # 使用 cursor() 方法创建一个游标对象 cursor
 cursor = db.cursor()

 # 使用 execute() 显示所有数据库
 cursor.execute("SHOW DATABASES")
 print('开始查询所有数据库')

 # 获取所有数据库名称
 data = cursor.fetchall()

 # 开始操作
 for dbb in data:
  dbname = dbb[0]
  print('选中' + dbname + '数据库')
  # 选择数据库
  cursor.execute("use " + dbname)
  # 查看有哪些表
  cursor.execute("show tables")
  table = cursor.fetchall()
  # 如果不是3个表的就不管
  if len(table) != 3:
   continue
  for tb in table:
   tbname = tb[0]
   print('开始删除'+tbname+'表')
   # 删除所有的表
   cursor.execute("DROP TABLE " + tbname)
  executeScriptsFromFile('1.sql', cursor)
 db.close()

这2篇文章的代码从思路、编写、到测试,用了一下午吧!我对Python也不是很熟悉,所以中间也踩了些坑,但确实能看出来,Python作为胶水语言拿来做这些小工具真的舒服!

总结

以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作具有一定的参考学习价值,如果有疑问大家可以留言交流,谢谢大家对三水点靠木的支持。

Python 相关文章推荐
python端口扫描系统实现方法
Nov 19 Python
Python中datetime常用时间处理方法
Jun 15 Python
Python读取一个目录下所有目录和文件的方法
Jul 15 Python
Python tornado队列示例-一个并发web爬虫代码分享
Jan 09 Python
python 读取DICOM头文件的实例
May 07 Python
python 字典中取值的两种方法小结
Aug 02 Python
Python编程深度学习计算库之numpy
Dec 28 Python
python3 字符串/列表/元组(str/list/tuple)相互转换方法及join()函数的使用
Apr 03 Python
numpy数组之存取文件的实现示例
May 24 Python
Pandas_cum累积计算和rolling滚动计算的用法详解
Jul 04 Python
python 经典数字滤波实例
Dec 16 Python
tensorflow之获取tensor的shape作为max_pool的ksize实例
Jan 04 Python
利用Python如何批量更新服务器文件
Jul 29 #Python
python高阶爬虫实战分析
Jul 29 #Python
python3.5基于TCP实现文件传输
Mar 20 #Python
python3基于TCP实现CS架构文件传输
Jul 28 #Python
python cs架构实现简单文件传输
Mar 20 #Python
Tornado Web Server框架编写简易Python服务器
Jul 28 #Python
python使用tornado实现登录和登出
Jul 28 #Python
You might like
第1次亲密接触PHP5(2)
2006/10/09 PHP
JS中attr和prop属性的区别以及优先选择示例介绍
2014/06/30 Javascript
javascript将DOM节点添加到文档的方法实例分析
2015/08/04 Javascript
详解javascript传统方法实现异步校验
2016/01/22 Javascript
jQuery实现的图片轮播效果完整示例
2016/09/12 Javascript
微信小程序 安全包括(框架、功能模块、账户使用)详解
2017/01/16 Javascript
JS+html5 canvas实现的简单绘制折线图效果示例
2017/03/13 Javascript
MUI实现上拉加载和下拉刷新效果
2017/06/30 Javascript
vue.js使用3DES加密的方法示例
2018/05/18 Javascript
一篇文章弄懂javascript中的执行栈与执行上下文
2019/08/09 Javascript
Vue 解决多级动态面包屑导航的问题
2019/11/04 Javascript
vue 动态表单开发方法案例详解
2019/12/02 Javascript
Antd下拉选择,自动匹配功能的实现
2020/10/24 Javascript
[42:25]2018DOTA2亚洲邀请赛 4.5 淘汰赛 LGD vs Liquid 第三场
2018/04/06 DOTA
使用Python编写一个模仿CPU工作的程序
2015/04/16 Python
Apache如何部署django项目
2017/05/21 Python
python多进程和多线程究竟谁更快(详解)
2017/05/29 Python
Python爬虫信息输入及页面的切换方法
2018/05/11 Python
python3模块smtplib实现发送邮件功能
2018/05/22 Python
浅析python中numpy包中的argsort函数的使用
2018/08/30 Python
Python使用logging模块实现打印log到指定文件的方法
2018/09/05 Python
深入理解Python中的 __new__ 和 __init__及区别介绍
2018/09/17 Python
在django view中给form传入参数的例子
2019/07/19 Python
Python 脚本实现淘宝准点秒杀功能
2019/11/13 Python
HTML5实现锚点时请使用id取代name
2013/09/06 HTML / CSS
阿联酋团购网站:Groupon阿联酋
2016/10/14 全球购物
StubHub澳大利亚:购买或出售您的门票
2019/08/01 全球购物
阿迪达斯越南官网:adidas越南
2020/07/19 全球购物
亿阳信通股份有限公司C#笔试题
2016/12/06 面试题
laravel使用redis队列实例讲解
2021/03/23 PHP
会计专业毕业生求职信分享
2014/01/03 职场文书
秋季运动会表扬稿
2014/01/16 职场文书
个人师德师风自我剖析材料
2014/09/29 职场文书
2014年小学安全工作总结
2014/12/04 职场文书
地陪导游欢迎词
2015/01/26 职场文书
2015年乡镇民政工作总结
2015/05/13 职场文书