装饰器定义

函数定义装饰器

def p_decorator(func):
    def wrapper(*args, **kwargs):
        return '<p>' + func(*args, **kwargs) + '<p>'

    return wrapper


@p_decorator  # 函数定义装饰器
def get_text():
    return '优品课堂'


@p_decorator
def get_upper_text(text):
    return text.upper()


if __name__ == '__main__':
    print(get_text())
    # <p>优品课堂<p>
    # html = p_decorator(get_text)  # html=wrapper
    # print(html()) # <p>优品课堂<p>

    print(get_upper_text('www.abc.com'))
    # <p>WWW.ABC.COM<p>

类定义装饰器

class P:
    def __init__(self, func):
        self.func = func

    def __call__(self, *args, **kwargs):
        return '<p>' + self.func(*args, **kwargs) + '<p>'


@P  # 类定义装饰器
def get_text():
    return '优品课堂'


@P
def get_upper_text(text):
    return text.upper()


if __name__ == '__main__':
    print(get_text())
    # <p>优品课堂<p>

    print(get_upper_text('www.abc.com'))
    # <p>WWW.ABC.COM<p>

参数化装饰器

# 参数化装饰器
def tags(tag):  # 再添加一参数函数
    def tag_decorator(func):
        def wrapper(*args, **kwargs):
            return f'<{tag}>{func(*args, **kwargs)}</{tag}>'

        return wrapper

    return tag_decorator


# def p_decorator(func):
#     def wrapper(*args, **kwargs):
#         return '<p>' + func(*args, **kwargs) + '<p>'
#
#     return wrapper

@tags('div')
@tags('p')
def get_text():
    return '优品课堂'


@tags('div')
@tags('p')
def get_upper_text(text):
    return text.upper()


if __name__ == '__main__':
    print(get_text())
    # <div><p>优品课堂</p></div>

    print(get_upper_text('www.abc.com'))
    # <div><p>WWW.ABC.COM</p></div>