python 设计模式（十）装饰者模式（Decorator pattern）

最新推荐文章于 2024-07-31 08:40:13 发布

ruguowoshiyu

最新推荐文章于 2024-07-31 08:40:13 发布

阅读量1.2k

点赞数

分类专栏： python python 设计模式文章标签：装饰者模式 python 设计模式

本文链接：https://blog.csdn.net/ruguowoshiyu/article/details/80921165

版权

本文介绍了Python中的装饰者模式，重点展示了装饰器在权限验证和添加日志方面的应用。通过示例代码，阐述了login_check装饰器如何实现登录验证以及with_log装饰器如何记录函数调用的日志。

摘要由CSDN通过智能技术生成

大家知道，装饰器可以给对象动态地添加行为。装饰器的两大应用场景是：

1 权限验证

2 添加日志

当然装饰器能应用的场景要远多于2种，下面分别用代码实现权限验证和日志：

1 权限验证

from functools import wraps


def login_check(func):
    '''
    Decorator 
    check whether user is logged in 
    :param func: 
    :return: 
    '''
    @wraps(func)
    def wrapper(*args, **kwargs):
        if args[0].status != 'login':
            # raise Exception('please log in first')
            print('please log in first')
        else:
            return func(*args, **kwargs)

    return wrapper


class Visitor(object):

    def __init__(self, status):
        self.status = status

    # decorate this method
    @login_check
    def access_2_resources(self):
        print('start downloading')

if __name__ == '__main__':
    visitor = Visitor('login')
    visitor.access_2_resources()