🔒Открыть пост
и посмотреть медиа
Знаете ли вы, что один True может скрыть исключение внутри with?
При выходе из with Python вызывает __exit__ и передает туда сведения об исключении, если оно возникло. Если исключение было, то ложное значение, включая None, пропускает исключение дальше; истинное сообщает Python, что оно обработано. При обычном выходе все три аргумента __exit__ равны None, а возвращаемое значение ни на что не влияет:
class Ignore:
def __enter__(self):
return self
def __exit__(self, exc_type, exc, tb):
return True
with Ignore():
1 / 0
print("эта строка не выполнится")
print("а эта выполнится")
# а эта выполнится
ZeroDivisionError возникает и прерывает выполнение тела блока – print внутри with не выполняется. Но __exit__ вернул True, поэтому исключение не выходит за пределы блока, а traceback необработанного исключения не выводится.
Еще __exit__ может случайно вернуть истинное значение, пробросив результат другого метода:
class Connection:
def __enter__(self):
return self
def close(self):
print("соединение закрыто")
return True
def __exit__(self, exc_type, exc, tb):
return self.close()
with Connection():
raise RuntimeError("запрос не прошел")
print("выполнение продолжилось")
# соединение закрыто
# выполнение продолжилось
Допустим, метод close() в этом API возвращает True как признак успешного закрытия. Но __exit__ передает этот результат дальше, и Python считает его обработанным исключением. Достаточно написать self.close() без return, и RuntimeError дойдет до вызывающего кода.
Осознанное подавление выглядит так же, только с явной проверкой типа исключения:
class SuppressZero:
def __enter__(self):
return self
def __exit__(self, exc_type, exc, tb):
return isinstance(exc, ZeroDivisionError)
with SuppressZero():
1 / 0
print("выполнение продолжилось")
with SuppressZero():
int("текст")
# ValueError: invalid literal for int() with base 10: 'текст'
Деление на ноль гасится, а ValueError распространяется дальше, потому что для него __exit__ вернул False. По тому же протоколу работает contextlib.suppress из стандартной библиотеки: он подавляет указанные классы исключений и их подклассы.
Возвращаемое значение __exit__ решает, выйдет ли возникшее исключение за пределы блока with: любое истинное значение останавливает его распространение. Если __exit__ не должен подавлять ошибки, заканчивайте метод без return или явно возвращайте False. Для намеренного локального подавления используйте contextlib.suppress(...), перечисляя только ожидаемые типы исключений.
Код на салфетке × Кусочки кода
