Jak radzić sobie z AssertionError w Pythonie i dowiedzieć się, na którym wierszu lub instrukcji wystąpił?

Question

Chcę obsługiwać AssertionError s zarówno w celu ukrycia niepotrzebnych części śladu stosu od użytkownika, jak i wydrukowania komunikatu, dlaczego błąd wystąpił i co powinien zrobić użytkownik.

Czy istnieje sposób sprawdzenia, w którym wierszu lub oświadczeniu nie powiodło się assert w bloku except?

try: 
    assert True 
    assert 7 == 7 
    assert 1 == 2 
    # many more statements like this 
except AssertionError: 
    print 'Houston, we have a problem.' 
    print 
    print 'An error occurred on line ???? in statement ???' 
    exit(1) 

nie chcę mieć do tego dodać do każdego rachunku assert:

assert 7 == 7, "7 == 7" 

ponieważ powtarza informacje.

Answer 1

pomocą modułu traceback:

import sys 
import traceback 

try: 
    assert True 
    assert 7 == 7 
    assert 1 == 2 
    # many more statements like this 
except AssertionError: 
    _, _, tb = sys.exc_info() 
    traceback.print_tb(tb) # Fixed format 
    tb_info = traceback.extract_tb(tb) 
    filename, line, func, text = tb_info[-1] 

    print('An error occurred on line {} in statement {}'.format(line, text)) 
    exit(1) 

Answer 2

Moduł ślad błędu i sys.exc_info są przesadą śledzenia źródło wyjątek. To wszystko w domyślnym tracebacku. Więc zamiast wywoływania exit (1) po prostu re-raise:

try: 
    assert "birthday cake" == "ice cream cake", "Should've asked for pie" 
except AssertionError: 
    print 'Houston, we have a problem.' 
    raise 

co daje następujący wynik, który zawiera oświadczenie przestępstwa i numer linia:

Houston, we have a problem. 
Traceback (most recent call last): 
    File "/tmp/poop.py", line 2, in <module> 
    assert "birthday cake" == "ice cream cake", "Should've asked for pie" 
AssertionError: Should've asked for pie 

Podobnie moduł rejestrowania ułatwia zalogować traceback dla dowolnego wyjątku (włączając te, które są przechwytywane i nigdy nie zostały przebudowane):

import logging 

try: 
    assert False == True 
except AssertionError: 
    logging.error("Nothing is real but I can't quit...", exc_info=True)