"""___author___ =JiangYiwei"""


# 1.為函數(shù)寫一個(gè)裝飾器阅嘶，在函數(shù)執(zhí)行之后輸出 after
def add_after(fn):
    def test(*args, **kwargs):
        fn(*args, **kwargs)
        print('after')

    return test


@add_after
def add_1(x):
    print(x * 2)


add_1(2)


# 2.為函數(shù)寫一個(gè)裝飾器箩做，把函數(shù)的返回值 +100 然后再返回。
def add_100(fn):
    def test(*args, **kwargs):
        re = fn(*args, **kwargs) + 100
        return re

    return test


@add_100
def add_2(x):
    return (x * 2)


print(add_2(2))  # 104


# 3.寫一個(gè)裝飾器@tag要求滿足如下功能:
def tag(fn):
    def test(*args, **kwargs):
        re = fn(*args, **kwargs)
        return "<p>" + re + "</p>"

    return test


@tag
def render(text):
    # 執(zhí)行其他操作
    return text


@tag
def render2():
    return 'abc'


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



# 4. 寫一個(gè)裝飾器@tag要求滿足如下功能(需要使用帶參的裝飾器，自己先自學(xué)正在一下):
def tag(name):
    def out_(fn):
        def test(*args, **kwargs):

            if name == "p":
                re = fn(*args, **kwargs)
                return '<p>' + re + '</p>'
            elif name == 'div':
                re = fn(*args, **kwargs)
                return '<div>' + re + '</div>'
            else:
                return

        return test

    return out_


# def say_hello(country):
#     def wrapper(func):
#         def deco(*args, **kwargs):
#             if country == 'china':
#                 print('你好熊昌！')
#             elif country == 'america':
#                 print('hello')
#             else:
#                 return
#             func(*args, **kwargs)
#         return deco
#     return wrapper
#
#
# @say_hello('china')
# def chinese():
#     print('我來(lái)自中國(guó)。')


print("====================4=====================")


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


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


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


# 為函數(shù)寫一個(gè)裝飾器北滥，根據(jù)參數(shù)不同做不同操作昂勒。
#
# flag為True，則 讓原函數(shù)執(zhí)行后返回值加100谤逼，并返回贵扰。
#
# flag為False，則 讓原函數(shù)執(zhí)行后返回值減100流部，并返回戚绕。

def say_hello(flag):
    def wrapper(func):
        def text(*args, **kwargs):
            if flag:
                re = func(*args, **kwargs) + 100
                return re
            else:
                re = func(*args, **kwargs) - 100
                return re

        return text

    return wrapper


@say_hello(flag=True)
def add_2(x):
    return (x * 2)


print(add_2(2))