Python 接口自动化实战:10 个代码示例教你搞定参数关联

import requestsdef test_hardcoded_correlation():    # 1. 登录    login_resp = requests.post("http://api.example.com/login", json={"user": "admin"})    token = login_resp.json()["data"]["token"]    # 2. 下单 - 直接使用变量    # 缺点：代码耦合度极高，无法复用，无法跨文件调用    order_data = {"token": token, "product_id": 101}    order_resp = requests.post("http://api.example.com/order", json=order_data)    assert order_resp.status_code == 200

import pytestimport requests@pytest.fixture()def get_token():    """前置操作：登录并返回token"""    resp = requests.post("http://api.example.com/login", json={"user": "admin"})    token = resp.json()["data"]["token"]    return tokendef test_order_with_fixture(get_token):    """测试下单：依赖get_token"""    # get_token 会自动触发登录逻辑    headers = {"Authorization": get_token}    resp = requests.post("http://api.example.com/order", json={"pid": 101}, headers=headers)    assert resp.status_code == 200

# context.py# 一个简单的单例模式或全局字典，用于存储动态参数GLOBAL_CONTEXT = {}def set_context(key, value):    GLOBAL_CONTEXT[key] = valuedef get_context(key):    return GLOBAL_CONTEXT.get(key)

import jsonpathdef extract_value(response_json, json_path, key_name):    """    从响应中提取数据并存入上下文    :param response_json: 接口返回的字典    :param json_path: JSONPath表达式，如 '$.data.token'    :param key_name: 存入上下文的键名    """    result = jsonpath.jsonpath(response_json, json_path)    if result:        # 存入全局上下文（示例3中的对象）        set_context(key_name, result[0])        print(f"💡 提取成功: {key_name} = {result[0]}")    else:        raise ValueError(f"提取失败: {json_path}")# 使用# extract_value(login_resp.json(), "$.data.token", "my_token")‍

import redef extract_by_regex(text_response, regex_pattern, key_name):    """    使用正则表达式提取数据    """    match = re.search(regex_pattern, text_response)    if match:        set_context(key_name, match.group(1))        print(f"💡 正则提取成功: {key_name}")    else:        print(f"❌ 正则匹配失败")# 使用场景：从 HTML 中提取# extract_by_regex(html_text, r'value="(.*?)"', "csrf_token")‍

def replace_params(data):    """    递归替换字典、列表或字符串中的 ${key}    """    if isinstance(data, dict):        return {k: replace_params(v) for k, v in data.items()}    elif isinstance(data, list):        return [replace_params(item) for item in data]    elif isinstance(data, str):        # 检查是否包含 ${...}        if data.startswith("${") and data.endswith("}"):            key = data[2:-1]            return get_context(key) or data # 如果找不到key，返回原值    return data# 效果# 输入: {"token": "${my_token}"} -> 输出: {"token": "xyz123"}‍

import jsonimport osCONTEXT_FILE = "context.json"def save_to_file(key, value):    # 1. 读取文件    data = {}    if os.path.exists(CONTEXT_FILE):        with open(CONTEXT_FILE, 'r') as f:            data = json.load(f)    # 2. 更新数据    data[key] = value    # 3. 写回文件    with open(CONTEXT_FILE, 'w') as f:        json.dump(data, f, indent=4)

class ApiClient:    def send(self, method, url, req_data=None, extract_rules=None):        # 1. 参数替换        if req_data:            req_data = replace_params(req_data)        # 2. 发送请求        resp = requests.request(method, url, json=req_data)        # 3. 后置处理：提取数据        if extract_rules:            for key, path in extract_rules.items():                extract_value(resp.json(), path, key)        return resp‍

def test_full_shopping_flow():    client = ApiClient()    # 步骤1：登录，并提取 token    client.send("POST", "/login",                 {"user": "admin"},                 extract_rules={"token": "$.data.token"})    # 步骤2：下单，使用 ${token}，并提取 order_id    client.send("POST", "/order",                 {"token": "${token}", "pid": 101},                 extract_rules={"oid": "$.data.order_id"})    # 步骤3：支付，使用 ${oid}    client.send("POST", "/pay", {"order_id": "${oid}"})

import pytest@pytest.fixture(scope="session", autouse=True)def clean_context():    yield    # 测试会话结束后，删除上下文文件    if os.path.exists(CONTEXT_FILE):        os.remove(CONTEXT_FILE)        print("🧹 上下文已清理")‍