coordt · May 8, 2023 18:00 · May 8, 2023
diff --git a/immutable.py b/immutable.py
@@ -0,0 +1,26 @@
+"""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)