Элегантный способ использования ресурса в Python?

Question

Я хочу найти элегантный способ сделать следующее:

try: 
    with some_resource: 
     # got it 
     do_something() 

except ResourceUnavailableError: 
    # didn't get it 
    do_something_else() 

This:

1. пытается получить ресурс
2. на успех, он делает что-то, а затем освобождает ресурс правильно
3. на провал, он делает что-то другое

Хотя не многие строки кода, я нахожу, что если мне придется писать это во многих разных местах, это не очень элегантно.

Я почти жаль, что я мог бы написать что-то следующим образом (я знаю, что это не код реального Python):

with some_resource: 
    # got it 
    do_something() 

# an optional else 
else: 
    # didn't get it 
    do_something_else() 

Кто-нибудь знает, если это возможно, чтобы написать что-то вдоль этих линий в Python? Спасибо!

Answer 1

Нет, with не имеет else блок. Но вы можете написать свой собственный менеджер контекста, который обрабатывает else части:

from contextlib import contextmanager 

@contextmanager 
def handle_unavailable(resource, exception, exception_handler): 
    try: 
     with resource: 
      yield resource 
    except exception: 
     exception_handler() 

и использовать его, как это везде так:

with handle_unavailable(some_resource, ResourceUnavailableError, do_something_else): 
    do_something() 

Итак, если ResourceUnavailableError поднят в with блоке, do_something_else вызываются для вы.

Это использует @contextlib.contextmanager() decorator, что делает запись ваших собственных менеджеров контекста практически тривиальной.