小编HHC*_*HHC的帖子

def foo():
    while <condition>:
        do something

def bar():
    for i in range(5):
        do something

假设我在一个文件名中定义了两个函数test.py。python 有没有办法编写具有以下行为的函数？

import test

def uses_while(fn: Callable) -> bool:
    (what goes here?)

>>> uses_while(test.foo)
True
>>> uses_while(test.bar)
False

我本质上需要以编程方式检查函数是否使用 while 循环，而不需要手动检查代码。我想过使用 pdb.getsourcelines() ，但是如果里面有注释或字符串中包含“while”一词，那么这不起作用。有任何想法吗？