python基础知识点大全-【python基础学习】基础重点难点知识汇总-CSDN博客

python中decorator装饰器

语法示例：

@decorator

什么是装饰器：

问题：

定义一个函数后

在运行时动态增加功能

又不想改动函数本身的代码

示例：

# 希望对下列函数调用增加log功能，打印出函数调用：

def f1(x):return x*2def f2(x):return x*x

def f3(x):return x*x*x

方法1：直接修改原函数的定义

def f1(x):

print('call f1')return x*2def f2(x):

print('call f2')return x*x

def f3(x):

print('call f3')return x*x*x

有没有更简单的方法？

高阶函数1. 可以接受函数作为参数2. 可以返回函数

是否可以接受一个函数，对其进行包装，然后返回一个新函数？

方法2：通过高阶函数返回新函数（装饰器函数）：

def f1(x):return x*2def new_fn(f):

def fn(x):

print('call'+f.__name__+'()')returnf(x)return fn

调用1：

g1 =new_fn(f1)

print( g1(5) )

调用2：

f1 =new_fn(f1)

print( f1(5) )

装饰器：

python内置@语法就是为了简化装饰器调用

@new_fn

def f1(x):return x*2

等同于：

def f1(x):return x*2f1= new_fn(f1)

装饰器的作用：

可以极大简化代码，避免每个函数编写重复性的代码

打印日志：@log

检测性能：@performance

数据库事务：@transaction

URL路由：@post('/register')

装饰器示例：

1. 如果自定义函数存在两个形参，上述装饰器函数内部固定写了一个形参，会出现错误怎么办：

@new_fn

def add(x,y):return x+y

# 这里会出现错误，所以需要更改 new_fn 函数如下:

def new_fn(f):

def fn(*args, **kwargs):

print('call'+f.__name__+'()')return f(*args, **kwargs)return fn

2. 请编写一个@performance，它可以打印出函数调用的时间。计算函数调用的时间可以记录调用前后的当前时间戳，然后计算两个时间戳的差。

import time

def performance(f):

def fn(*args, **kw):

t1=time.time()

r= f(*args, **kw)

t2=time.time()

print'call %s() in %fs' % (f.__name__, (t2 -t1))returnrreturnfn

@performance

def factorial(n):return reduce(lambda x,y: x*y, range(1, n+1))

print factorial(10)

3.上述装饰器函数只能接受一个函数参数，如果装饰器函数需要传入额外的参数怎么办？

@log('DEBUG')

def my_func():

pass

把上面的定义翻译成高阶函数的调用，就是：

my_func = log('DEBUG')(my_func)

上面的语句看上去还是比较绕，再展开一下：

log_decorator = log('DEBUG')

my_func= log_decorator(my_func)

上面的语句又相当于：

log_decorator = log('DEBUG')

@log_decorator

def my_func():

pass

所以，带参数的log函数首先返回一个decorator函数，再让这个decorator函数接收my_func并返回新函数：

def log(prefix):

def log_decorator(f):

def wrapper(*args, **kw):

print'[%s] %s()...' %(prefix, f.__name__)return f(*args, **kw)returnwrapperreturnlog_decorator

@log('DEBUG')

def test():

pass

print test()

上述第二个例子：@performance只能打印秒，请给 @performace 增加一个参数，允许传入's'或'ms'：

import time

import functools

def performance(unit):

def perf_decorator(f):

@functools.wraps(f)

def wapper(*args, **kw):

t1=time.time()

r= f(*args, **kw)

t2=time.time()

t= (t2 - t1) * 1000 if unit=='ms' else (t2 -t1)

print'call %s() in %f %s' %(f.__name__, t,unit)returnrreturnwapperreturnperf_decorator