"""Utilities for diot"""
from typing import Any, Iterable, cast
class DiotFrozenError(Exception):DOCS
"""When try to modify a frozen diot"""
def nest(DOCS
value: Any,
types: Iterable[type],
dest_type: type,
frozen: bool,
) -> Any:
"""Convert values with certain types recursively"""
# nothing to convert
if not types or not isinstance(value, tuple(types)): # type: ignore
return value
if (list in types and isinstance(value, list)) or ( # type: ignore
tuple in types and isinstance(value, tuple) # type: ignore
):
# use value.__class__ to keep user-subclassed list or tuple
out = [nest(val, types, dest_type, frozen) for val in value] # type: ignore
try:
return value.__class__(out) # type: ignore[call-arg, operator]
except Exception: # pragma: no cover
return out
if dict in types and isinstance(value, dict): # type: ignore
dct = cast(dict[Any, Any], value)
if issubclass(dct.__class__, dest_type):
return dct
return dest_type(
[
(key, nest(val, types, dest_type, frozen))
for key, val in dct.items()
]
)
return value
def to_dict(value: Any) -> Any:DOCS
"""Convert converted Diot objects back to dict"""
if isinstance(value, dict):
d = cast(dict[Any, Any], value)
return {key: to_dict(val) for key, val in d.items()}
if isinstance(value, tuple):
return tuple(to_dict(val) for val in cast(tuple[Any, ...], value))
if isinstance(value, list):
return [to_dict(val) for val in cast(list[Any], value)]
return value