python 利用pymysql 匯入sql檔案,生成資料庫
阿新 • • 發佈:2022-04-17
pip install pymysql
import pymysql host = "127.0.0.1" user = "root" password = "" db = "ceshi_install" conn = pymysql.connect(host=host, user=user, password=password, database=db) # 連線資料庫 cursor = conn.cursor() # 建立遊標 a = 0 try: with open('xc_ghavu_com.sql', 'r', encoding='utf-8') as f: sql = '' for i in f: if i == '\n' or i[0] == '/' or i[0] == '-': pass else: a = a + 1 # 處理空行 i = i.strip() i = i.strip('\r') i = i.strip('\n') # 構造字串 sql = sql + i # 判斷此行sql語句中是否只含有 ‘;’ ,如果含有,則進行判斷是否在結尾,反之,繼續拼接 if ';' in i: pot = i.rfind(';') if pot + 1 == len(i): cursor.execute(sql) sql = '' conn.commit() cursor.close() # 關閉遊標 conn.close() # 關閉連線 print("建立成功") except Exception as e: # 失敗後的回滾操作 conn.rollback() cursor.close() conn.close() print("建立失敗: ", e) print('----------', a)