>  기사  >  백엔드 개발  >  일반 데코레이터를 사용할 때 원래 함수 서명을 보존하는 방법은 무엇입니까?

일반 데코레이터를 사용할 때 원래 함수 서명을 보존하는 방법은 무엇입니까?

Barbara Streisand
Barbara Streisand원래의
2024-10-17 16:58:021077검색

How to Preserve Original Function Signatures when Using Generic Decorators?

Preserving Original Function Signatures When Using Generic Decorators

Problem Overview

Decorators can enhance functions with additional functionality, but generic decorators often replace the original function's signature with a wildcard signature like "args, *kwargs." This can be problematic, especially when generating documentation or introspecting function metadata.

Solution 1: Utilizing the decorator Module

The decorator module provides an easy solution. By wrapping the decorator function with decorator.decorator, we can preserve the original function's signature:

<code class="python">import decorator

@decorator.decorator
def args_as_ints(f, *args, **kwargs):
    args = [int(x) for x in args]
    kwargs = dict((k, int(v)) for k, v in kwargs.items())
    return f(*args, **kwargs)

@args_as_ints
def funny_function(x, y, z=3):
    """Computes x*y + 2*z"""
    return x*y + 2*z</code>

This approach maintains the original function's signature:

<code class="python">help(funny_function)
# Help on function funny_function in module __main__:
# 
# funny_function(x, y, z=3)
#     Computes x*y + 2*z</code>

Solution 2: Python 3.4+ Enhancements

In Python 3.4 and later, functools.wraps() automatically preserves the original function signature:

<code class="python">import functools

def args_as_ints(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        args = [int(x) for x in args]
        kwargs = dict((k, int(v)) for k, v in kwargs.items())
        return func(*args, **kwargs)
    return wrapper

@args_as_ints
def funny_function(x, y, z=3):
    """Computes x*y + 2*z"""
    return x*y + 2*z

help(funny_function)
# Help on function funny_function in module __main__:
#
# funny_function(x, y, z=3)
#     Computes x*y + 2*z</code>

However, functools.wraps() did not exhibit this behavior in earlier versions of Python.

Other Approaches

Duplicating the signature in the function's docstring or creating custom decorators for each specific signature are flawed workarounds that introduce duplication and maintenance issues.

Conclusion

By utilizing the decorator module or taking advantage of functools.wraps() in Python 3.4+, decorators can be used generically while preserving the original function signatures, ensuring robust documentation and introspection capabilities.

위 내용은 일반 데코레이터를 사용할 때 원래 함수 서명을 보존하는 방법은 무엇입니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!

성명:
본 글의 내용은 네티즌들의 자발적인 기여로 작성되었으며, 저작권은 원저작자에게 있습니다. 본 사이트는 이에 상응하는 법적 책임을 지지 않습니다. 표절이나 침해가 의심되는 콘텐츠를 발견한 경우 admin@php.cn으로 문의하세요.