22

✏️ Editing: immutabletypes.py

"""
    :codeauthor: Pedro Algarvio (pedro@algarvio.me)

salt.utils.immutabletypes
    ~~~~~~~~~~~~~~~~~~~~~~~~~

Immutable types
"""

import copy
from collections.abc import Mapping, Sequence, Set

class ImmutableDict(Mapping):
    """
    An immutable dictionary implementation
    """

def __init__(self, obj):
        self.__obj = obj

def __len__(self):
        return len(self.__obj)

def __iter__(self):
        return iter(self.__obj)

def __getitem__(self, key):
        return freeze(self.__obj[key])

def __repr__(self):
        return f"<{self.__class__.__name__} {repr(self.__obj)}>"

def __deepcopy__(self, memo):
        return copy.deepcopy(self.__obj)

def copy(self):
        """
        Return an un-frozen copy of self
        """
        return copy.deepcopy(self.__obj)

class ImmutableList(Sequence):
    """
    An immutable list implementation
    """

def __init__(self, obj):
        self.__obj = obj

def __len__(self):
        return len(self.__obj)

def __iter__(self):
        return iter(self.__obj)

def __add__(self, other):
        return self.__obj + other

def __radd__(self, other):
        return other + self.__obj

def __getitem__(self, key):
        return freeze(self.__obj[key])

def __repr__(self):
        return f"<{self.__class__.__name__} {repr(self.__obj)}>"

def __deepcopy__(self, memo):
        return copy.deepcopy(self.__obj)

def copy(self):
        """
        Return an un-frozen copy of self
        """
        return copy.deepcopy(self.__obj)

class ImmutableSet(Set):
    """
    An immutable set implementation
    """

def __init__(self, obj):
        self.__obj = obj

def __len__(self):
        return len(self.__obj)

def __iter__(self):
        return iter(self.__obj)

def __contains__(self, key):
        return key in self.__obj

def __repr__(self):
        return f"<{self.__class__.__name__} {repr(self.__obj)}>"

def __deepcopy__(self, memo):
        return copy.deepcopy(self.__obj)

def copy(self):
        """
        Return an un-frozen copy of self
        """
        return copy.deepcopy(self.__obj)

def freeze(obj):
    """
    Freeze python types by turning them into immutable structures.
    """
    if isinstance(obj, dict):
        return ImmutableDict(obj)
    if isinstance(obj, list):
        return ImmutableList(obj)
    if isinstance(obj, set):
        return ImmutableSet(obj)
    return obj

MR SKB