From 05c645c5ea884dc0e0ed4299825332b8e5287ea7 Mon Sep 17 00:00:00 2001 From: Un7corn Date: Sat, 21 Mar 2026 16:33:27 +0800 Subject: [PATCH 1/2] Create batch_templates.yaml --- httpie/cli/examples/batch_templates.yaml | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) create mode 100644 httpie/cli/examples/batch_templates.yaml diff --git a/httpie/cli/examples/batch_templates.yaml b/httpie/cli/examples/batch_templates.yaml new file mode 100644 index 0000000000..a0c9f5fc23 --- /dev/null +++ b/httpie/cli/examples/batch_templates.yaml @@ -0,0 +1,22 @@ +requests: + - name: get-ip + method: GET + url: https://httpbin.org/ip + headers: + Accept: application/json + + - name: post-json + method: POST + url: https://httpbin.org/post + headers: + Content-Type: application/json + json_body: + message: hello + source: batch-yaml + + - name: with-query + method: GET + url: https://httpbin.org/get + params: + page: 1 + size: 10 From 37f3d0fbd0bc7825244f2e4b7a887b124911021b Mon Sep 17 00:00:00 2001 From: Un7corn Date: Sat, 21 Mar 2026 16:33:55 +0800 Subject: [PATCH 2/2] Create batch.py --- httpie/cli/examples/httpie/cli/batch.py | 167 ++++++++++++++++++++++++ 1 file changed, 167 insertions(+) create mode 100644 httpie/cli/examples/httpie/cli/batch.py diff --git a/httpie/cli/examples/httpie/cli/batch.py b/httpie/cli/examples/httpie/cli/batch.py new file mode 100644 index 0000000000..a90d976b5a --- /dev/null +++ b/httpie/cli/examples/httpie/cli/batch.py @@ -0,0 +1,167 @@ +import json +from pathlib import Path +from dataclasses import dataclass, asdict +from typing import Any, Dict, List, Optional + +try: + import yaml +except ImportError: + yaml = None + + +@dataclass +class BatchRequestItem: + name: str + method: str + url: str + headers: Optional[Dict[str, str]] = None + params: Optional[Dict[str, Any]] = None + json_body: Optional[Any] = None + data: Optional[Any] = None + auth: Optional[Dict[str, Any]] = None + group: Optional[str] = None + + +@dataclass +class BatchExecutionResult: + name: str + method: str + url: str + success: bool + status_code: Optional[int] = None + error: Optional[str] = None + response_text: Optional[str] = None + + +class BatchExecutor: + def __init__( + self, + template_file: str, + exec_mode: str = "all", + preview: bool = False, + output_dir: str = "batch_results", + output_format: str = "json", + verbose: bool = False, + ) -> None: + self.template_file = Path(template_file) + self.exec_mode = exec_mode + self.preview = preview + self.output_dir = Path(output_dir) + self.output_format = output_format + self.verbose = verbose + self.requests: List[BatchRequestItem] = [] + + def load_templates(self) -> List[BatchRequestItem]: + if not self.template_file.exists(): + raise FileNotFoundError(f"Template file not found: {self.template_file}") + + suffix = self.template_file.suffix.lower() + content = self.template_file.read_text(encoding="utf-8") + + if suffix in [".yaml", ".yml"]: + if yaml is None: + raise RuntimeError("PyYAML is required for YAML batch templates") + raw = yaml.safe_load(content) + elif suffix == ".json": + raw = json.loads(content) + else: + raise ValueError(f"Unsupported template format: {suffix}") + + items = raw.get("requests", raw if isinstance(raw, list) else []) + self.requests = [BatchRequestItem(**item) for item in items] + return self.requests + + def preview_requests(self) -> None: + print("Batch request preview:") + for idx, req in enumerate(self.requests, start=1): + print(f"{idx}. [{req.method}] {req.name} -> {req.url}") + + def select_requests(self) -> List[BatchRequestItem]: + if self.exec_mode == "all": + return self.requests + + if self.exec_mode == "interactive": + print("Interactive mode is not fully verified yet.") + selected = [] + for req in self.requests: + answer = input(f"Execute '{req.name}'? [y/N]: ").strip().lower() + if answer == "y": + selected.append(req) + return selected + + if self.exec_mode == "preview": + return [] + + raise ValueError(f"Unsupported exec mode: {self.exec_mode}") + + def execute_one(self, req: BatchRequestItem) -> BatchExecutionResult: + """ + 这里暂时只放占位逻辑。 + 真正集成时，应接 HTTPie 现有请求执行流程，而不是自己重复造轮子。 + """ + try: + # TODO: 替换为 HTTPie 核心请求执行调用 + if self.verbose: + print(f"Executing: [{req.method}] {req.url}") + + return BatchExecutionResult( + name=req.name, + method=req.method, + url=req.url, + success=False, + error="Execution path not integrated with HTTPie core yet", + ) + except Exception as exc: + return BatchExecutionResult( + name=req.name, + method=req.method, + url=req.url, + success=False, + error=str(exc), + ) + + def execute(self) -> List[BatchExecutionResult]: + self.load_templates() + + if self.preview or self.exec_mode == "preview": + self.preview_requests() + return [] + + selected_requests = self.select_requests() + results: List[BatchExecutionResult] = [] + + for req in selected_requests: + result = self.execute_one(req) + results.append(result) + + self.export_results(results) + return results + + def export_results(self, results: List[BatchExecutionResult]) -> None: + self.output_dir.mkdir(parents=True, exist_ok=True) + + if self.output_format == "json": + output_file = self.output_dir / "results.json" + output_file.write_text( + json.dumps([asdict(r) for r in results], ensure_ascii=False, indent=2), + encoding="utf-8", + ) + return + + if self.output_format == "md": + output_file = self.output_dir / "results.md" + lines = [ + "# Batch Execution Report", + "", + "| Name | Method | URL | Success | Status | Error |", + "|---|---|---|---|---|---|", + ] + for r in results: + lines.append( + f"| {r.name} | {r.method} | {r.url} | {r.success} | " + f"{r.status_code or ''} | {r.error or ''} |" + ) + output_file.write_text("\n".join(lines), encoding="utf-8") + return + + raise ValueError(f"Unsupported output format: {self.output_format}")