python3操作excel和數據庫

1. 讀取excel數據並寫入數據庫

'''
Created on 2019年
讀取excel數據並寫入數據庫
@author: Root
'''
import xlrd
from DBUtils import DBUtils

# 獲取excel表格
def get_sheet():
    #定義測試用例的文檔位置
    filename = "E:/eclipse/workspace/Py36/com/wdzy/db2/t1.xlsx"
    #打開測試文檔
    excel = xlrd.open_workbook(filename)
    #讀取sheet頁面位置
    sheet = excel.sheet_by_name("test")
    return sheet

# 讀取excel表格數據並寫入數據庫
def read_write():
    sheet = get_sheet()
    sql = "insert into t1qian values(%s,%s)"
    for i in range(sheet.nrows):
        l = sheet.row_values(i)
        count = (l[0],l[1])
        DBUtils().operate_dateByCount(sql, count)
        
read_write()

excel只設置了兩行內容id,name

2. 讀取數據庫數據寫入excel表格

'''
Created on 2019年
讀取數據庫數據寫入excel表格
@author: Root
'''
import xlwt
from DBUtils import DBUtils

# 設置表格樣式
def set_style(name,height,bold=True):
    style = xlwt.XFStyle()
    font = xlwt.Font()
    font.name = name
    font.bold = bold
    font.colour_index = 4
    font.height = height
    style.font = font
    return style

# 將數據寫入excel表格
def write_excel(sql,path):
    # 創建工作薄
    workbook = xlwt.Workbook(encoding='utf8')
    # 創建sheet
    sheet = workbook.add_sheet('test',cell_overwrite_ok=True)
    # 設置表頭
    sheet.write(0,0,'id',set_style('Times New Roman', 220, True))   # 1行1列
    sheet.write(0,1,'name',set_style('Times New Roman', 220, True)) # 1行2列

    # 查詢數據庫中數據   
    result = DBUtils().get_allDate(sql)

    for i in range(len(result)):
        # 寫入數據
        sheet.write(i+1,0,result[i][0])   # i+1行1列
        sheet.write(i+1,1,result[i][1])   # i+1行2列
    
    # 保存文件
    workbook.save(path)
    
#     workbook.close()

sql = 'select * from t1qian'
path = 'E:/eclipse/workspace/Py36/com/wdzy/db3/t2.xlsx'
write_excel(sql, path)