装饰器

 1 import time
 2 def bar():#bar是这个函数的门牌号
 3     time.sleep(2)
 4     print(‘in the bar‘)
 5 def test1(func):#不是装饰器，是因为函数的调用方式test1(bar)被更改了
 6     start_time=time.time()
 7     func()
 8     stop_time=time.time()
 9     print(‘the func fun time is %s‘ % (stop_time-start_time))
10 
11 test1(bar) #func=bar

 1 import time
 2 def bar():#bar是这个函数的门牌号
 3     time.sleep(2)
 4     print(‘in the bar‘)
 5 
 6 def test2(func):
 7     print(func)
 8     return func
 9 
10 print(test2(bar))
11 print(‘--------------------‘)
12 bar=test2(bar)
13 print(‘--------------------‘)
14 bar()

1 def foo():
2     print(‘in the foo‘)
3     def bar():#bar具有局部变量的特性
4         print(‘in the bar‘)

 1 import time
 2 
 3 def timer(func):#在被装饰函数之外嵌套装饰器函数，添加装饰功能
 4     def deco(*args,**kwargs):#不固定参数参数
 5         start_time=time.time()
 6         result=func(*args,**kwargs)#被装饰的函数在这运行
 7         stop_time=time.time()
 8         print(‘the func run time is %s‘ % (stop_time-start_time))
 9 
10         return result#加入了返回值
11     return deco#timer这个函数返回了deco函数的地址
12 
13 @timer#@timer等于test1=timer(test1)：返回了deco函数的内存地址赋给了test1，此时test1是新的test1，这时的test1已经有了装饰器的功能
14 def test1():
15     time.sleep(3)
16     print(‘in the test1‘)
17 
18 @timer#test2=timer(test2),deco=test2
19 def test2(name):
20     time.sleep(3)
21     print(‘test2:‘,name)
22     print(‘in the test2‘)
23     return ‘from test2‘#加入了返回，装饰器内的函数也要加返回
24 
25 test1()
26 print(‘-----------------‘)
27 print(test2(‘tao‘))#print打印函数的返回值

 1 user,passwd=‘Tao‘,‘abc‘
 2 def auth(auth_type):#与未加参数的装饰器区别：装饰器多加了一个参数，要多嵌套一层函数；auth返回的仍是wrapper这个函数
 3     def outer_wrapper(func):#注意加入的参数和func的位置
 4         def wrapper(*args, **kwargs):#auth函数返回out_wrapper,out_wrapper函数返回wrapper，所以auth函数返回的仍是wrapper这个函数
 5 
 6             #判断参数
 7             if auth_type==‘local‘:
 8                 username = input(‘username:‘).strip()
 9                 password = input(‘passwd:‘).strip()
10 
11                 if user == username and passwd == password:
12                     print(‘欢迎‘)
13                     res = func(*args, **kwargs)  #当被装饰函数有返回值时
14                     return res
15                 else:
16                     exit(‘错误‘)
17 
18             #判断参数
19             elif auth_type==‘ldap‘:
20                 print(‘ldap无法实现‘)
21 
22         return wrapper
23 
24     return outer_wrapper
25 
26 
27 
28 def index():
29     print(‘welcome to index page‘)
30 
31 @auth(auth_type=‘local‘)#auth返回了wrapper这个函数赋给了home
32 def home():
33     print(‘welcome to home page‘)
34     return "from home" #加入了返回，装饰器内的函数也要加返回
35 
36 @auth(auth_type=‘ldap‘)#auth返回了wrapper这个函数赋给了bbs
37 def bbs():
38     print(‘welcome to bbs page‘)
39 
40 index()
41 print(‘--------------------‘)
42 print(home())#打印函数的返回值即from home
43 print(‘--------------------‘)
44 bbs()