python装饰器
Posted
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了python装饰器相关的知识，希望对你有一定的参考价值。
一.基本装饰器
基本装饰器的作用:
在不改变原函数的基础上, 通过装饰器, 给原函数新增某些功能
实现方法:
在原函数上加
@装饰器名字
其中@叫做语法糖
定义装饰器
第一层函数传入参数(用于传入原函数)
第二层使用原函数的同时, 加入需要新增的功能
第一层函数要返回第二层函数名
整个函数形成闭包
import time
def runtime(func):
def wrapper():
start = time.time()
for i in range(100):
func()
end = time.time()
print("程序运行时间为 {} ".format((end - start)/1000.0))
return wrapper
@runtime
def hello():
print("hello world")
hello()
二.三层装饰器
现在需要在装饰器的基础上, 调用 @装饰器 的时候传入参数
就需要在原有的装饰器的基础上, 在外层写一个函数, 从而又形成闭包的结构
import time
def runtime(msg="默认值"):
def decorator(func):
def wrapper():
start = time.time()
for i in range(100):
func()
end = time.time()
print(msg)
print("程序运行时间为 {} ".format((end - start) / 1000.0))
return wrapper
return decorator
@runtime("hello()")
def hello():
print("hello world")
hello()
三.完善参数传递
在之前的装饰器中, 由于原函数可能存在不同种类的参数, 可能有各种各样的返回值, 所以要进行一下两点修改
1 将装饰器实际执行函数的参数设置为(*, **)的形式
2 改函数需要return 原函数
import time
def log(msg="默认值"):
def decorator(func):
def wrapper(*args, **kwargs):
print(func.__name__, msg)
return func(*args, **kwargs)
return wrapper
return decorator
@log("hello()")
def hello():
print("hello world")
hello()
以上是关于python装饰器的主要内容，如果未能解决你的问题，请参考以下文章