利用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 StringIO模块实现在内存缓冲区中读写数据
Apr 08 Python
python中Genarator函数用法分析
Apr 08 Python
python单例模式实例分析
Apr 08 Python
Python入门_浅谈逻辑判断与运算符
May 16 Python
对Pycharm创建py文件时自定义头部模板的方法详解
Feb 12 Python
python解析yaml文件过程详解
Aug 30 Python
python常用数据重复项处理方法
Nov 22 Python
Python爬虫库BeautifulSoup的介绍与简单使用实例
Jan 25 Python
pandas.DataFrame.drop_duplicates 用法介绍
Jul 06 Python
如何在windows下安装配置python工具Ulipad
Oct 27 Python
python中复数的共轭复数知识点总结
Dec 06 Python
python使用tkinter实现透明窗体上绘制随机出现的小球(实例代码)
May 17 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
php代码优化及php相关问题总结
2006/10/09 PHP
PHP的数组中提高元素查找与元素去重的效率的技巧解析
2016/03/03 PHP
php面向对象编程self和static的区别
2016/05/08 PHP
PHP的时间戳与具体时间转化的简单实现
2016/06/13 PHP
快速保存网页中所有图片的方法
2006/06/23 Javascript
SUN的《AJAX与J2EE》全文译了
2007/02/23 Javascript
js 禁止选择功能实现代码(兼容IE/Firefox)
2010/04/23 Javascript
原生js实现半透明遮罩层效果具体代码
2013/06/06 Javascript
js写的评论分页(还不错)
2013/12/23 Javascript
PHP+jQuery+Ajax+Mysql如何实现发表心情功能
2015/08/06 Javascript
基于jquery实现的树形菜单效果代码
2015/09/06 Javascript
使用jQuery判断Div是否在可视区域的方法 判断div是否可见
2016/02/17 Javascript
复杂的javascript窗口分帧解析
2016/02/19 Javascript
js 连续赋值的简单实现
2016/06/13 Javascript
详解js的事件处理函数和动态创建html标记方法
2016/12/16 Javascript
微信小程序去哪里找 小程序到底如何使用(附小程序名单)
2017/01/09 Javascript
js时间戳格式化成日期格式的多种方法介绍
2017/02/16 Javascript
Angular2学习教程之ng中变更检测问题详解
2017/05/28 Javascript
Vue的状态管理vuex使用方法详解
2020/02/05 Javascript
浅析Vue 中的 render 函数
2020/02/28 Javascript
python中的闭包用法实例详解
2015/05/05 Python
python中类变量与成员变量的使用注意点总结
2017/04/29 Python
利用python操作SQLite数据库及文件操作详解
2017/09/22 Python
Python实现拷贝/删除文件夹的方法详解
2018/08/29 Python
django的分页器Paginator 从django中导入类
2019/07/25 Python
python数据爬下来保存的位置
2020/02/17 Python
使用Pycharm分段执行代码
2020/04/15 Python
在C语言中"指针和数组等价"到底是什么意思?
2014/03/24 面试题
信用社实习人员自我鉴定
2013/09/20 职场文书
九月份红领巾广播稿
2014/01/22 职场文书
《长相思》听课反思
2014/04/10 职场文书
《放小鸟》教学反思
2014/04/20 职场文书
领导班子整改措施
2014/10/24 职场文书
pandas中对文本类型数据的处理小结
2021/11/01 Python
python中pymysql包操作数据库方法
2022/04/19 Python
Ubuntu Server 安装Tomcat并配置systemctl
2022/04/28 Servers