10、python-装饰器

简介

作用：用一个新函数封装旧函数（是旧函数代码不变的情况下增加功能）然后会返回一个新函数，新函数就叫做装饰器

一.装饰器前生（闭包）

1. 先执行w1()函数，返回inner函数
2. 再执行inner()函数，inner()函数执行中会执行func()函数
3. 执行func()函数，是由w1函数传进来的函数决定，这里执行func1()

def w1(func):
    def inner():
        print('正在验证权限...')
        func()

    return inner

def func1():
    print('------func1------')

w1(func1)()

二.装饰器使用

def w1(func):
    def inner():
        print('正在验证权限...')
        func()

    return inner

@w1
def func1():
    print('------func1------')

func1()

三.多个装饰器的使用

def make_bold1(func):
    def wrapped():
        print('---1---')
        return "<b>" + func() + "</b>"

    return wrapped

def make_bold2(func):
    def wrapped():
        print('---2---')
        return "<i>" + func() + "</i>"

    return wrapped

@make_bold1
@make_bold2
def test():
    print('---3---')
    return 'hello world'

print(test())

四.装饰器固定参数传参

def w1(func):
    def inner(aa, bb):
        print('正在验证权限...')
        func(aa, bb)

    return inner

@w1
def func1(a, b):
    print('------func1------{},{}'.format(a, b))

func1(1, 2)

五.装饰器不定参数传参

def w1(func):
    def inner(*args, **kwargs):
        print('正在验证权限...')
        func(*args, **kwargs)

    return inner

@w1
def func1(a, b):
    print('------func1------{},{}'.format(a, b))

@w1
def func2(a, b, c):
    print('------func1------{},{},{}'.format(a, b, c))

func1(1, 2)
func2(1, 2, 3)

六.装饰器带有返回值

def w1(func):
    def inner():
        print('正在验证权限...')
        r = func()
        return r + "456"

    return inner

@w1
def func1():
    print('------func1------')
    return '123'

print(func1())

七.带有参数的装饰器

def w_arg(a):
    def w1(func):
        def inner():
            print('正在验证权限...{}'.format(a))
            if a == '通过':
                print('验证通过')
            else:
                print('验证失败')
            func()

        return inner

    return w1

@w_arg('通过')
def func1():
    print('------func1------')

@w_arg('123')
def func2():
    print('------func2------')

func1()
func2()