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

小編給大家分享一下利用Python怎么批量修改數(shù)據(jù)庫(kù)執(zhí)行Sql文件，希望大家閱讀完這篇文章之后都有所收獲，下面讓我們一起去探討吧！

環(huán)境

• 版本:Python3.6

• 系統(tǒng):MacOS

• IDE:PyCharm

• 第三方庫(kù):pymysql

Show Code

import pymysql

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

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

 # 使用 cursor() 方法創(chuàng)建一個(gè)游標(biāo)對(duì)象 cursor
 cursor = db.cursor()

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

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

 # 開始操作
 for dbb in data:
  dbname = dbb[0]
  print('選中' + dbname + '數(shù)據(jù)庫(kù)')
  # 選擇數(shù)據(jù)庫(kù)
  cursor.execute("use " + dbname)
  # 查看有哪些表
  cursor.execute("show tables")
  table = cursor.fetchall()
  # 如果不是3個(gè)表的就不管
  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文件,這里第一句就有個(gè)坑,最好設(shè)置encoding='utf-8'否則可能會(huì)報(bào)錯(cuò)UnicodeEncodeError: 'latin-1' codec can't encode characters in position 41-44: ordinal not in range(256),當(dāng)讀取了sql文件后用;分割語(yǔ)句然后用for循環(huán)依次執(zhí)行sql語(yǔ)句

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ù)庫(kù),注意還是最好設(shè)置一下charset='utf8' ,因?yàn)槲乙僮鞫鄠€(gè)數(shù)據(jù)庫(kù)執(zhí)行sql文件,所以先把數(shù)據(jù)庫(kù)名稱全部獲取出來(lái),這里獲取出來(lái)的結(jié)果是元組類型,即使for循環(huán)后出來(lái)的也是一個(gè)元組，所以用[0]取出元組中的值,然后選中數(shù)據(jù)庫(kù)執(zhí)行刪表操作(當(dāng)然不是每個(gè)人都要按我這些操作哈,需要執(zhí)行啥操作就自己改SQL語(yǔ)句),表刪完了,然后開始執(zhí)行1.sql文件,執(zhí)行完成后關(guān)閉數(shù)據(jù)庫(kù)

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

 # 使用 cursor() 方法創(chuàng)建一個(gè)游標(biāo)對(duì)象 cursor
 cursor = db.cursor()

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

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

 # 開始操作
 for dbb in data:
  dbname = dbb[0]
  print('選中' + dbname + '數(shù)據(jù)庫(kù)')
  # 選擇數(shù)據(jù)庫(kù)
  cursor.execute("use " + dbname)
  # 查看有哪些表
  cursor.execute("show tables")
  table = cursor.fetchall()
  # 如果不是3個(gè)表的就不管
  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篇文章的代碼從思路、編寫、到測(cè)試，用了一下午吧！我對(duì)Python也不是很熟悉，所以中間也踩了些坑，但確實(shí)能看出來(lái)，Python作為膠水語(yǔ)言拿來(lái)做這些小工具真的舒服！

看完了這篇文章，相信你對(duì)“利用Python怎么批量修改數(shù)據(jù)庫(kù)執(zhí)行Sql文件”有了一定的了解，如果想了解更多相關(guān)知識(shí)，歡迎關(guān)注億速云行業(yè)資訊頻道，感謝各位的閱讀！