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)