unravel_pickle

unravel_pickle.py

from pickle import PickleBuffer


def unravel_pickle(obj: object, seen: list[int] | None = None) -> Any:
    if isinstance(obj, (int, float, complex, bytes, str, bool, type(None))):
        return obj
    if isinstance(obj, (bytearray, memoryview, PickleBuffer)):
        return str(type(obj)), bytes(obj)
    if callable(obj):
        return str(obj)

    seen = seen or []
    try:
        return "__seen", seen.index(id(obj)), str(obj)
    except ValueError:
        pass
    seen.append(id(obj))

    if isinstance(obj, (list, tuple)):
        return str(type(obj)), [unravel_pickle(v, seen) for v in obj]
    if isinstance(obj, (set, frozenset)):
        return str(type(obj)), [unravel_pickle(v, seen) for v in sorted(obj, key=str)]
    if isinstance(obj, dict):
        return str(type(obj)), [
            (unravel_pickle(k, seen), unravel_pickle(v, seen)) for k, v in obj.items()
        ]
    return unravel_pickle(obj.__reduce_ex__(5), seen)