Python读取含有VBA宏的Excel数据

因为现在项目在开发新的API,我需要根据API方法论去校验这个API的计算逻辑和计算结果是否正确,而且需要将计算逻辑用Excel记录下来,所以就用了Excel的VBA。

1. 什么是VBA?

具体含义大家可以看百度百科,我就直接上图了,每个单元格都是调用了VBA的宏计算语言函数,我就简称这个Excel含有macro计算逻辑吧。

2.问题是什么?

大家都知道要比较一个API的准确性,我们需要自己根据方法论去把所有原始数据都推演计算一遍,得到结果。然后在跟开发做的API response做下对比,如果两个结果相同,则表示这个API计算正确。

现在我们PO不让我重写一遍开发的计算逻辑了,规定所有计算相关的逻辑都在Excel里面做好计算模板sheet,然后我将所有原始数据都放到计算模板同一个文件夹下。在含有Marco计算逻辑的那个excel(计算模板sheet)里面完成数据的引用和计算。

然后将API返回的Response也写入到同目录下的一个Excel中(样式跟计算模板计算出来的结果相同),模板里面将比较API response和Excel中的计算结果,然后输出一个如上图一样的一个sheet页(二维矩阵,单元格里的内容是TRUE或者FALSE)。

我要做的就是:将原始数据放到规定名字和格式的Excel中,将最终API的response也放到Excel中,然后打开这个含有macro函数的excel,查看sheet页中是否含有FALSE,如果有,告诉对应的行和列名称并输出。若是没有FALSE,这说明比对通过,API pass。

难点是:

我用Python写函数,现在Python用pandas打开这个含有Macro函数的Excel后,读出的数据永远都是Nan,因为它不能识别那些macro函数。。。所以我无法判断该API是否pass。

3. 手动操作Excel的时候要怎样看到是TRUE或FALSE呢?

在打开这个有macro函数的Excel之前,我们需要打开所有它引用的其他Excel。然后回到这个Excel的时候,这些宏会自动引用其他excel相应的数据,并得出计算结果。

4. 用Python要怎样实现这一手动逻辑呢?

在Python中调用VBA去控制这些Excel即可。

5. 具体要怎样实现呢?

a. 下载Python扩展包: pywin32 (这个有32和64位之分)

可以直接去下载zip文件,然后安装;

但我建议直接用命令: pip install pywin32 

b. 在自己python文件中引用:  from win32com import client

from win32com import client

c. 在python中写VBA:

#打开Excel应用,并且依次打开该计算模板Excel依赖的所有excel文件,然后打开最终计算模板excel

xlApp = client.Dispatch('Excel.Application')

# xlApp.visible默认是0,就是不显示所有Excel文件窗口哦

xlApp.visible = 1

# Open original excels in the given absolute path one by one

suite_name = suite_path.split(".")[-1]

file_names = filter(lambda x: x.find("File") > -1, input_params.keys())

for file_name in file_names:

    # excel = pd.ExcelFile("OutputExcel/%s/%s" % (suite_name, input_params[file_name]))

    xlApp.Workbooks.Open(r"%s\APG_API_RobotFramework\OutputExcel\%s\%s" % (workspace_path, suite_name, input_params[file_name]))

# Open the result excel and check data

xlBook = xlApp.Workbooks.Open("%s\APG_API_RobotFramework\OutputExcel\%s\%s" % (workspace_path, suite_name, RESULT_EXCEL))

xlsheet = xlBook.Worksheets("Result")

# 操作Excel可能会出错,所以需要用到try, finally结构块

try:

    # Store calculation result into result_list

    rows = xlsheet.UsedRange.Rows.Count

    columns = xlsheet.UsedRange.Columns.Count

    # 读取做好的比较数据模板sheet页中的数据并放入二维数组中

    result_list = []

    for row in range(1, rows):

        temp_list = []

        for column in range(1, columns):

            temp_list.append(str(xlsheet.Cells(row, column)))

        result_list.insert(row-1, temp_list)

    # print result_list

    # Get row and column title, then remove titles' values and put left values into new_list

    row_title = [x[0] for x in result_list][1:]

    column_title = [y for y in result_list[0]][1:]

    # print("row title : ", row_title)

    # print("column title :", column_title)

    new_list = []

    for x in result_list[1:]:

        new_list.append(x[1:])

    # print new_list

    # 因为最终要提示用户FALSE单元格对应的行和列名称,所以我用pandas来构造DataFrame,这样效率很高

    # Put data into DataFrame then select values which are "FALSE"

    df = pd.DataFrame(data=new_list, index=row_title, columns=column_title)

    fail_list = []

    for index, row in df.iterrows():

        if row[0] == "False":

            fail_list.extend([index, row[0]])

    # 若找到FALSE单元格,则提示具体错误信息,否则打印比较成功的提示

    assert fail_list.__len__() == 0, "Failed cells are : "+str(fail_list)

    print "Compare successfully !"

finally:

    # Don't let the save dialog pop up when close (Since the result excel shouldn't be saved)

    xlApp.DisplayAlerts = 0

    # xlBook.Save()

    xlBook.Close()

    xlApp.Workbooks.Close()

    # xlApp.visible = 0

    xlApp.Quit()

finally中的语句非常关键,相当于open一个文件之后进行读写,最终一定要关闭对应的io流一样!

那个计算模板是不允许改动的,所以在关闭那个xlBook的时候总是提示是否需要保存,在Jenkins上配置这个Project之后,所有相关操作肯定是无法手动干预的。所以我就采取了最直接的方式:

xlApp.DisplayAlerts = 0  

就是不弹出所有提示框。

先关闭计算模板Excel,然后关闭该Excel依赖的所有其他Excel。

最后退出Excel应用程序。

(若是最后退出程序以后发现还有一个空白的Excel窗口,这时候可以用 xlApp.visible=0 来隐藏)

所有这些都执行完了以后,这些被操作过的文件就不会被锁住了。(若是被锁住,以后用python读取的时候都会提示IOException哦~~)

最终的表现是:

系统会在运行该函数的时候先打开所有依赖的Excel文件,并打开最终的计算模板,在后台读出模板中比较数据,然后查看是否有FALSE单元格,并关闭所有打开过的Excel,一切都是自动化的哦~

RobotFramework Test Case如下:

因为项目太忙了,所以写的都是流水账,如果大家有什么更好的建议或者疑问,可以给我留言哦,谢谢阅读~~

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,684评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 87,143评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,214评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,788评论 1 277
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,796评论 5 368
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,665评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,027评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,679评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 41,346评论 1 299
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,664评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,766评论 1 331
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,412评论 4 321
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,015评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,974评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,203评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,073评论 2 350
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,501评论 2 343

推荐阅读更多精彩内容