大家好~我是
米洛
!
我正在从0到1打造一个开源的接口测试平台, 也在编写一套与之对应的教程
,希望大家多多支持。
欢迎关注我的公众号米洛的测开日记
,获取最新文章教程!
上一节我们构思了一下怎么去支持Python脚本,在和@joker讨论以后,考虑到python脚本不会太复杂,所以我们这一节我们就先实现exec
的版本。
exec是一个比较厉害的内置库
,它为想要执行python代码的用户提供了便利,与之对应的还有execfile版本,这个是文件版本的exec,也就是说,他接受的参数从一段代码变成了一个指定的文件
。
上一节我们似乎介绍了exec的普通用法,那今天我们就讲讲怎么通过exec拿到对应的执行结果,毕竟对于执行结果我们是会渴望能拿到的。
import datetime def get_first_day_of_month(): # 把day指定为01,肯定就是第一天啦 return datetime.today().strftime("%Y-%m-01")
python代码是如此编写的,那exec里面我们怎么拿到这个方法的结果呢?
首先呢,exec是没用返回值的。我们只能通过,在exec执行的代码块里面定义变量去接收返回值。
import datetime return_value = datetime.today().strftime("%Y-%m-01")
那么在我们的后台代码
里面,就可以拿到return_value作为变量。
但我实际在fastapi的方法里面,发现是不生效的。会报return_value not defined,但是单个执行文件确实是可以。
于是我采用了新的方法,定义一个dict,传给exec,这样exec里面用到的变量等信息,都会返回到dict里面。
command = """ import datetime return_value = datetime.today().strftime("%Y-%m-01") """ loc = dict() exec(command, loc) print(loc['return_value'])
其实本身是不想提供这个接口的,因为大部分脚本都可以在本地测试后传上来,但想了下,有就有吧,对于os.remove这样危险的语句还是得禁止掉。
不过由于是演示,我就不禁止了。
先看一张效果图,执行完成后会有数据返回。但一定要注意的是,如果想拿到返回值,必须要有:
current_time = xxx
,并且把返回值设置为current_time。
为什么呢?因为exec本身是拿不到返回值的,如果你去exec语句里面定义一个变量:
current_time2 = "234"
这样就能在loc里面取到返回值了,所以如果要求有返回值,则必须写上上述的python语句。
我们就是通过exec执行代码,定义变量为返回值,接着取出该变量的方式实现整个过程的。
来看下代码怎么写:
和上文提到的一样,如果没有返回值的话,直接就是None了。
和redis操作很相似,实现run方法,也是记录数据并拿到返回值。
新增Python类型,注意这里我把这些数字,0-3都放到了Config类里面。
提醒一下各位,比如我这里的方法没有给出对应的文件路径,大家可以全局搜索,找到之。
对应Config.py也增加了这样的子类。
页面基本是沿用的Redis的部分,在外部也稍微改造了一下:
改用开关的形式,看起来更直观。
今天的内容就介绍到这里了,欢迎大家积极提出建议,一起打造一款开源工具。