使用不可变键但可变值定义python字典

时间:2013-02-11 16:25:31

标签: python dictionary key immutability

嗯,问题在于标题:如何定义具有不可变键但可变值的python字典?我想出了这个(在python 2.x中):

class FixedDict(dict):
    """
    A dictionary with a fixed set of keys
    """

    def __init__(self, dictionary):
        dict.__init__(self)
        for key in dictionary.keys():
            dict.__setitem__(self, key, dictionary[key])

    def __setitem__(self, key, item):
        if key not in self:
            raise KeyError("The key '" +key+"' is not defined")
        dict.__setitem__(self, key, item)

但它看起来(不出所料)相当草率。特别是,这是安全还是存在实际更改/添加某些密钥的风险,因为我是从dict继承的? 感谢。

3 个答案:

答案 0 :(得分:14)

dict直接继承的问题在于,很难遵守完整的dict合同(例如,在您的情况下,update方法不会在一致的方式)。

你想要的是扩展collections.MutableMapping

import collections

class FixedDict(collections.MutableMapping):
    def __init__(self, data):
        self.__data = data

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

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

    def __setitem__(self, k, v):
        if k not in self.__data:
            raise KeyError(k)

        self.__data[k] = v

    def __delitem__(self, k):
        raise NotImplementedError

    def __getitem__(self, k):
        return self.__data[k]

    def __contains__(self, k):
        return k in self.__data

请注意,原始(包装)字典将被修改,如果您不希望这种情况发生,请使用copy or deepcopy

答案 1 :(得分:13)

考虑代理dict而不是子类化它。这意味着只允许您定义的方法,而不是回退到dict的实现。

class FixedDict(object):
        def __init__(self, dictionary):
            self._dictionary = dictionary
        def __setitem__(self, key, item):
                if key not in self._dictionary:
                    raise KeyError("The key {} is not defined.".format(key))
                self._dictionary[key] = item
        def __getitem__(self, key):
            return self._dictionary[key]

此外,您应该使用字符串格式而不是+来生成错误消息,否则它将因任何不是字符串的值而崩溃。

答案 2 :(得分:1)

如何阻止某人添加新密钥完全取决于某人可能尝试添加新密钥的原因。正如评论所述,修改密钥的大多数字典方法都不会通过__setitem__,因此.update()调用会添加新密钥。

如果您只希望某人使用d[new_key] = v,那么您的__setitem__就可以了。如果他们可能使用其他方式添加密钥,那么您必须投入更多工作。当然,无论如何,他们总是可以这样做:

dict.__setitem__(d, new_key, v)

你不能在Python中使事物真正不可变,你只能停止特定的改变。