Скрыть вызовы после операции с помощью оператора

Question

У меня есть функция f, которая возвращает два параметра. После этой функции я использую первый параметр. Мне нужен второй, чтобы склеить вещи вместе с другой функцией g. Так что мой код будет выглядеть следующим образом:

a, b = f() 
# do stuff with a 
g(b) 

Это очень повторяющееся, поэтому я полагал, что я мог бы использовать что-то вроде RAII подхода. Но, так как я не знаю, когда объекты будут умирать и моя главная цель состоит в том, чтобы избавиться от повторяющегося кода, я использовал with заявление:

with F() as a: 
    # do stuff with a 

Вот так. Я в основном создал объект F вокруг этой функции, поставляя функции __enter__ и __exit__ (и, очевидно, __init__).

Тем не менее, я все еще задаюсь вопросом, является ли это правильным "Pythonic" способом справиться с этим, поскольку инструкция with предназначена для обработки исключений. Особенно это функция __exit__ имеет три аргумента, которые я не использую в данный момент.

Редактировать (дальнейшие пояснения): Всякий раз, когда я звоню f() Мне НУЖНО что-то делать с b. Неважно, что произошло между ними. Я выразил это как g(b). И именно это я скрываю в заявлении. Поэтому программисту не нужно вводить g(b) снова и снова после каждого звонка f(), который может стать очень запутанным и запутанным, поскольку #do stuff with a может затянуться.

Answer 1

Чуть более вещий способ сделать это было бы использовать contextlib модуль. Вы можете отказаться от развертывания собственного класса с функциями __enter__ и __exit__.

from contextlib import contextmanager 

def g(x): 
    print "printing g(..):", x 

def f(): 
    return "CORE", "extra" 

@contextmanager 
def wrap(): 
    a, b = f() 
    try: 
     yield a 
    except Exception as e: 
     print "Exception occurred", e 
    finally: 
     g(b) 
if __name__ == '__main__': 
    with wrap() as x: 
     print "printing f(..):", x 
     raise Exception() 

Выход:

$ python temp.py 
printing f(..): CORE 
Exception occurred 
printing g(..): extra 

Answer 2

Для кого-то, кто читает ваш код (или вы через 6 месяцев), менеджер контекста, вероятно, будет подразумевать создание/инициализацию и закрытие/удаление ресурса, и я был бы очень осторожен в повторном назначении идиомы.

Вы можете использовать композицию, а не менеджером контекста:

def func(a, b): 
    # do stuff with a & optionally b 

def new_f(do_stuff): 
    a, b = f() 
    do_stuff(a, b) 
    g(b) 


new_f(func)