patch_requests.py

import json
from functools import wraps

import requests

allow_methods = [
    'options',
    'head',
    'get',
    'post',
    'put',
    'delete',
]


def patch_requests_method(method: str):
    assert method in allow_methods, f'method must be in {allow_methods}'
    fn_method = getattr(requests, method)

    @wraps(fn_method)
    def new_fn(*args, **kw):
        request = {'method': method, 'arguments': {'args': args, 'kwargs': kw}}
        res = fn_method(*args, **kw)
        try:
            js = res.json()
        except json.decoder.JSONDecodeError:
            js = None
        response = {
            'status_code': res.status_code,
            'headers': dict(res.headers),
            'cookies': dict(res.cookies),
        }
        if js is None:
            response['text'] = res.text
        else:
            response['json'] = js
        print(
            json.dumps(
                {'request': request, 'response': response}, indent=2, ensure_ascii=False
            )
        )
        return res

    setattr(requests, '_' + method, fn_method)
    setattr(requests, method, new_fn)
    return new_fn


def patch_requests_all():
    for x in allow_methods:
        print(f'patched requests.{x}')
        patch_requests_method(x)