1. 程式人生 > >Python 將資料庫資料儲存到txt

Python 將資料庫資料儲存到txt

這幾天剛剛寫了一個python的小程式,目的是從資料庫拉取資料然後儲存到本地檔案的txt中。

我這裡用的是python2.7 一開始裝的是python3,但是mysqldb並不能用,所以下了一個python2.7。

關鍵的程式碼

#coding=utf-8
import MySQLdb as db
import sys
import codecs
#你會發現我們下面的程式碼出現了中文,如果不加入這句,那麼系統就報錯
reload(sys)
sys.setdefaultencoding('utf-8')

def connectionDB():
    conn=db.connect(host='localhost',user='root',db="test",passwd=123456,port=3306,charset="utf8")
    cur = conn.cursor() 

    #在開始的時候 先在e:\get_key\中生成一個 農學_獸醫學的txt檔案
    defau_subject_first = '醫學'
    defau_subject_second = '臨床醫學'
    path = 'E:\\get_key2\\' + defau_subject_first + '_' + defau_subject_second + '.txt' #注意我們這裡路徑是e:\\get_key2\\ 主要是\是一個特殊符號,我們需要轉義
    paper_keywords1 = codecs.open(path.decode('utf-8'), 'wb',encoding="utf-8") #在開啟檔案的時候,為了防止出現中文亂碼的問題,我用codecs.open去開啟檔案

    #把結果先放到sql裡面
    sql = "SELECT name_chinese,subject_first ,subject_second  FROM periodical"
    cur.execute(sql)
    results = cur.fetchall()
    #把結果寫到txt裡面
    for result in results:
	paper_keywords1.write(result[0])
    #關閉資源
    cur.close()
    conn.close()
    paper_keywords1.close()

connectionDB()
print(end)