python装饰器

装饰器本质上也是一个函数,装饰器函数用来修饰其他函数,在不改变原函数代码的情况下为函数增加功能而无需修改调用方式

不带参函数的装饰

import time
def show_time(func):# 装饰器函数调用原功能函数并返回一个新的函数对象
    def inner():
        start = time.time()
        func()
        end = time.time()
        print('函数执行时间为;%s' % (end - start))
    return inner

@show_time# 这一步相当于func1 = show_time(func1)
def func1():
    print('func1')
    time.sleep(1)
    print('func1 again')
func1()# 在这里调用func1实际上是执行了show_time函数里的inner函数

带参函数的装饰

import time
def show_time2(func):# 带参数函数的装饰,修改inner的参数就好了
    def inner(values):
        start = time.time()
        func(values)
        end = time.time()
        print('函数执行时间为;%s' % (end - start))
    return inner
@show_time2
def func2(values):
    print(values)
    time.sleep(1)
    print('%s again' % values)
    print(type(values))
func2(123)


# *************************假装是分割线*************************
def add_show_time(func):# 带参数函数的装饰,修改inner的参数就好了
    def inner(*args):
        start = time.time()
        func(*args)
        end = time.time()
        print('函数执行时间为;%s' % (end - start))
    return inner
@add_show_time
def add(*args):
    amount = 0
    for i in args:
        amount += i
    print(amount)
    time.sleep(1)
    return amount

add(1,2,3,4)

小结

得益于python强大的功能,python的函数可以像普通对象一样存在,可以作为参数传递给另一个函数也可以被重新赋值like

def func3():
    print('1234')
func3()
print(type(func3))

func3 = '1234'
print(func3)
print(type(func3))

装饰器其实也是一个函数,如果难以理解,可以将其记为一种特定的格式like
装饰器将需要修饰的功能函数作为参数传入==>意味着装饰器需要接受一个参数,我们将需要修饰的函数作为参数传入
装饰器内置一个函数并返回这个内置函数
在内置函数中调用需要修饰的功能函数(这里用到了参数),并增加新功能
@show_time相当于func1 = show_time(func1)

扩展-->带参数的装饰器

def add_show_time(flag):
    def show_time(func):
        def inner(*args):
            start = time.time()
            func(*args)
            end = time.time()
            print('函数执行时间为;%s' % (end - start))
            print('这是装饰器传入的参数:%s' % flag)
        return inner
    return show_time
@add_show_time('Nathaniel')# 实际上就是将装饰器进行一次封装,再返回装饰器对象
# add = add_show_time('Nathaniel')
def add(*args):
    amount = 0
    for i in args:
        amount += i
    print(amount)
    time.sleep(1)
    return amount

add(1,2,3,4)

 多次装饰

import time

# 编写装饰器,为函数加上统计时间的功能
def show_time(func):
    def inner(*args):
        start_time = time.time()
        func(*args)
        end_time = time.time()
        number = '1998'
        print('add函数执行了%s' %str(end_time - start_time))
    return inner

# 编写装饰器,为函数加上登录功能
def login(func):
    def inner(*args):
        name = input('请输入用户名:')
        password = input('请输入密码:')
        if name == '111' and password == '111':
            func(*args)
        else:print('用户名或密码错误')
    return inner

@login
@show_time
# 随便写个函数
def add(a,b):
    time.sleep(1)
    values = a + b
    print(values)
    return values

add(1,2)

装饰器之登陆

需求:三个页面,进入页面之前需要登录,每个页面登录用的账号都不一样,如果输错账号密码相当于不登录,成功登录一次后再进入其他页面不需要重复登录

userID = {
    '微信账户':('111','111'),
    'QQ账户':('222','222'),
    '支付宝账户':('333','333'),
}

token = False


def login(login_account_type):
    def account_type(func):
        def inner():
            global token
            if not token:
                name = input('请输入你的%s用户名>>: ' % login_account_type).strip()
                password = input('请输入你的%s密码>>: ' % login_account_type).strip()
                if name == userID[login_account_type][0] and password == userID[login_account_type][1]:
                    token = True
                    func()
                else:
                    print('账号或密码错误,默认等于不登录')
                    func()
            else:
                func()
        return inner
    return account_type

@login('微信账户')
def page1(): # 使用微信登录
    print('welcome to the page1')
    while True :# 输入校验,通过break
        choose = input('输入2进入page2,输入3进入page3,输入q退出')
        if choose =='q' or choose == 'Q':
            break
        if choose == '2':
            page2()
            break
        elif choose == '3':
            page3()
            break
        else:
            print('输入错误')
@login('QQ账户')
def page2(): # QQ登录
    print('welcome to the page2')
    while True :# 输入校验,通过break
        choose = input('输入1进入page1,输入3进入page3,输入q退出')
        if choose =='q' or choose == 'Q':
            break
        if choose == '1':
            page1()
            break
        elif choose == '3':
            page3()
            break
        else:
            print('Invalid enter')
@login('支付宝账户')
def page3():
    print('welcome to the page3')
    while True:  # 输入校验,通过break
        choose = input('输入1进入page1,输入2进入page2,输入q退出')
        if choose =='q' or choose == 'Q':
            break
        if choose == '1':
            page1()
            break
        elif choose == '2':
            page2()
            break
        else:
            print('Invalid enter')
page1()