当前位置:首页 > 技术 > 正文内容

基于Requests的链式HTTP请求执行器设计

访客 技术 2026年9月12日 12

在接口自动化测试中,常需按顺序执行多个依赖性HTTP请求(如先登录获取token,再调用业务接口)。为保障流程可控性与错误中断能力,可构建一个支持步骤化、条件终止的请求执行引擎。

核心设计思路:将每个请求步骤抽象为结构化字典,统一调度入口根据请求类型分发至对应方法;执行器以for循环遍历步骤列表,并在任一请求返回非成功状态码时立即退出,避免后续无效调用。

步骤驱动型请求分发

以下方法接收单步配置,识别method字段并路由至内部私有处理函数:

def dispatch_request(self, step_config):
    method = step_config.get("method", "").lower()
    if method == "get":
        return self._execute_get(step_config)
    elif method == "post":
        return self._execute_post(step_config)
    else:
        return {"status": "error", "code": 405, "message": f"Unsupported HTTP method: {method}"}

顺序执行控制器

该方法逐条执行步骤,一旦某步失败(status != "success"),即刻中止流程并返回当前结果:

def run_sequential_steps(self, step_list):
    last_result = None
    for step in step_list:
        last_result = self.dispatch_request(step)
        if last_result.get("status") != "success":
            break
    return last_result

完整实现示例

以下为精简重构后的工具类,移除了冗余注释与调试代码,增强健壮性与可读性:

# -*- coding: utf-8 -*-
import json
import requests
from utils.config_utils import local_config

class SequentialRequestExecutor:
    def __init__(self):
        self.base_url = local_config.HOSTS
        self.session = requests.Session()

    def _execute_get(self, config):
        url = self.base_url + config.get("path", "")
        params = json.loads(config.get("query_params", "{}"))
        headers = json.loads(config.get("headers", "{}")) if config.get("headers") else {}
        
        try:
            resp = self.session.get(url=url, params=params, headers=headers, timeout=10)
            return {
                "status": "success",
                "code": resp.status_code,
                "body": resp.text,
                "headers": dict(resp.headers),
                "elapsed_ms": int(resp.elapsed.total_seconds() * 1000)
            }
        except Exception as e:
            return {"status": "error", "code": 0, "message": str(e)}

    def _execute_post(self, config):
        url = self.base_url + config.get("path", "")
        json_payload = json.loads(config.get("json_body", "{}"))
        headers = json.loads(config.get("headers", "{}")) if config.get("headers") else {}
        
        try:
            resp = self.session.post(
                url=url,
                json=json_payload,
                headers=headers,
                timeout=10
            )
            resp.encoding = resp.apparent_encoding
            return {
                "status": "success",
                "code": resp.status_code,
                "body": resp.text,
                "headers": dict(resp.headers),
                "elapsed_ms": int(resp.elapsed.total_seconds() * 1000)
            }
        except Exception as e:
            return {"status": "error", "code": 0, "message": str(e)}

    def dispatch_request(self, step_config):
        method = step_config.get("method", "").lower()
        if method == "get":
            return self._execute_get(step_config)
        elif method == "post":
            return self._execute_post(step_config)
        else:
            return {"status": "error", "code": 405, "message": f"Method '{method}' not supported"}

    def run_sequential_steps(self, step_list):
        result = None
        for step in step_list:
            result = self.dispatch_request(step)
            if result["status"] != "success":
                break
        return result

# 使用示例
if __name__ == "__main__":
    executor = SequentialRequestExecutor()
    
    steps = [
        {
            "method": "get",
            "path": "/cgi-bin/token",
            "query_params": '{"grant_type":"client_credential","appid":"wx55614004f367f8ca","secret":"65515b46dd758dfdb09420bb7db2c67f"}',
            "headers": '{}'
        },
        {
            "method": "post",
            "path": "/cgi-bin/tags/delete",
            "json_body": '{"tag":{"id":456}}',
            "headers": '{"Content-Type":"application/json"}'
        }
    ]
    
    final_output = executor.run_sequential_steps(steps)
    print(json.dumps(final_output, indent=2, ensure_ascii=False))
标签: requests

相关文章

Linux crontab 详解

1) crontab 是什么cron 是 Linux 的定时任务守护进程;crontab 是用来编辑/查看“按时间周期执行命令”的表(cron table)。常见两类:用户 crontab:每个用户一份(crontab -e 编辑)系统级 crontab / cron.d:可指定执行用户(/etc/crontab、/etc/cron.d/*)2) crontab 时间...

富文本里可以允许的 HTML 属性

一、所有标签默认允许的安全属性(极少)class        (可选)id           (通常建议禁用)title️ 注意:id 容易被滥用做锚点注入,很多系统直接禁用class 允许的话最好只允许固定前缀(如 editor-*)二、a 标签允许属性<a href="" t...

Mac 安装 Node.js 指南

方法一:通过官网安装包(最简单,适合初学者)如果你只是想快速安装并开始使用,这是最直接的方法。访问 Node.js 官网。页面会显示两个版本:LTS (Recommended For Most Users):长期支持版,最稳定。建议选这个。Current:最新特性版,包含最新功能但可能不够稳定。下载 .pkg 安装包并运行。按照安装向导点击“下一步”即可完成。方法二:使用 Homebrew 安装(...

Dom\HTML_NO_DEFAULT_NS 的副作用:自动加闭合标签

在使用Dom\HTMLDocument时,Dom\HTML_NO_DEFAULT_NS 将禁止在解析过程中设置元素的命名空间, 此设置是为了与DOMDocument向后兼容而存在的。当使用它时,已知的一个副作用就是:自动加闭合标签例如 </img> 为什么会这样?当你使用:Dom\HTML_NO_DEFAULT_NS文档会变成 无命名空间模式,此时内部更接近 XML...

Laravel 事件和监听器创建

在 Laravel 中,使用 Artisan 命令创建 Events(事件) 和 Listeners(监听器) 是非常高效的。你可以通过以下几种方式来实现:1. 手动创建单个 Event如果你只想创建一个事件类,可以使用 make:event 命令:Bashphp artisan make:event UserRegistered执行后,文件将生成在 app/Even...

自定义域名解析神器 dnsmasq

什么是 dnsmasq?dnsmasq 是一个轻量级、功能强大的网络服务工具,专为小型和中等规模网络设计。它是一个综合的网络基础设施解决方案[1]。dnsmasq 能做什么?功能说明应用场景DNS 转发与缓存将 DNS 查询转发到上游服务器(ISP、Google DNS 等),并在本地缓存结果加快 DNS 查询速度,减少外部 DNS 流量本地 DNS解析本地网络设备的主机名,无需编辑&n...

发表评论

访客

◎欢迎参与讨论,请在这里发表您的看法和观点。