record-session.py

import json
import pickle
from contextlib import contextmanager

import requests


def hash_args(args, kwargs):
    return hash(json.dumps((args, kwargs), sort_keys=True))


@contextmanager
def record(session=None):
    recorded = {}

    if session is None:
        session = requests.Session()

    def rec_request(*args, **kwargs):
        key = hash_args(args, kwargs)

        if key in recorded:
            return pickle.loads(recorded[key])

        response = do_request(*args, **kwargs)
        recorded[key] = pickle.dumps(response)

        return response

    do_request = session.request
    session.request = rec_request

    yield session

    session.request = do_request