Pythonクラスのデコレータの引数

Question

Pythonでクラスのデコレータにオプションの引数を渡そうとしています。私は現在持っているコードの下 ：

class Cache(object): 
    def __init__(self, function, max_hits=10, timeout=5): 
     self.function = function 
     self.max_hits = max_hits 
     self.timeout = timeout 
     self.cache = {} 

    def __call__(self, *args): 
     # Here the code returning the correct thing. 


@Cache 
def double(x): 
    return x * 2 

@Cache(max_hits=100, timeout=50) 
def double(x): 
    return x * 2 

デフォルトの1（私の__init__機能でmax_hits=10, timeout=5）を上書きする引数と第二デコレータは、動作していないと私は例外TypeError: __init__() takes at least 2 arguments (3 given)を得ました。私は多くのソリューションを試して記事を読んでいましたが、ここではまだそれを動作させることはできません。

これを解決する方法はありますか？ありがとう！

Answer 1

@Cache(max_hits=100, timeout=50)__init__(max_hits=100, timeout=50)を呼び出しているため、function引数を満たしていません。

関数が存在するかどうかを検出するラッパーメソッドを使用してデコレータを実装できます。関数を見つけると、Cacheオブジェクトを返すことができます。それ以外の場合は、デコレータとして使用されるラッパー関数を返すことができます。

class _Cache(object): 
    def __init__(self, function, max_hits=10, timeout=5): 
     self.function = function 
     self.max_hits = max_hits 
     self.timeout = timeout 
     self.cache = {} 

    def __call__(self, *args): 
     # Here the code returning the correct thing. 

# wrap _Cache to allow for deferred calling 
def Cache(function=None, max_hits=10, timeout=5): 
    if function: 
     return _Cache(function) 
    else: 
     def wrapper(function): 
      return _Cache(function, max_hits, timeout) 

     return wrapper 

@Cache 
def double(x): 
    return x * 2 

@Cache(max_hits=100, timeout=50) 
def double(x): 
    return x * 2 

Answer 2

@Cache 
def double(...): 
    ... 

@Cache(max_hits=100, timeout=50) 
def double(...): 
    ... 

が

def double(...): 
    ... 
double = Cache(max_hits=100, timeout=50)(double) 

と同等であるが

def double(...): 
    ... 
double=Cache(double) 

と等価ですは、Cache(double)とは非常に異なるセマンティックスを持っています。

Cacheを両方の使用例に対応させることは賢明ではありません。

代わりに、オプションmax_hitsとtimeout引数を取ることができデコレータ・ファクトリーを使用して、デコレータを返すことができます：

class Cache(object): 
    def __init__(self, function, max_hits=10, timeout=5): 
     self.function = function 
     self.max_hits = max_hits 
     self.timeout = timeout 
     self.cache = {} 

    def __call__(self, *args): 
     # Here the code returning the correct thing. 

def cache_hits(max_hits=10, timeout=5): 
    def _cache(function): 
     return Cache(function,max_hits,timeout) 
    return _cache 

@cache_hits() 
def double(x): 
    return x * 2 

@cache_hits(max_hits=100, timeout=50) 
def double(x): 
    return x * 2 

PSを。 Cacheクラスに__init__と__call__以外のメソッドがない場合は、_cache関数内のすべてのコードを移動してCacheを完全に削除することができます。

Answer 3

私はこの質問から多くのことを学びました。ありがとうございます。最初の@Cacheに空の括弧を入れるだけの答えはありませんか？次に、functionパラメータを__call__に移動できます。

class Cache(object): 
    def __init__(self, max_hits=10, timeout=5): 
     self.max_hits = max_hits 
     self.timeout = timeout 
     self.cache = {} 

    def __call__(self, function, *args): 
     # Here the code returning the correct thing. 

@Cache() 
def double(x): 
    return x * 2 

@Cache(max_hits=100, timeout=50) 
def double(x): 
    return x * 2 

私は、このアプローチは、よりシンプルで簡潔だと思いますが。

def cache(max_hits=10, timeout=5): 
    def caching_decorator(fn): 
     def decorated_fn(*args ,**kwargs): 
      # Here the code returning the correct thing. 
     return decorated_fn 
    return decorator 

デコレータを使用しているときは、括弧を忘れてしまった場合は、残念ながらまだ外として、実行時までエラーを得ることはありませんデコレータパラメータは、デコレートしようとしている関数に渡されます。そして、実行時に内部デコレータは文句：

TypeError: caching_decorator() takes exactly 1 argument (0 given).

あなたはデコレータのパラメータは呼び出し可能になるだろうことはありません知っている場合は、これをキャッチすることができますしかし：

def cache(max_hits=10, timeout=5): 
    assert not callable(max_hits), "@cache passed a callable - did you forget to parenthesize?" 
    def caching_decorator(fn): 
     def decorated_fn(*args ,**kwargs): 
      # Here the code returning the correct thing. 
     return decorated_fn 
    return decorator 

をあなたが今しようとした場合：

@cache 
def some_method() 
    pass 

宣言にはAssertionErrorが表示されます。

合計で、この記事では、装飾するクラスではなく、クラスをデコレートするデコレータを探していました。他の誰かがそうであれば、this questionが便利です。