时间戳

首页 > 解决方案 > Python / PyCharm 的 __slots__ 类型注释

python - Python / PyCharm 的 __slots__ 类型注释

问题描述

如何为通过定义的属性提供类型注释__slots__？例如对于这个类：

class DiffMatch:
  __slots__ = ["ref_seq_idx", "ref_txt", "hyp_txt", "start_time", "end_time"]

  def __repr__(self):
    return "%s(%s)" % (self.__class__.__name__, ", ".join(["%s=%r" % (a, getattr(self, a)) for a in self.__slots__]))

（如果可能的话，在注释中兼容 Python 2（否则只有 Python 3 很好），并且 PyCharm 可以处理它（实际上这对我来说现在最重要）。）

标签： pythonpycharm

解决方案

__slots__只告诉type()对象为潜在属性腾出空间。列表中的名称本身不是属性。你所拥有的只是一堆描述符。没有值，所以没有类型。

因此，您需要创建实际属性，并且这些属性具有类型注释的所有常规选项。

对于 Python 3.6 及更高版本，使用变量类型注释：

class DiffMatch:
    __slots__ = ["ref_seq_idx", "ref_txt", "hyp_txt", "start_time", "end_time"]

    ref_seq_idx: int
    # ...

或者对于 3.6 之前的版本（包括 Python 2），您必须在可以设置属性的方法上注释类型。否则没有真实的实例属性支持。您可以为此添加一个虚拟方法：

class DiffMatch:
    __slots__ = ["ref_seq_idx", "ref_txt", "hyp_txt", "start_time", "end_time"]

    def __type_hints__(self, ref_seq_idx, ...):
        """Dummy method to annotate the instance attribute types
        # type: (int, ...) -> None
        """
        self.ref_seq_idx = ref_seq_idx
        # ...

其中每个参数的所有类型都列在文档字符串中。如果您__init__的类的方法也涉及所有属性，则不需要虚拟方法。

请注意，您不能为这些设置类默认值，这意味着您不能使用ref_seq_idx = None # type: int(pre-Python 3.6) 或ref_seq_idx: int = None(Python 3.6 and up)；__slots__名称被转换为类上的描述符对象，因此名称已设置。

最后但同样重要的是，我会认真研究一下为您构建这些类型的attrs库。PyCharm 2018.2 最近添加了对该库的支持，因此会自动获取类型信息：

@attr.s(slots=True)
class DiffMatch:
    ref_seq_idx = attr.ib(init=False)  # type: int
    # ...

您将__repr__免费生成您的方法。init=False注释告诉attrs不要在 中包含该名称，__init__此时实例在实例化时根本不会设置该属性。

推荐阅读