web自动化注册登录---（4）excel 模块的编写

上一篇：自动生成手机号方法的编写

个人还是喜欢利用excel 来保存生成的一些日志文件，下面创建一个excel 的类

from openpyxl import workbook,load_workbook


# 初始化生成一个excel，用该excel来记录一些日志信息
def init_excel(path, sheetname=None):
	# 新建用 workbook.Workbook()
    wb = workbook.Workbook()
    if not sheetname:
        ws = wb.active
    else:
        ws = wb[sheetname]
    ws['A1'] = 'phone'
    ws['B1'] = 'time_stamp'
    ws['C1'] = 'register_result'
    ws['D1'] = 'login_result'
    # ws['E1'] = 'pic_url'
    # 记得保存
    wb.save(path)
    wb.close()


# 将日志信息写进excel中
def write(path, data, row_index, sheetname=None):
	# 追加用load_workbook
    wb = load_workbook(path)
    if not sheetname:
        ws = wb.active
    else:
        ws = wb[sheetname]
    ws.cell(row_index, 1).value = data.get('phone')
    ws.cell(row_index, 2).value = data.get('time_stamp')
    ws.cell(row_index, 3).value = data.get('register_result')
    ws.cell(row_index, 4).value = data.get('login_result')
    # ws.cell(row_index, 5).value = data.get('pic_url')
    row_index += 1
    # 记得保存
    wb.save(path)
    # 记得关闭
    wb.close()
    return row_index


if __name__ == '__main__':
    init_excel('test.xlsx')
    val = {
        "id": 1,
        "h5_pic_url": 2,
        "acture_pic_url": 3,
        "differentia": 4
    }
    row_index = 2
    row_index = write('test.xlsx',val,row_index)
    print(row_index)

下一篇：公共类的编写