forked from DxvLwRYF/apiAutoTest
53 lines
1.3 KiB
Python
53 lines
1.3 KiB
Python
#!/usr/bin/env python
|
||
# _*_ coding: utf-8 _*_
|
||
"""
|
||
@project: apiAutoTest
|
||
@file: hooks.py
|
||
@author: zy7y
|
||
@time: 2021/2/27
|
||
@site: https://cnblogs.com/zy7y
|
||
@github: https://github.com/zy7y
|
||
@gitee: https://gitee.com/zy7y
|
||
@desc: 扩展方法, 2021/02/27
|
||
关于exec执行python代码可查阅资料：https://python3-cookbook.readthedocs.io/zh_CN/latest/c09/p23_executing_code_with_local_side_effects.html
|
||
|
||
"""
|
||
import json
|
||
import time
|
||
|
||
|
||
def exec_func(func: str) -> str:
|
||
"""执行函数(exec可以执行Python代码)
|
||
:params func 字符的形式调用函数
|
||
: return 返回的将是个str类型的结果
|
||
"""
|
||
# 得到一个局部的变量字典，来修正exec函数中的变量，在其他函数内部使用不到的问题
|
||
loc = locals()
|
||
exec(f"result = {func}")
|
||
return str(loc['result'])
|
||
|
||
|
||
def get_current_highest():
|
||
"""获取当前时间戳"""
|
||
return int(time.time())
|
||
|
||
|
||
def sum_data(a, b):
|
||
"""计算函数"""
|
||
return a + b
|
||
|
||
|
||
def set_token(token: str):
|
||
"""设置token，直接返回字典"""
|
||
return {"Authorization": token}
|
||
|
||
|
||
|
||
if __name__ == '__main__':
|
||
# 实例, 调用无参数方法 get_current_highest
|
||
result = exec_func("get_current_highest()")
|
||
print(result)
|
||
|
||
# 调用有参数方法 sum_data
|
||
print(exec_func("sum_data(1,3)"))
|