Pythonのコンテキストマネージャデコレータする（および逆）

Question

私がしたいと思います：この例では

# Simple example, one could replace try/except by any other nested construct 
def mycontextmanager_generator(foo): 
    try: 
     yield 
    except: 
     print 'bar' 
   raise 

mycontextmanager = build_contextmanager(mycontextmanager_generator) 
mydecorator = build_decorator(mycontextmanager_generator) 


>>> with mycontextmanager(): 
>>>  raise Exception('baz gone bar in context manager') 
... bar 


>>> @mydecorator() 
>>> def bazzer(): 
>>>  raise Exception('baz gone bar in decorator') 
>>> bazzer() 
... bar 

は、Iが生成機能からコンテキストマネージャを構築し、同じ機能のデコレータ。これは私が失敗した方法で達成しようとしたものです。私が欲しいものより一般的

は、DRYになることです：書き込みがtry/exceptブロックたら、再度デコレータとコンテキストマネージャ両方を通してそれを再使用：試みを書き込むことによって/圏除いて一度だけを、ジェネレータ関数またはその他のラッパーに関係なく使用できます。

（PY2でPY3/contextlib2でcontextlib中）ContextDecorator事はクラスでのみ使用可能ですが、その場合には無用のようですが...私は何かが足りないのですか？ __enter__と__exit__を使用して、クラスベースのContextManagerでtry/exceptブロックを実装する方法はありますか？

yield構文で構築されたコンテキストマネージャをデコレータに変換する可能性はありますか？

または逆の（デコレータからコンテキストマネージャ）？

もしそうでない場合は、その点に関してPythonの制限が何であるかを知っておきたいと思います。

私の理解では、yieldの構文はPythonインタプリタとコンテキストの切り替えに非常に厳密に縛られており、その点でその動作を変更できるかどうかはわかりません。

Answer 1

contextmanagerがそのコンテキスト（_GeneratorContextManager）とContextDecoratorクラスの管理に使用するクラスを組み合わせることで、必要なものを簡単に達成できます。例えば。

from contextlib import ContextDecorator, _GeneratorContextManager 
from functools import wraps 

class MyContextManager(_GeneratorContextManager, ContextDecorator): 
    pass 

def contextmanager(func): 
    @wraps(func) 
    def helper(*args, **kwds): 
     return MyContextManager(func, args, kwds) 
    return helper 

@contextmanager 
def print_bar_on_error(): 
    try: 
     yield 
    except: 
     print('bar') 
     raise 

with print_bar_on_error(): 
    raise Exception('baz gone bar in context manager') 

は生成します。

bar 
Traceback (most recent call last): 
    File "run.py", line 28, in <module> 
    raise Exception('baz gone bar in context manager') 
Exception: baz gone bar in context manager 

とデコレータとして

@print_bar_on_error() 
def bazzer(): 
    raise Exception('baz gone bar in decorator') 
bazzer() 

を使用し生成します

bar 
Traceback (most recent call last): 
    File "run.py", line 32, in <module> 
    bazzer() 
    File "c:\Users\User\AppData\Local\Programs\Python\Python35-32\lib\contextlib.py", line 30, in inner 
    return func(*args, **kwds) 
    File "run.py", line 31, in bazzer 
    raise Exception('baz gone bar in decorator') 
Exception: baz gone bar in decorator 
    return func(*args, **kwds) 
Exception: baz gone bar in decorator 

Answer 2

ないとはいえ、砂丘よりもソリューションを理解することが単純を活用する二重構文。

import contextlib 
import functools 

def handler(): 
    try: 
     yield 
    except: 
     print 'bar' 


my_contextmanager = contextlib.contextmanager(handler) 


def my_decorator(func): 
    @functools.wraps(func) 
    def wrapper(*args, **kwargs): 
     with my_contextmanager(): 
      func(*args, **kwargs) 
    return wrapper 


with my_contextmanager(): 
    raise Exception('baz') 

@my_decorator 
def f(): 
    raise Exception('baz') 

f() 

ができます：

bar 
bar