1. ホーム
  2. python

[解決済み] Python functools.wrapsのクラスと同等の機能

2023-07-14 16:38:16

質問

クラスを使用してデコレータを定義する場合、どのようにすれば自動的に __name__ , __module____doc__ ? 通常、私はfunctoolsの@wrapsデコレータを使用します。 以下は、私が代わりにクラスに対して行ったことです(これは完全に私のコードではありません)。

class memoized:
    """Decorator that caches a function's return value each time it is called.
    If called later with the same arguments, the cached value is returned, and
    not re-evaluated.
    """
    def __init__(self, func):
        super().__init__()
        self.func = func
        self.cache = {}

    def __call__(self, *args):
        try:
            return self.cache[args]
        except KeyError:
            value = self.func(*args)
            self.cache[args] = value
            return value
        except TypeError:
            # uncacheable -- for instance, passing a list as an argument.
            # Better to not cache than to blow up entirely.
            return self.func(*args)

    def __repr__(self):
        return self.func.__repr__()

    def __get__(self, obj, objtype):
        return functools.partial(self.__call__, obj)

    __doc__ = property(lambda self:self.func.__doc__)
    __module__ = property(lambda self:self.func.__module__)
    __name__ = property(lambda self:self.func.__name__)

ネームモジュールとdocの作成を自動化するための標準的なデコレータはありますか? また、getメソッドを自動化するために(バインドメソッドを作成するためだと思いますが)足りないメソッドはありませんか?

どのように解決するのですか?

皆さん、明らかな解決策を見逃しているようです。

>>> import functools
>>> class memoized(object):
    """Decorator that caches a function's return value each time it is called.
    If called later with the same arguments, the cached value is returned, and
    not re-evaluated.
    """
    def __init__(self, func):
        self.func = func
        self.cache = {}
        functools.update_wrapper(self, func)  ## TA-DA! ##
    def __call__(self, *args):
        pass  # Not needed for this demo.

>>> @memoized
def fibonacci(n):
    """fibonacci docstring"""
    pass  # Not needed for this demo.

>>> fibonacci
<__main__.memoized object at 0x0156DE30>
>>> fibonacci.__name__
'fibonacci'
>>> fibonacci.__doc__
'fibonacci docstring'