一丶接口测试平台之替换请求参数中的变量

# author: 倾丶几回断肠

import random
import string
import json
import re


def change_globe_val(obj):
    """ 替换全局变量 """
    if isinstance(obj, dict):
        obj = json.dumps(obj)
    p = '\${(.*?)}'
    keys = re.findall(p, obj)
    values = []
    for i in keys:
        # value = Val.objects.filter(name=i)
        value = ''.join(random.choices(string.ascii_letters, k=6))
        values.append(value)
    z = list(zip(keys, values))
    print(z)
    newObj = obj
    for k, v in z:
        patt = '\${%s}' % k
        newObj = re.sub(patt, v, newObj)
    return newObj


d = '{"userName": "admin", "password": "${password}", "token": "${token}"}'
dd = {"user-agent": "${usreAgent}"}
new_obj = change_globe_val(d)
print(new_obj)
# new_obj = change_globe_val(dd)
# print(new_obj)

 

结果如下：