快捷導(dǎo)航

利用Python如何批量修改數(shù)據(jù)庫執(zhí)行Sql文件

更新時間：2018年07月29日 14:33:01 作者：YxYYxY

這篇文章主要給大家介紹了關(guān)于利用Python如何批量修改數(shù)據(jù)庫執(zhí)行Sql文件的相關(guān)資料，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

前言

由于上篇文章中批量修改了文件,有的時候數(shù)據(jù)庫也需要批量修改一下，之前的做法是使用寶塔的phpMyAdmin導(dǎo)出一個已經(jīng)修改好了的sql文件,然后依次去其他數(shù)據(jù)庫里導(dǎo)入,效率不說極低,也算低了,且都是些重復(fù)性的勞動,所以打算用Python來批量執(zhí)行sql

環(huán)境

版本:Python3.6
系統(tǒng):MacOS
IDE:PyCharm
第三方庫:pymysql

Show Code

import pymysql

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

def connectMySQL():
 print('開始連接數(shù)據(jù)庫')
 # 打開數(shù)據(jù)庫連接
 db = pymysql.connect(host,username,password,charset='utf8')

 # 使用 cursor() 方法創(chuàng)建一個游標對象 cursor
 cursor = db.cursor()

 # 使用 execute() 顯示所有數(shù)據(jù)庫
 cursor.execute("SHOW DATABASES")
 print('開始查詢所有數(shù)據(jù)庫')

 # 獲取所有數(shù)據(jù)庫名稱
 data = cursor.fetchall()

 # 開始操作
 for dbb in data:
  dbname = dbb[0]
  print('選中' + dbname + '數(shù)據(jù)庫')
  # 選擇數(shù)據(jù)庫
  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執(zhí)行完成')


if __name__ == "__main__":
 connectMySQL()

解釋代碼

這是用于執(zhí)行sql文件,這里第一句就有個坑,最好設(shè)置encoding='utf-8'否則可能會報錯UnicodeEncodeError: 'latin-1' codec can't encode characters in position 41-44: ordinal not in range(256),當讀取了sql文件后用;分割語句然后用for循環(huán)依次執(zhí)行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執(zhí)行完成')

這一段比較容易理解了,首先是連接數(shù)據(jù)庫,注意還是最好設(shè)置一下charset='utf8' ,因為我要操作多個數(shù)據(jù)庫執(zhí)行sql文件,所以先把數(shù)據(jù)庫名稱全部獲取出來,這里獲取出來的結(jié)果是元組類型,即使for循環(huán)后出來的也是一個元組，所以用[0]取出元組中的值,然后選中數(shù)據(jù)庫執(zhí)行刪表操作(當然不是每個人都要按我這些操作哈,需要執(zhí)行啥操作就自己改SQL語句),表刪完了,然后開始執(zhí)行1.sql文件,執(zhí)行完成后關(guān)閉數(shù)據(jù)庫

def connectMySQL():
 print('開始連接數(shù)據(jù)庫')
 # 打開數(shù)據(jù)庫連接
 db = pymysql.connect(host,username,password,charset='utf8')

 # 使用 cursor() 方法創(chuàng)建一個游標對象 cursor
 cursor = db.cursor()

 # 使用 execute() 顯示所有數(shù)據(jù)庫
 cursor.execute("SHOW DATABASES")
 print('開始查詢所有數(shù)據(jù)庫')

 # 獲取所有數(shù)據(jù)庫名稱
 data = cursor.fetchall()

 # 開始操作
 for dbb in data:
  dbname = dbb[0]
  print('選中' + dbname + '數(shù)據(jù)庫')
  # 選擇數(shù)據(jù)庫
  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作為膠水語言拿來做這些小工具真的舒服！

總結(jié)

以上就是這篇文章的全部內(nèi)容了，希望本文的內(nèi)容對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，如果有疑問大家可以留言交流，謝謝大家對腳本之家的支持。

您可能感興趣的文章: