jiekou-python3/Public/get_excel.py

# -*- coding: utf-8 -*-
# @Time    : 2017/6/4 20:35
# @Author  : lileilei
# @File    : get_excel.py
import xlrd,os
from Public.log import LOG, logger


@logger('解析测试用例文件')
def datacel(filepath):
    try:
        file = xlrd.open_workbook(filepath)
        print(file)
        rslut = file.sheets()[0]
        nrows = rslut.nrows
        listid = []
        listkey = []
        listconeent = []
        listurl = []
        listfangshi = []
        listqiwang = []
        listname = []
        for i in range(1, nrows):
            listid.append(rslut.cell(i, 0).value)
            listkey.append(rslut.cell(i, 2).value)
            listconeent.append(rslut.cell(i, 3).value)
            listurl.append(rslut.cell(i, 4).value)
            listname.append(rslut.cell(i, 1).value)
            listfangshi.append((rslut.cell(i, 5).value))
            listqiwang.append((rslut.cell(i, 6).value))
        return listid, listkey, listconeent, listurl, listfangshi, listqiwang, listname
    except Exception as e:
        print(e)
        LOG.info('打开测试用例失败，原因是:%s' % e)
        return
@logger('生成数据驱动所用数据')
def makedata():
    path = os.path.join(os.path.join(os.getcwd(),'test_case_data'),'case.xlsx')
    listid, listkey, listconeent, listurl, listfangshi, listqiwang, listname=datacel(path)
    i=0
    make_data=[]
    for i in range(len(listid)):
        make_data.append({'url':listurl[i],'key':listkey[i],'coneent':listconeent[i],'fangshi':listfangshi[i],'qiwang':listqiwang[i]})
        i+=1
    return make_data