PyReprHelpers.py

import re
import sys
from collections import deque
from queue import Queue


def is_python_repr_format(filename):
    try:
        f = open(filename, "r")
        first_byte = f.read(1)
        # List, dict, tuple, string or number.
        if first_byte in "[{(\"'0123456789-":
            return True
        f.seek(0)
        beginning = f.read(100)
        # Maybe some identifier.
        if re.match("[_A-Za-z][_a-zA-Z0-9.]*\(.*", beginning):
            return True
    except UnicodeDecodeError:
        return False
    return False


def load_python_repr_format(filename, env=None, default_constructor=None):
    code = open(filename, "r").read()
    if not env:
        env = {}
    elif isinstance(env, list):
        env = dict([(o.__name__, o) for o in env])
    else:
        env = dict(env)
    env["loadQueue"] = load_queue
    if hasattr(default_constructor, "__module__"):
        env.update(vars(sys.modules[default_constructor.__module__]))
    elif hasattr(default_constructor, "__name__"):
        env[default_constructor.__name__] = default_constructor
    return eval(code, env)


def load_queue(l):
    q = Queue()
    q.queue = q.queue.__class__(l)
    return q


def better_repr(o):
    # the main difference: this one is deterministic
    # the orig dict.__repr__ has the order undefined.
    if isinstance(o, list):
        return "[\n" + "".join([better_repr(v) + ",\n" for v in o]) + "]"
    if isinstance(o, deque):
        return "deque([\n" + "".join([better_repr(v) + ",\n" for v in o]) + "])"
    if isinstance(o, Queue):
        return "loadQueue([\n" + "".join([better_repr(v) + ",\n" for v in list(o.queue)]) + "])"
    if isinstance(o, tuple):
        return "(" + ", ".join(map(better_repr, o)) + ")"
    if isinstance(o, dict):
        return "{\n" + "".join([better_repr(k) + ": " + better_repr(v) + ",\n" for (k, v) in sorted(o.items())]) + "}"
    if isinstance(o, set):
        return "set([\n" + "".join([better_repr(v) + ",\n" for v in sorted(o)]) + "])"
    # fallback
    return repr(o)