python装饰器

学习版本3.5.2

装饰器就是在不改变其它函数代码的情况下为它们加上新的功能。

1.装饰器的语法糖@

def deco(func):
    def _deco():
        print("log before func")
        func()
        print("log affter func")
    return _deco

def func1():
    print("hello world")

@deco
def func2():
    print("hello world")

func1 = deco(func1)

func1()
func2()

运行结果：

log before func
hello world
log affter func
log before func
hello world
log affter func

从运行结果中可以看出来，@符相当于做了func2 = deco(func2)这个操作。

2.类装饰器

当一个类实现了__call__函数的时候，该类的对象就会当作函数来使用。

class Deco(object):
    def __init__(self, func):
        self._func = func

    def __call__(self):
        print("log before func")
        self._func()
        print("log affter func")

def func1():
    print("hello world")

@Deco
def func2():
    print("hello world")

func1 = Deco(func1)

func1()
func2()

运行结果：

log before func
hello world
log affter func
log before func
hello world
log affter func

运行结果与1中一摸一样。

3.装饰带参数的函数

使用（*args, **kwargs）来适应变参和命名参数

def deco(func):
    def _deco(*args, **kwargs):
        print("log before func")
        func(*args, **kwargs)
        print("log affter func")
    return _deco

def func1(a,b):
    print("hello world",a,b)

@deco
def func2(a,b,c):
    print("hello world",a,b,c)

func1 = deco(func1)

func1(1,2)
func2(1,b=2,c=3)

运行结果：

log before func
hello world 1 2
log affter func
log before func
hello world 1 2 3
log affter func

4.装饰器带参数

def deco1(arg):
    def deco(func):
        def _deco(*args, **kwargs):
            print("log before func",arg)
            func(*args, **kwargs)
            print("log affter func",arg)
        return _deco
    return deco

def func1(a,b):
    print("hello world",a,b)

@deco1("args")
def func2(a,b,c):
    print("hello world",a,b,c)

func1 = deco1("args")(func1)

func1(1,2)
func2(1,b=2,c=3)

运行结果：

log before func args
hello world 1 2
log affter func args
log before func args
hello world 1 2 3
log affter func args