1
0
mirror of https://github.com/httpie/cli.git synced 2025-08-10 22:42:05 +02:00

core: support custom request/response classes (#1205)

* core: support custom request/response classes

* Move to `httpie.models`, prefix with `Requests`
This commit is contained in:
Batuhan Taskaya
2021-11-25 02:45:39 +03:00
committed by GitHub
parent c000886546
commit ef62fc11bf
4 changed files with 46 additions and 15 deletions

View File

@@ -4,7 +4,7 @@ import json
import sys
from contextlib import contextmanager
from pathlib import Path
from typing import Callable, Iterable, Union
from typing import Callable, Iterable
from urllib.parse import urlparse, urlunparse
import requests
@@ -14,6 +14,7 @@ from . import __version__
from .adapters import HTTPieHTTPAdapter
from .cli.dicts import HTTPHeadersDict
from .encoding import UTF8
from .models import RequestsMessage
from .plugins.registry import plugin_manager
from .sessions import get_httpie_session
from .ssl import AVAILABLE_SSL_VERSION_ARG_MAPPING, HTTPieHTTPSAdapter
@@ -36,7 +37,7 @@ def collect_messages(
args: argparse.Namespace,
config_dir: Path,
request_body_read_callback: Callable[[bytes], None] = None,
) -> Iterable[Union[requests.PreparedRequest, requests.Response]]:
) -> Iterable[RequestsMessage]:
httpie_session = None
httpie_session_headers = None
if args.session or args.session_read_only: