由于上篇文章中批量修改了文件,有的時候數據庫也需要批量修改一下，之前的做法是使用寶塔的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作為膠水語言拿來做這些小工具真的舒服！
文中有錯誤的地方還望指出.
