调试where方法返回None

Question

我有一个很大的遗留Python方法，它包含大约20个return语句。

该方法不应返回None但它会返回None 。 它在一个简单的测试用例中是可重复的。

到目前为止，我使用了一个调试器并逐行遍历代码以找到匹配的return语句。

但是有更简单的方法吗？

一旦方法返回None，有没有办法提出异常？

当然我需要看到包含return语句的行。

例：

def big_method(arg1, some_var):
    #.... many returns
    if arg1:
        return some_var # <------
    #... many returns


assert not big_method(True, None) is None

上面是一个简单的代码片段。 结果：

Traceback (most recent call last):
  File "/home/modwork_vums_d/src/manyreturns.py", line 8, in <module>
    assert not big_method(True, None) is None
AssertionError

上面的回溯并没有多大帮助，因为我想在 big_method()看到这一行。 在上面的例子中，我想看看哪个标有<------ 。

我使用PyCharm，但欢迎使用纯Python或其他解决方案。

仅供记录。 有一个后续问题试图在PyCharm中启用此功能： PyCharm：调试：r（eturn）继续执行，直到当前函数返回

Answer 1

pdb有一个r(eturn)命令来满足这个需求：

r（eturn）继续执行，直到当前函数返回。

例：

> /Users/georgexsh/wasteland/tmp/app.py(6)<module>()
-> assert not big_method(True, None) is None
(Pdb) s
--Call--
> /Users/georgexsh/wasteland/tmp/app.py(1)big_method()
-> def big_method(arg1, some_var):
(Pdb) r
--Return--
> /Users/georgexsh/wasteland/tmp/app.py(3)big_method()->None
-> return some_var

在pdb doc中查看更多细节。