带参数的 Python 装饰器

17 Mar 2025 | 4 分钟阅读

在本教程中，我们将讨论 Python 中的带参数装饰器，但在开始此主题之前，用户必须学习 Python 中的装饰器和函数装饰器。

装饰器是 Python 中一个非常强大且有用的工具，因为它可以让用户修改函数或类的行为。

Python 函数可以被视为头等公民，就像对象一样。

函数可以被引用给一个变量。
函数可以作为参数传递给其他函数。
函数可以从函数中返回。

带参数的装饰器就像普通的装饰器一样。

语法

@decorator(params)
def function_name():
	'''Function implementation'''

带参数装饰器的代码实现

def function_name():
	'''Function implementation'''
function_name = (decorator(params))(function_name)
"""

当我们执行此代码时，执行将从左到右开始，它将调用 decorator(params) 来返回函数对象 func_obj。然后 func_obj(function_name) 将通过使用 func_obj 被调用。在内部函数中，将执行所需的操作，并将实际函数的引用返回以分配给 function_name。现在，用户可以使用 function_name() 来调用已应用装饰器的函数。

如何实现带参数的装饰器

首先，我们将看到如果我们直接运行参数代码而不实现任何值，我们可以得到什么输出。

def decorators_1(*args, **kwargs):
    def inner_1(func_1):
        '''
           doing operations with func_1
        '''
        return func_1
    return inner_1 # this is the function_object mentioned in the above content
 
@decorators_1(params)
def func_1():
    """
         function implementation
    """

在这里，在上面的代码中，由于 params 为空，我们可能会遇到一些错误。

让我们一步一步来理解

def decorator_function(function_name):
  print("Inside the Decorator: ")
 
  def inner_1(*args, **kwargs):
    print("Inside the Inner Function: ")
    print("'Decorated the function'")
    # perform this operations with function_name
     
    function_name()
     
  return inner_1
 
@decorator_function
def function_to():
    print("Inside the actual function")
 
function_to()

输出

Inside the Decorator: 
Inside the Inner Function: 
'Decorated the function'
Inside the actual function

代码执行的可视化表示

装饰器执行内部

内部函数执行内部

装饰器函数执行

最终输出执行

在上面的代码中，我们将获得通过带参数装饰器调用的函数的输出。

替代方法

在下面的代码中，我们将看到如何以替代方式编写使用装饰器的函数代码。

def decorator_fun(function_name):
  print ("Inside the Decorator: ")
 
  def inner_1(*args, **kwargs):
    print ("Inside the Inner Function: ")
    print ("'Decorated the function'")
    # Perform this operations with function_name
     
    function_name()
     
  return inner_1
 
def function_to():
    print ("Inside the actual function")
 
# This is another way of using decorators
decorator_fun(function_to)()

输出

Inside the decorator
Inside the inner function
Decorated the function
Inside the actual function

现在我们将看到使用参数装饰器的不同示例，以便更好地理解该概念。

示例 1

def decorator_1(*args, **kwargs):
    print("Inside the Decorator")
     
    def inner_1(function_1):
         
        # Here, we will see the functionality of the code:
        print ("Inside the inner function")
        print ("I am studying ", kwargs['JTP'])
         
        function_1()
         
    # Returning the inner function   
    return inner_1
 
@decorator_1(JTP = "COMPUTER SCIENCE AND ENGINEERING ")
def my_function():
    print ("Inside the actual function")

输出

Inside the Decorator
Inside the inner function
I am studying COMPUTER SCIENCE AND ENGINEERING 
Inside the actual function

代码执行的可视化表示

最终输出执行

示例 2

def decorator_function(A, B):
 
    def Inner_1(function_1):
 
        def wrapper_1(*args, **kwargs):
            print ("I am studying COMPUTER SCIENCE AND ENGINEERING ")
            print ("Summation of values - {}".format(A + B) )
 
            function_1(*args, **kwargs)
             
        return wrapper_1
    return Inner_1
 
 
# here, we are not using decorator
def my_function(*args):
    for ele in args:
        print (ele)
 
# another way of using decorators
decorator_function(22, 14)(my_function)('Computer', 'Science', 'and', 'Engineering')

输出

I am studying COMPUTER SCIENCE AND ENGINEERING 
Summation of values - 36
Computer
Science
and
Engineering

上面的示例也表明，内部函数可以访问外部函数的参数。

代码执行的可视化表示

示例 3

def deco_decorator(dataType, message_1, message_2):
    def decorator_1(function_1):
        print (message_1)
        def wrapper_1(*args, **kwargs):
            print(message_2)
            if all([type(arg) == dataType for arg in args]):
                return function_1(*args, **kwargs)
            return "Invalid Input"
        return wrapper_1
    return decorator_1
 
 
@deco_decorator(str, "Decorator for 'string_Join'", "stringJoin process started ...")
def string_Join(*args):
    st1 = ''
    for K in args:
        st1 += K
    return st1
 
 
@deco_decorator(int, "Decorator for 'summation_1'\n", "summation process started ...")
def summation_1(*args):
    summ1 = 0
    for arg in args:
        summ1 += arg
    return summ1
 
 
print (string_Join("I ", 'am ', "studying ", 'Computer ', "Science ", "and ", "Engineering"))
print ()
print ("The sum is equal to: ", summation_1(22, 12, 48, 133, 627, 181, 219))

输出

Decorator for 'string_Join'
Decorator for 'summation_1'

stringJoin process started ...
I am studying Computer Science and Engineering

summation process started ...
The sum is equal to: 1242