哪个 python 静态检查器可以捕获“忘记等待”问题？

Question

代码：

from typing import AsyncIterable

import asyncio


async def agen() -> AsyncIterable[str]:
    print('agen start')
    yield '1'
    yield '2'


async def agenmaker() -> AsyncIterable[str]:
    print('agenmaker start')
    return agen()


async def amain():
    print('amain')
    async for item in agen():
        pass
    async for item in await agenmaker():
        pass
    # Error:
    async for item in agenmaker():
        pass


def main():
    asyncio.get_event_loop().run_until_complete(amain())


if __name__ == '__main__':
    main()

正如您所看到的，它带有类型注释，并且包含一个容易错过的错误。

但是，既没有pylint也没有mypy发现该错误。

除了单元测试之外，还有哪些选项可以捕获此类错误？

Answer 1

MyPy 完全有能力发现这个问题。问题是没有检查未注释的函数。将有问题的函数注释为-> None，它会被正确检查并拒绝。

# annotated with return type
async def amain() -> None:
    print('amain')
    async for item in agen():
        pass
    async for item in await agenmaker():
        pass
    # Error:
    async for item in agenmaker(): # error: "Coroutine[Any, Any, AsyncIterable[str]]" has no attribute "__aiter__" (not async iterable)

        pass

如果您想消除此类问题，请使用标志--disallow-untyped-defs或--check-untyped-defs。

MyPy：函数签名和动态与静态类型

没有类型注释的函数被认为是由 mypy 动态类型化的：

def greeting(name):
    return 'Hello ' + name

默认情况下，mypy 不会对动态类型函数进行类型检查。这意味着，除了少数例外，mypy 不会报告常规未注释 Python 的任何错误。