Functions to make things immutable

immutable.py

"""Functions to make things immutable."""
from collections import OrderedDict
from collections.abc import Hashable
from typing import Any

from immutabledict import immutabledict
from pydantic import BaseModel


def freeze_data(obj: Any) -> Any:
    """Check type and recursively return a new read-only object."""
    if isinstance(obj, (str, int, float, bytes, type(None), bool)):
        return obj
    elif isinstance(obj, tuple) and type(obj) != tuple:  # assumed namedtuple
        return type(obj)(*(freeze_data(i) for i in obj))
    elif isinstance(obj, (tuple, list)):
        return tuple(freeze_data(i) for i in obj)
    elif isinstance(obj, (dict, OrderedDict, immutabledict)):
        return immutabledict({k: freeze_data(v) for k, v in obj.items()})
    elif isinstance(obj, (set, frozenset)):
        return frozenset(freeze_data(i) for i in obj)
    elif isinstance(obj, BaseModel):
        return freeze_data(obj.dict())
    elif isinstance(obj, Hashable):
        return obj
    raise ValueError(obj)