# 1.为函数写一个装饰器，在函数执行之后输出 after
def add_after(fn):
    def text(*args, **kwargs):
        fn(*args, **kwargs)
        print('after')
    return text


@add_after
def yt_sum(x, y):
    print(x+y)
    return x + y


yt_sum(1, 2)


# 2.为函数写一个装饰器，把函数的返回值 +100 然后再返回。
def add100(fn):
    def text(*args, **kwargs):
        re = fn(*args, **kwargs) + 100
        return re
    return text


@add100
def yt_sum1(x, y):
    return x + y


print(yt_sum1(1, 2))


# 3.写一个装饰器@tag要求满足如下功能:
def tag(fn):
    def text(*args, **kwargs):
        re = '<p>'+fn(*args, **kwargs)+'</p>'
        return re
    return text


@tag
def render2():
    return 'abc'


@tag
def render(text):
    text1 = text
    return text1


print(render2())
print(render('Hello'))

print('====================================================')
# 4.写一个装饰器@tag要求满足如下功能(需要使用带参的装饰器，自己先自学正在一下):


def tag(name):
    def i(fn):
        def i2(*args, **kwargs):
            re = '<{}>{}<{}>'.format(name, fn(*args, **kwargs), name)
            return re
        return i2
    return i


@tag('a')
def render2():
    return 'abc'


@tag('b')
def render(text):
    text1 = text
    return text1


print(render2())
print(render('Hello'))


# 5.为函数写一个装饰器，根据参数不同做不同操作。
# flag为True，则 让原函数执行后返回值加100，并返回。
# flag为False，则 让原函数执行后返回值减100，并返回。


def add100(fn):
    def text(*args, **kwargs):
        re = fn(*args, **kwargs) + 100
        return re
    return text


def sub100(fn):
    def text(*args, **kwargs):
        re = fn(*args, **kwargs) - 100
        return re
    return text


def add_sub(flag, *args):
    if flag == True:
        return add100
    elif flag == False:
        return sub100


@add_sub(False)
def yt_sum(x, y):
    print(x + y)
    return x + y


print(yt_sum(1, 2))