с-утверждением, что может пропустить свой блок

Question

Я надоело писать (2.7) код Python, как это:

if "george" in cats and cats["george"] is not None: 
    special_cat = cats["george"] 
    # do something with special_cat 

Так что я написал генератор, который возвращает либо ноль или один объект:

def maybe_get(d, k): 
    if k in d and d[k] is not None: 
     yield d[k] 
    else: 
     return 

Теперь я могу сделать более оригинальный фрагмент кода более компактной, например, так:

for special_cat in maybe_get(cats, "george"): 
    # do something with special_cat 

Использование для цикла для этой цели Лоо ks нечетно. Я бы предпочел использовать контекстный менеджер:

with maybe_get(cats, "george") as special_cat: 
    # do something with special_cat 

Но я не могу понять, какой-либо способ сделать менеджер контекста пропустить блок кода он управляет. (Декодер @contextmanager, например, поднимет RuntimeError, если генератор не даст хотя бы одного значения.) Есть ли трюк, который мне не хватает?

Answer 1

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

Я хотел бы использовать:

special_cat = cats.get("george") 
if special_cat is not None: 
    # do something with special_cat 

Это имеет преимущество в том, читаемый и простой. Любые махинации с генераторами или менеджерами контекста увеличили бы удивление для новых разработчиков вашего кода. И помните: вы являетесь новым сопровождающим через год или два от кодовой базы.