python 基础——常用日志装饰器

from functools import wraps

class logit():
    def __init__(self, logfile='out.log'):
        self.log = logfile
        
    def __call__(self, func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            # args 为元组;kwargs 为字典
            info = func.__name__ + "was called"
            with open(self.log, 'rw') as f:
                f.write(info)
            self.notify()
            # 把传递进来的参数元组 args 和 参数字典 kwargs 展开,并调用原始函数
            return func(*args, **kwargs)
        return wrapper
    
    def notify(self):
        # TODO send email
        pass
    
def func1(arg1, arg2):
    print arg1, arg2
    
# 不使用装饰器
func = logit('test.log')(func1)
func(1, 2)

#使用装饰器
@logit('test.log')
def func2(arg1, arg2):
    print arg1, arg2
func2(3, 4) 

 

posted @ 2016-08-26 16:21  弋痕夕的残影  阅读(989)  评论(0编辑  收藏  举报