python之装饰器使用场景

1.用户登录认证

• 有三个界面需要登录认证通过后才能查看
• 一个登录函数作为装饰器
• 调用使用@函数名
• 调用中如果加参数，则装饰器需要再加一层

def login(func):
    def desc(*args,**kwargs):
        user='test'
        passwd ='123456'
        n=0
        if n<=3:
            for i in range(3):
                user_input=input('用户名')
                passwd_input=input('密码')
                if user_input==user and passwd_input==passwd:
                    func(*args,**kwargs)
                    break
                else:
                    print('输入错误，请重新输入')
                    n+=1
    return desc

@login
def home():
    print('welcome to home')
@login
def product():
    print('welcome to product')
@login
def solution():
    print('welcome to solution')

home()
product()
solution()

2.日志操作时间

• 在三个界面记录操作时间
• 使用logger日志

def logger(func):
    '''日志装饰器'''
    def desc(*args,**kwargs):
        start_time=time.time()
        func()
        end_time=time.time()
        print('操作了%s时间'%(end_time-start_time))
    return desc
@logger
def home(*args,**kwargs):
    time.sleep(2)
    print('welcome to home')
@logger
def product():
    time.sleep(2)
    print('welcome to product')
@logger
def solution():
    time.sleep(2)
    print('welcome to solution')

home()
product()
solution()