functools.wraps 데코레이터-파이썬 튜토리얼-php.cn

functools.wraps 데코레이터

高洛峰

풀어 주다： 2016-11-05 14:11:19

원래의

1108명이 탐색했습니다.

랩은 실제로 실용적인 용도가 없습니다.

데코레이터가 함수 func로 인해 발생한 원래 함수 이름의 속성 변경 문제를 해결하는 데 사용됩니다. , 그리고 이때 func는 데코레이터의 장식된 함수를 가리킵니다

import sys

debug_log = sys.stderr

def trace(func):
        if debug_log:
                def callf(*args, **kwargs):
                        """A wrapper function."""
                        debug_log.write(&#39;Calling function: {}\n&#39;.format(func.__name__))
                        res = func(*args, **kwargs)
                        debug_log.write(&#39;Return value: {}\n&#39;.format(res))
                        return res
                return callf
        else:
                return func

@trace
def square(x):
        """Calculate the square of the given number."""
        return x * x

로그인 후 복사

여기의 사각형은 실제로 호출을 가리킵니다. help(square) 또는 square.__name__을 사용하여 할 수 있습니다. 확인해 보세요.

사각형의 장식된 기능은 실제로는 다른 기능입니다(함수명 등 기능속성은 변경됩니다)

장식용으로 랩을 사용하는 경우

def trace(func):
        if debug_log:
          @functools.wraps(func)
                def callf(*args, **kwargs):
                        """A wrapper function."""
                        debug_log.write(&#39;Calling function: {}\n&#39;.format(func.__name__))
                        res = func(*args, **kwargs)
                        debug_log.write(&#39;Return value: {}\n&#39;.format(res))
                        return res
                return callf
        else:
                return func

로그인 후 복사

이때 트레이스로 장식된 사각형의 속성은 변경되지 않습니다.

를 보려면 help(square)를 참조하세요. 이유: 랩 장식 코드를 다음과 같이 번역하며 이에 해당합니다.

def trace(func):
        if debug_log:
                def _callf(*args, **kwargs):
                        """A wrapper function."""
                        debug_log.write(&#39;Calling function: {}\n&#39;.format(func.__name__))
                        res = func(*args, **kwargs)
                        debug_log.write(&#39;Return value: {}\n&#39;.format(res))
                        return res

                callf = functools.update_wrapper(_callf, wrapped = func,assigned = functools.WRAPPER_ASSIGNMENTS,updated = functools.WRAPPER_UPDATES)

                return callf
        else:
                return func

로그인 후 복사

Update_wrapper는 매우 간단한 작업을 수행합니다. 이는 래핑된 매개변수(예: 사각형)로 표시되는 함수 개체의 일부 속성(예: __name__, __doc__)을 사용하여 다음으로 표시되는 함수 개체를 덮어쓰는 것입니다. 매개변수 래퍼(예: callf , 여기서 callf 는 단순히 square 함수를 호출하므로 callf 는 square의 래퍼 함수라고 말할 수 있습니다).

따라서 이 예에서는 Wraps 데코레이터를 사용하여 callf를 "장식"한 후 callf의 __doc__, __name__ 및 기타 속성은 함수 square의 속성과 완전히 동일합니다. 추적으로 "장식"합니다.