大家好~我是米洛

我正在从0到1打造一个开源的接口测试平台, 也在编写一套与之对应的教程,希望大家多多支持。

欢迎关注我的公众号米洛的测开日记,获取最新文章教程!

回顾

上一节我们构思了一下怎么去支持Python脚本,在和@joker讨论以后,考虑到python脚本不会太复杂,所以我们这一节我们就先实现exec的版本。

exec初体验

exec是一个比较厉害的内置库,它为想要执行python代码的用户提供了便利,与之对应的还有execfile版本,这个是文件版本的exec,也就是说,他接受的参数从一段代码变成了一个指定的文件

上一节我们似乎介绍了exec的普通用法,那今天我们就讲讲怎么通过exec拿到对应的执行结果,毕竟对于执行结果我们是会渴望能拿到的。

  • 实现第一个python方法: 获取本月第一天的日期
  1. from datetime import datetime
  2. def get_first_day_of_month():
  3. # 把day指定为01,肯定就是第一天啦
  4. return datetime.today().strftime("%Y-%m-01")

python代码是如此编写的,那exec里面我们怎么拿到这个方法的结果呢?

首先呢,exec是没用返回值的。我们只能通过,在exec执行的代码块里面定义变量去接收返回值。

  1. from datetime import datetime
  2. return_value = datetime.today().strftime("%Y-%m-01")

那么在我们的后台代码里面,就可以拿到return_value作为变量。

但我实际在fastapi的方法里面,发现是不生效的。会报return_value not defined,但是单个执行文件确实是可以。

于是我采用了新的方法,定义一个dict,传给exec,这样exec里面用到的变量等信息,都会返回到dict里面。

  1. command = """
  2. from datetime import datetime
  3. return_value = datetime.today().strftime("%Y-%m-01")
  4. """
  5. loc = dict()
  6. exec(command, loc)
  7. print(loc['return_value'])

测试平台系列(95) 前置条件支持简单的python脚本 - 图1

编写在线执行exec接口

其实本身是不想提供这个接口的,因为大部分脚本都可以在本地测试后传上来,但想了下,有就有吧,对于os.remove这样危险的语句还是得禁止掉。

不过由于是演示,我就不禁止了。

测试平台系列(95) 前置条件支持简单的python脚本 - 图2

先看一张效果图,执行完成后会有数据返回。但一定要注意的是,如果想拿到返回值,必须要有:

current_time = xxx,并且把返回值设置为current_time。

为什么呢?因为exec本身是拿不到返回值的,如果你去exec语句里面定义一个变量:

  1. current_time2 = "234"

这样就能在loc里面取到返回值了,所以如果要求有返回值,则必须写上上述的python语句。

我们就是通过exec执行代码,定义变量为返回值,接着取出该变量的方式实现整个过程的。

来看下代码怎么写:

测试平台系列(95) 前置条件支持简单的python脚本 - 图3

和上文提到的一样,如果没有返回值的话,直接就是None了。

编写python数据构造器

测试平台系列(95) 前置条件支持简单的python脚本 - 图4

和redis操作很相似,实现run方法,也是记录数据并拿到返回值。

改写获取数据构造器的方法

测试平台系列(95) 前置条件支持简单的python脚本 - 图5

新增Python类型,注意这里我把这些数字,0-3都放到了Config类里面。

提醒一下各位,比如我这里的方法没有给出对应的文件路径,大家可以全局搜索,找到之。

对应Config.py也增加了这样的子类。

测试平台系列(95) 前置条件支持简单的python脚本 - 图6

前端部分的改动

测试平台系列(95) 前置条件支持简单的python脚本 - 图7

页面基本是沿用的Redis的部分,在外部也稍微改造了一下:

测试平台系列(95) 前置条件支持简单的python脚本 - 图8

改用开关的形式,看起来更直观。


今天的内容就介绍到这里了,欢迎大家积极提出建议,一起打造一款开源工具。