一次Python操做數據庫和excel過程

師從‘百測’besttest

　　最近牛大溼教了操做數據庫和操做excel，寫了一個小小的腳本，傳入一個表名後，將表中全部數據寫入excel中。

　　使用了pymysql,xlwt，須要自行安裝。

import pymysql,xlwt
def sql_into_excel(table_name):
        db_info = {'user': 'xxx', 'password': '123456','host': '8.8.8.8', 'db': 'asd', 'port': 3306, 'charset': 'utf8','autocommit': True}
        conn = pymysql.connect(**db_info)  # 創建鏈接
        cur = conn.cursor(pymysql.cursors.DictCursor) #獲取表頭的遊標

        sql = 'select * from %s' %(table_name) #若是參數在實際查詢中須要用‘’，在SQL中寫法：'%s'
        cur.execute(sql)
        all = cur.fetchall()

        cur.close()
        conn.close()

        key = []
        for i in all[0]:
                key.append(i)

        book = xlwt.Workbook()
        sheet = book.add_sheet('sheet1')

        for j in range(len(key)):
                sheet.write(0,j,key[j])
                for x in range(len(all)):
                        a = key[j]
                        sheet.write(x+1,j,all[x][a])

        book.save('test.xls')

sql_into_excel('table_name')