接口测试框架第一节：重构操作Excel工具类

#coding=utf-8
#@Time    : 2019/2/3 10:45
#@File    : operation_excel.py.py
#@Author  : 老鹿（Fighter.Lu）

# import xlrd
# wb = xlrd.open_workbook('../dataconfig/interface.xlsx')
# sheet1 = wb.sheet_by_name('Sheet1')
# #获取excel行数
# rows = sheet1.nrows
# print('rows:',rows)
# #获取单元格cell
# cell = sheet1.cell_value(rowx=1 ,colx=1)
# print("url",cell)

import xlrd
#在已有的excel中追加数据
from openpyxl import load_workbook  #导入 load_workbook方法

class Operate_Excel():

    def __init__(self,fileName=None,sheetName=None):

        if fileName != None and sheetName != None:
            self.fileName = fileName
            self.sheetName = sheetName
        else:
            self.fileName = '../dataconfig/interface.xlsx'
            self.sheetName = 'Sheet1'

    #打开excel文件
    def open_work(self):
        wb = xlrd.open_workbook(self.fileName)
        sheet1 = wb.sheet_by_name(self.sheetName)
        return sheet1

    #获取excel页面使用的行数
    def get_rows(self):
        return  self.open_work().nrows


    #获取excel指定单元格的值
    def get_cell_value(self,row,col):
        return self.open_work().cell_value(row,col)

    #向指定单元格中写入数据
    def input_value(self,row,col,value):
        wb = load_workbook(self.fileName)
        sheet = wb.get_sheet_by_name(self.sheetName)
        #写入数据
        sheet.cell(row+1,col+1,value)
        #保存文件
        wb.save(self.fileName)

if __name__ == '__main__':
    oe = Operate_Excel('../dataconfig/interface.xlsx')
    print(oe.get_rows())
    print(oe.get_cell_value(1,1))
    oe.input_value(1,11,'PASS')