Python装饰器

不带参数装饰器（两层）

• 举个栗子

import time

# 不带参数的装饰器
def decorator(func):
	def wrapper(*args, **kwargs):
		print(f'{func.__name__} is running.')
		result = func(*args, **kwargs)
		return result
	return wrapper

# 例子：测量函数运行时间
def decorator(func):
	def wrapper(*args, **kwargs):
		start_time = time.time()
		result = func(*args, **kwargs)
		end_time = time.time()
		print(f'{func.__name__} execution time:{end_time-start_time} seconds.')
		return result
	return wrapper

# 使用装饰器
# 方式一
decorated_square = decorator(square)
decorated_square(10)
# 或
square = decorator(square)
square(10)

# 方式二
@decorator
def square(x):
	return x * x
square(10)

带参数的装饰器

import time

# 带参数的装饰器
def transfer_parameters(param):
	def decorator(func):
		def wrapper(*args, **kwargs)
			result = func(*args, **kwargs)
			return result
		return wrapper
	return decorator

# 例子：判断测量的时间有没有超过设置的阈值
def timer(threshold):
	def decorator(func):
		def wrapper(*args, **kwargs):
			start_time = time.time()
			result = func(*args, **kwargs)
			end_time = time.time()
			if end_time - start_time > threshold:
				print(f'{func__name__} took longer than {threshold} seconds.')
			return result
		return wrapper
	return decorator

# 使用装饰器timer
# 方式一
@timer(0.2)
def sleeped():
	time.sleep(0.4)
sleeped()

# 方式二
sleep = timer(0.2)(sleeped)


# 注意点：当我们打印他的函数名，返回的是wrapper，因为装饰器里定义的就是这个函数wrapper
sleeped.__name__  # 返回的是wrapper
# 但是我们是想要wrapper代替func这个函数,因此我们是希望wrapper可以继承func的属性，像是他的name或者其他的元属性，为此我们可以在wrapper上面加一行`@functools.wraps(func)`,这也是一个装饰器，是python中已经定义好的，所以我们需要导入`functools`这个模块，再打印sleeped.__name__的结果就是我们定义的函数原来的名字
```python
import functools
import time

#带参数的装饰器
def transfer_parameters(param):
	def decorator(func):
		def wrapper(*args, **kwargs)
			result = func(*args, **kwargs)
			return result
		return wrapper
	return decorator

@timer(0.2)
def sleeped():
	time.sleep(0.4)
sleeped()

sleeped.__name__  # 打印后得到的结果就是`sleeped`，不是wrapper了