作业009：装饰器

1. 为函数写一个装饰器，在函数执行之后输出 after

def print_after(fn):
    def test(*args, **kwargs):
        re = fn(*args, **kwargs)
        print('after')
        return re

    return test


@print_after
def yyd_sum(x, y):
    return x + y


print(yyd_sum(10, 20))

2. 为函数写一个装饰器，把函数的返回值 +100 然后再返回。

def add_num(fn):
    def test(*args, **kwargs):
        num = fn(*args, **kwargs)   # 运行yyd_sum()
        if isinstance(num, int) or isinstance(num, float):
            return num + 100
        return num

    return test


@add_num    # yyd_num = add_num(yydsum)
def yyd_sum(x, y):
    return x + y


print(yyd_sum(10, 20))
# res = add_num(yyd_sum)
# print(res(10, 20))    # test()

3. 写一个装饰器@tag要求满足如下功能:

def tag(fn):
    def test(*args, **kwargs):
        char = fn(*args, **kwargs)
        return '<p>' + str(char) + '</p>'

    return test


@tag
def render(text):
    # 执行其他操作
    return text


@tag
def render2():
    return 'abc'


print(render('Hello'))  # 打印出: <p>Hello</p>
print(render2())  # 打印出: <p>abc</p>

4. 写一个装饰器@tag要求满足如下功能

def tag(name):
    def decorator(fn):
        def wrapper(*args, **kwargs):
            value = fn(*args, **kwargs)
            return "<{name}>{value}</{name}>".format(name=name, value=value)

        return wrapper

    return decorator


@tag(name='p')
def render(text):
    # 执行其他操作
    return text


@tag(name='div')
def render2():
    return 'abc'


print(render('Hello'))  # 打印出: <p>Hello</p>
print(render2())  # 打印出: <div>abc</div>

5. 为函数写一个装饰器，根据参数不同做不同操作。

• flag为True，则 让原函数执行后返回值加100，并返回。
• flag为False，则 让原函数执行后返回值减100，并返回。

def yyd_change(fn):
    def test(flag, *args, **kwargs):
        value = fn(*args, **kwargs)
        if flag:
            value += 100
        else:
            value -= 100
        return value

    return test


@yyd_change
def yyd_sum(x, y):
    return x + y


print(yyd_sum(True, 12, 23))