Как узнать переменные при ошибке в коде?

Question

[Андрей Сальников]

Скажем у меня есть код:

def b(a):
    d = int(a)
    return d * 6

Я вызываю эту функцию так:

try:
    r = b('asd')
except Exception as e:
    print(e)

Вопрос - как мне узнать переменные внутри функции при ошибке?
Код я написал для примера. У меня код скачивает файл json из интернета с ним работает и т.п. Но т.к. структура у файла иногда меняется мне нужно отслеживать ошибки и что-где поменялось, в частности знать при каких новых данных у меня ошибка.

Comments

[Андрей Сальников]

javedimka, Имеете в виду через pdb.set_trace останавливать код и т.п.? Если так - не могу. Код должен выполняться полностью, а в случае ошибки должны выводиться переменные.

[javedimka]

Андрей Сальников, я не так понял проблему потому и удалил коммент

[javedimka]

Андрей Сальников, Как лучше всего обрабатывать ошибки в python?

[Андрей Сальников]

javedimka, За ссылку спасибо.

Answer 1

[Егор Стаховский]

Попробуйте вот такое решение. Код для Python 3. Основано на setprofile, так что может вам не подойти, если ваш код многопоточный.

import sys
import logging


LOG = logging.getLogger(__name__)


def handle_exception_with_locals(func):
    def wrapped(*args, **kwargs):
        locals = {}
        def tracer(frame, event, arg):
            """ Эта функция будет инспектировать код,
                который будет выполнятся после того, 
                как она будет установлена профайлером.
                Профайлер срабатывает только на события
                retrun и call (при этом exception интерпретируется как return)
            """
            nonlocal locals
            if event == 'return':
                locals = frame.f_locals.copy()
        sys.setprofile(tracer)
        try:
            res = func(*args, **kwargs)
        except Exception as e:
            log_pattern = ('{}(*{}, **{}) failed with {}. ' + 
                           'Local function variables: {}; ' +
                           'Global variables: {};')
            LOG.error(log_pattern.format(
                func.__name__, repr(args), repr(kwargs),
                repr(e), repr(locals), repr(globals()))
            )
            raise
        finally:
            sys.setprofile(None)
        return res
    return wrapped


@handle_exception_with_locals
def test(a, b):
    a = 16
    b = 7
    c = 9
    raise ValueError('Oops!')


test(1, 2)