# 为函数写一个装饰器，在函数执行之后输出 after
def af(fn):
    def test(*args,**kwargs):
        re=fn(*args,**kwargs)
        print('after')
        return re

    return test

@af
def ys_sum(x,y):
    return x+y


print(ys_sum(10,20))

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

@jia
def ys_2(x,y):
    return x+y

print(ys_2(15,25))
print('===============================================')
# 写一个装饰器@tag要求满足如下功能:
"""
@tag
def render(text):
    # 执行其他操作
    return text

@tag
def render2():
    return 'abc'

print(render('Hello'))   # 打印出: <p>Hello</p>
print(render2())     # 打印出: <p>abc</p>
"""
def tag1(fn):
    def test(*args,**kwargs):
        re=fn(*args,**kwargs)
        return '<p>'+re+'</p>'
    return test

@tag1
def render(text):
    return text

print(render('Hello'))

print('===============================================')
# 写一个装饰器@tag要求满足如下功能(需要使用带参的装饰器，自己先自学正在一下):
"""
@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>
"""
def tag2(name):
    def mid(fn):
        def test(*args,**kwargs):
            re=fn(*args,**kwargs)
            return '<'+name+'>'+re+'</'+name+'>'
        return test
    return mid

@tag2(name='p')
def render2():
    return 'abc'

print(render2())

print('===============================================')
"""
为函数写一个装饰器，根据参数不同做不同操作。
flag为True，则 让原函数执行后返回值加100，并返回。
flag为False，则 让原函数执行后返回值减100，并返回
"""
def ys5(flag:bool):
    def mid(fn):
        def test(*args,**kwargs):
            re=fn(*args,**kwargs)
            if flag:
                return re+100
            else:
                return re-100
        return test
    return mid


@ys5(flag=True)
def zj(x,y):
    return x + y


print(zj(2,1))

@ys5(flag=False)
def zj(x,y):
    return x + y


print(zj(2,1))