python3 装饰器_Python3中装饰器的使用

较为复杂的装饰器使用：

user,passwd = 'hjc',111111

def auth(type):

print('auth type:',type)

def outwrapper(func):

def wrapper(*args,**kwargs):

if type == 'lll':

username = input('username:').strip()

password = int(input('password:').strip())

if user == username and passwd == password:

print("\033[32;1muser pass ~~~~~~\033[0m")

func(*args,**kwargs)

else:

exit("\033[31;1minvalid user ~~~~~~\033[0m")

elif type == 'kkk':

print('fuck off')

return wrapper

return outwrapper

def index():

print("welcome to index page")

@auth('lll')

def home():

print("welcome to home page")

@auth('kkk')

def bbs():

print("welcome to bbs page")

index()

home()

bbs()

输出结果如下：

auth type: lll

auth type: kkk

welcome to index page

username:hjc

password:111111

user pass ~~~~~~

welcome to home page

fuck off

2.较为简单的装饰器：

import time

def timer(func):

def inner(*args,**kwargs):

start_time = time.time()

func(*args,**kwargs)

stop_time = time.time()

print(stop_time-start_time)

return inner

@timer

def wt():

print('hello wt!!!')

wt()