Python3装饰器的使用

装饰器

简易装饰器模板

def wrapper(func):

    def inner(*args,**kwargs):

        print('主代码前添加的功能')

        ret=func(*args,**kwargs)

        print('主代码后添加的功能')

        return ret

    return inner

@wrapper

def func():

    print('主代码')

利用装饰器完成的登录认证

def wrapper(func):

    def inner(*args, **kwargs):

        username = input('请输入用户名：')

        password = input('请输入密码：')

        ret = func(username, password)

        if ret == '':

            print('登录成功')

        else:

            print('登录失败')

        return ret

    return inner

@wrapper

def func(username, password):

    if username == 'zhaoruofei' and password == '':

        return ''

    else:

        return ''

func()

装饰器的执行流程

wrapper(func)