Python 类型提示 - 为 dict 子类指定键、值类型

Question

说我有

class A(dict): pass

有没有办法指定 A 的键和值的类型？是否可以以继承类型的方式完成 - 所以class B(A) 将继承类型值，例如，键并能够覆盖值的类型？

理想情况下，这将通过pep-0484 中引入的类型提示来完成 - 但请注意我在 python 2 上，所以我需要一个类型为 cmets 的解决方案。但是，如果这是不可能的，那么涉及元类或任何其他黑客的解决方案是可以接受的。

Answer 1

与您的类的所有类型交互都通过方法进行，因此请对这些方法进行注释。 A.__setitem__(self, key, value): 被调用来设置一个键值对并且可以被注解来指示期望的类型。

任何有效的 PEP 484 注释都可以在这里使用，包括类型 cmets 如果您需要 Python 2 兼容性。

除非 B 覆盖方法，否则这些注释将被继承到 B。

Answer 2

详细说明@MartijnPieters answer（我使用的是 Pycharm 5.0.4，因此以下行为可能是由于标准或 IDE 错误/功能所致）

_key_type_global = unicode

class A(dict):

    _key_type = unicode

    def __setitem__(self, key, value):
        """A._key_type is not recognised, _key_type_global is
        :type key: _key_type_global
        :type value: int
        """
        super(A, self).__setitem__(key, value)

class B(A): pass


a = A()
a['str'] = 'uy' # expected type unicode got str instead
a[u'str'] = 'uy' # no warn (?)
a[u'str'] = 1 # no warn

val = a[u'str'] # Pycharm does not give any type info on value so I have to 
# define __getitem__ too

# type info is inherited
b = B()
b['str'] = 'uy' # expected type unicode got str instead
b[u'str'] = 'uy' # no warn (?)
b[u'str'] = 1 # no warn

因此需要覆盖所有方法（键值类型信息是各个方法的本地信息）。 IOW，不像参数或属性等可以做的情况：

class A(object):

    def __init__(self, dct):
        self.dct = dct # type : dict[unicode, int]

这是一个 NOOP：

class A(dict): # type dict[unicode, int]

此外，似乎无法将类型指定为可以在子类中轻松覆盖的类变量。