让 mypy 识别该对象具有属性

Question

例如：

from typing import Union, List, TypeVar

foo: object

if hasattr(foo, 'bar'):
    print(foo.bar)

返回

main.py:6: error: "object" has no attribute "bar"
Found 1 error in 1 file (checked 1 source file)

但是，我们知道foo 具有属性bar，因为我们只是断言它 - 有没有办法告诉 mypy 这个？

Answer 1

您可以使用Protocol 而不是hasattr 来做到这一点：

from typing import Protocol, runtime_checkable


@runtime_checkable
class HasBar(Protocol):
    bar: int


foo: object

if isinstance(foo, HasBar):
    print(foo.bar)