1. 程式人生 > 其它 >python 利用pymysql 匯入sql檔案,生成資料庫

python 利用pymysql 匯入sql檔案,生成資料庫

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)