python里函数装饰器如何使用？

Python中的函数装饰器是一个非常重要的概念。它可以用来修改现有的函数，或者在不改变现有函数的情况下，给函数添加额外的功能。在这篇文章中，我们将从多个角度来探讨函数装饰器的使用方法。

一、什么是函数装饰器？

装饰器本质上是一个Python函数，它可以接收一个函数作为参数，并返回一个新的函数作为输出。这个新函数通常会在原始函数调用之前或之后进行一些额外的操作。例如，可以使用装饰器来记录函数的执行时间、检查函数的输入参数、缓存函数的输出结果等等。

二、如何定义函数装饰器？

定义一个装饰器需要使用Python的@语法。例如，下面是一个简单的装饰器，它可以记录函数的执行时间：

```

import time

def timer(func):

def wrapper(*args, **kwargs):

start_time = time.time()

result = func(*args, **kwargs)

end_time = time.time()

print("函数 %s 执行时间为 %f 秒" % (func.__name__, end_time - start_time))

return result

return wrapper

@timer

def my_function():

time.sleep(2)

```

这个装饰器定义了一个timer函数，它接收一个函数作为参数，并返回一个wrapper函数。wrapper函数会记录函数的执行时间，并在函数执行结束后打印出来。注意，wrapper函数使用了*args和**kwargs参数，这样它可以接收任意数量和任意类型的参数。

三、如何使用函数装饰器？

要使用函数装饰器，只需要在函数定义的上面加上@装饰器名称即可。例如，在上面的例子中，我们用@timer装饰了my_function函数。这样，当我们调用my_function函数时，它会自动执行timer函数，并将my_function函数作为参数传递给它。

四、如何链式调用多个装饰器？

Python允许我们使用多个装饰器来修改一个函数。例如，我们可以定义一个新的装饰器，它可以检查函数的输入参数是否符合要求：

```

def check_input(func):

def wrapper(*args, **kwargs):

if len(args) < 2:

raise ValueError("参数数量不足")

if type(args[0]) != str:

raise TypeError("第一个参数必须是字符串")

if type(args[1]) != int:

raise TypeError("第二个参数必须是整数")

return func(*args, **kwargs)

return wrapper

@check_input

@timer

def my_function(name, age):

print("姓名：%s，年龄：%d" % (name, age))

my_function("张三", 18)

```

这个例子定义了一个新的装饰器check_input，它会检查函数的输入参数是否符合要求。我们将它和之前的timer装饰器一起使用，来对my_function函数进行装饰。注意，我们可以使用@语法来链式调用多个装饰器。

五、如何使用类装饰器？

除了函数装饰器之外，Python还支持类装饰器。类装饰器是一个类，它可以接收一个函数作为参数，并返回一个新的函数作为输出。例如，下面是一个类装饰器的例子，它可以记录函数的执行时间：

```

import time

class Timer:

def __init__(self, func):

self.func = func

def __call__(self, *args, **kwargs):

start_time = time.time()

result = self.func(*args, **kwargs)

end_time = time.time()

print("函数 %s 执行时间为 %f 秒" % (self.func.__name__, end_time - start_time))

return result

@Timer

def my_function():

time.sleep(2)

```

这个类装饰器定义了一个Timer类，它接收一个函数作为参数，并返回一个新的函数。新的函数会记录函数的执行时间，并在函数执行结束后打印出来。注意，Timer类实现了__call__方法，这样它就可以像函数一样被调用。

六、总结

函数装饰器是Python中一个非常重要的概念，它可以用来修改现有的函数，或者在不改变现有函数的情况下，给函数添加额外的功能。我们可以使用@语法来定义和使用函数装饰器，也可以使用类装饰器来实现类似的功能。在实际编程中，函数装饰器常常被用来记录函数的执行时间、检查函数的输入参数、缓存函数的输出结果等等。通过合理地使用函数装饰器，我们可以使我们的代码更加简洁、优雅、易于维护。