Debugowanie, w którym zwracana jest metoda Brak

Question

Mam dużą starszą metodę Pythona, która zawiera około dwudziestu return instrukcji. Ta metoda nie powinna zwracać None, ale robi to. Jest powtarzalny w prostym przypadku testowym.

Do tej pory użyłem debuggera i przeszedłem przez linię kodu po linii, aby znaleźć zgodną instrukcję return.

Ale czy jest łatwiejszy sposób?

Czy istnieje sposób podniesienia wyjątku, gdy tylko metoda zwróci None?

i Oczywiście muszę zobaczyć wiersz zawierający instrukcję return.

przykład:

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


assert not big_method(True, None) is None 

Powyżej prosty fragment kodu. Rezultat:

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 

Przede traceback nie pomaga bardzo dużo, ponieważ chcę zobaczyć linię wewnątrzbig_method(). W powyższym przykładzie chcę zobaczyć, który z nich oznaczono jako <------.

Używam PyCharm, ale czysty pyton lub inne rozwiązanie jest mile widziane.

Tylko dla rekordów. Istnieje kolejne pytanie, które próbuje włączyć tę funkcję w PyCharm: PyCharm: Debugging: r(eturn) Continue execution until the current function returns

Answer 1

PDB ma r(eturn) polecenia dla tego potrzebował:

r (eturn) Kontynuuj realizację aż prąd powraca funkcyjnych.

przykład:

> /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 

zobaczyć bardziej szczegółowo w pdb doc.

Answer 2

jeśli możesz zmienić kod źródłowy, to będzie to łatwe. Ale pozwoli ci tylko uzyskać numer linii i musi się zmienić wszystkie oświadczenie zwrotne.

#!/usr/bin/env python 
import os 
import sys 
import inspect 
def get_head_info(): 
    try: 
     raise Exception 
    except: 
     f = sys.exc_info()[2].tb_frame.f_back 
    return '%s, %s' % (f.f_code.co_name, str(f.f_lineno)) 

def big_method(arg1, some_var): 
    if arg1: 
     return (print(get_head_info()), some_var) 
    return (print(get_head_info()), True) 
if __name__ == '__main__': 
    assert not big_method(True, None)[1] is None 

w python3.5 na win10, będzie się

big_method, 17 
Traceback (most recent call last): 
    File "C:\test.py", line 21, in <module> 
    assert not big_method(True, None)[1] is None 
AssertionError 

return (print(get_head_info()), some_var) to punkt, gdy spust był powrót, to najpierw ocenić oświadczenie druku, wydrukować numer wiersza (w tym sprawa 17), następnie uzyskaj wartość Brak, w końcu otrzymasz (None, some_var).

Answer 3

Dla mnie najprościej jest użyć dekoratora, który sprawdzi wyjście twojej funkcji.Mały przykład:

def check_none_in_response(func): 
    def wrapper(arg1, some_var): 
     rs = func(arg1, some_var) 
     assert rs is not None 
     return rs 

    return wrapper 

Zastosowanie:

@check_none_in_response 
<your method> 

Hope, to pomoże.