イミュータブルにしたり戻したり。 dict やら list やら bytearray やらを tuple や byte やらに。

immutable.py

# coding: utf8


def immutable(o):
    recursion = immutable
    if isinstance(o, dict):
        return tuple((k, recursion(v)) for k, v in o.iteritems())
    elif isinstance(o, list):
        return tuple(recursion(v) for v in o)
    elif isinstance(o, bytearray):
        return bytes(o)
    else:
        return o


def mutable(o, *args):
    recursion = mutable
    if not args:
        return o
    first, rest = args[0], args[1:]
    if issubclass(first, dict):
        return first((k, recursion(v, *rest)) for k, v in o)
    elif issubclass(first, list):
        return first(recursion(v, *rest) for v in o)
    else:
        return first(o)


if __name__ == '__main__':
    assert immutable(
        {'a': ['b', {'c': 'd'}]}) == (('a', ('b', (('c', 'd'),))),)
    k = [{'point': {'x': 1, 'y': 2}}, {'point': {'x': 2, 'y': 1}}]
    assert mutable(immutable(k), list, dict, dict) == k
    assert mutable(immutable(k), list, dict, dict, float)

乾燥ワカメみたいだ。

再帰つかいすぎなので、効率は悪いかもなー。