python 装饰器

一、需求

foo():
    print("执行方法")

遵循开闭原则，即关闭修改，开放扩展。显示函数执行时间。

import time


def show_time(func):
    def wrapper():
        start = time.time()
        func()
        print("excute cost time %s" % (time.time() - start))
        return

    return wrapper  # 形成闭包函数 wrapper能够调用func


@show_time  # foo = show_time(foo)
def foo():
    time.sleep(1)
    print("执行方法")


foo()

二、被装饰函数传参

def detect(func):
    def wrapper(x, y):  # 为wrapper传参即为func传参
        if type(x) == int and type(y) == int:
            return func(x, y)  # func也需接收参数
        else:
            return print("请使用数字类型数据")

    return wrapper


@detect  # add = detect(add)  add即指向wrapper的内存地址
def add(x, y):
    result = x + y
    return result


add("s", 8)

三、装饰器传参

def show(flag):
    if not flag == "True":
        return print("该方法不可用")

    def detect(func):
        def wrapper(x, y, *args, **kwargs):
            if type(x) == int and type(y) == int:
                return func(x, y)
            else:
                return print("请使用数字类型数据")
        return wrapper
    return detect


@show("True")  # @show("True")=@detect
def add(x, y):
    result = x + y
    return result


add("s", 8)