装饰器

装饰器功能：

1. 引入日志
2. 函数执行时间统计
3. 执行函数前预备处理
4. 执行函数后清理功能
5. 权限校验等场景
6. 缓存

装饰器：

# 不带参数多装饰器
# 定义函数：完成包裹数据
def makeBold(fn):
    def wrapped():
        return "<b>" + fn() + "</b>"
    return wrapped

# 定义函数：完成包裹数据
def makeItalic(fn):
    def wrapped():
        return "<i>" + fn() + "</i>"
    return wrapped

# 先装饰离函数最近的，执行按照从上到下
@makeBold   # 后被装饰 先执行
@makeItalic # 先被装饰 后执行
def test1():
    return "hello world-1"

print(test1())
print("*" * 50)

# 带参数装饰器
from time import ctime, sleep
def timefun(func):
    def wrapped_func(a, b):
        print("%s called at %s" % (func.__name__, ctime()))
        print(a, b)
        func(a, b)
    return wrapped_func

@timefun
def foo(a, b):
    print(a+b)

foo(3,5)
sleep(2)
foo(1,4)

print("*" * 50)

# 不定长参数
def sum_num(func):
    def sums(*args, **kwargs):
        return func(*args, **kwargs)
    return sums

@sum_num
def test2(a, b, c):
    return a + b + c

print(test2(1, 2, 3))
print("*" * 50)

# 装饰器带参数
URL_FUNC_DICT = dict()

def route(url):
    def set_func(func):
        URL_FUNC_DICT[url] = func
        def call_func():
            route = "route:" + url + "----" + func()
            return route
        return call_func
    return set_func

@route("/index.py")  # "/index.py"传给了参数url
def index():
    return "index.html"

print("------路由------")
print(index())
print("*" * 50)

from time import ctime, sleep

def timefun_arg(pre="hello"):
    def timefun(func):
        def wrapped_func():
            print("%s called at %s %s" % (func.__name__, ctime(), pre))
            return func()
        return wrapped_func
    return timefun

# 下面的装饰过程
# 1. 调用timefun_arg("itcast")
# 2. 将步骤1得到的返回值，即time_fun返回， 然后time_fun(foo)
# 3. 将time_fun(foo)的结果返回，即wrapped_func
# 4. 让foo = wrapped_fun，即foo现在指向wrapped_func
@timefun_arg("itcast")  # 相当于 foo()==timefun_arg("itcast")(foo)()
def foo():
    print("I am foo")

@timefun_arg("python")
def too():
    print("I am too")

foo()
print("*" * 50)

# 类装饰器
class Test(object):
    def __init__(self, func):
        self.func = func

    def __call__(self):
        print("这里是装饰器添加的功能")
        return self.func()
@Test  # 相当于get_str = Test(get-str)
def get_str():
    return "haha"

print(get_str())

结果：

<b><i>hello world-1</i></b>
**************************************************
foo called at Fri Sep  6 14:16:04 2019
3 5
8
foo called at Fri Sep  6 14:16:06 2019
1 4
5
**************************************************
6
**************************************************
------路由------
route:/index.py----index.html
**************************************************
foo called at Fri Sep  6 14:16:06 2019 itcast
I am foo
**************************************************
这里是装饰器添加的功能
haha

进程已结束,退出代码0