python整理十三——迷你日志

写个小点的程序，写日志太麻烦了，大点的程序，不写日志是万万不行的，鉴于日志频繁的使用，对python的logging稍微封装一下，适合自己的需求：

 

 

1. #coding=utf-8
3. import os, logging, sys
4. from logging.handlers import RotatingFileHandler as RFHandler
6. class CommLogger:
7.     
8.     LOG_FILE_SIZE_MAX    = 1024  # byte
9.     LOG_FILE_BACKUP_MAX  = 5 # 
11.     def __init__(self):
12.         self.__log_argv = {'dirname'      : os.path.join(os.path.dirname(__file__), 'log' ),
13.                            'name_default' : 'log_default',
14.                            'extend'       : '.log',
15.                            }
16.         self.__log_fmt_t = '%(f_info)s%(ctrl)s%(ctime)s%(ctrl)s%(msg)s%(ctrl)s%(tail)s%(ctrl)s' % /
17.             {'f_info' : 'File:%(pathname)s,line %(lineno)d,function: %(funcName)s',    
18.              'ctime'  : '%(asctime)s [%(levelname)s]',     
19.              'msg'    : '%(message)s', 
20.              'tail'   : '-' * 80 ,
21.              'ctrl'   : os.linesep,}
22.         self.__log_fmt = logging.Formatter(self.__log_fmt_t)
23.         self.__loggers = {}
25.     def __getLogPath(self, log_name):
26.         fpath = os.path.join(self.__log_argv['dirname'], log_name)
27.         if not os.path.exists(fpath):
28.             os.makedirs(fpath)
29.         return fpath
31.     def __initLog(self, log_name, level, bPrint):
32.         if log_name not in self.__loggers:
33.             logger = logging.getLogger(log_name)
34.             if len(logger.handlers) == 0:
35.                 if bPrint:
36.                     chdlr = logging.StreamHandler(sys.stdout)
37.                     chdlr.setFormatter(self.__log_fmt)
38.                     logger.addHandler(chdlr) 
39.                 else:
40.                     fname = os.path.join(self.__getLogPath(log_name), log_name + self.__log_argv['extend'])
41.                     fhdlr = RFHandler(fname, 'a', CommLogger.LOG_FILE_SIZE_MAX, CommLogger.LOG_FILE_BACKUP_MAX,)
42.                     fhdlr.setFormatter(self.__log_fmt)
43.                     logger.addHandler(fhdlr)
44.                 logger.setLevel(level)
45.                 self.__loggers[log_name] = logger  
46.         return self.__loggers[log_name]
47.     
48.     def getLogger(self, log_name = None, bPrint = False, level= logging.DEBUG, ):
49.         if log_name is None:
50.             log_name = self.__log_argv['name_default']
51.         if log_name not in self.__loggers:
52.             self.__initLog(log_name, level, bPrint)
53.         return self.__loggers[log_name]
55. def mini_log(fn):
56.     def moniter(*args, **argkv):
57.         return fn(*args, **argkv)
58.     logger = CommLogger().getLogger(fn.func_name)
59.     log_info = {'fn_name' : repr(fn.func_name), 'fn_code' : fn.func_code, 'ctrl' : os.linesep}
60.     logger.info('The function of %(fn_name)s has been running.%(ctrl)s%(fn_code)s' % log_info)
61.     return moniter    
63. @mini_log
64. def test():
65.     print 'test logging module.'
67. if __name__ == '__main__':
68.     cCls = CommLogger()
69.     log0 = cCls.getLogger()
70.     log1 = cCls.getLogger('log1', True)
71.     for i in range(3):
72.         log0.debug('debug')
73.         log0.info('info')
74.         log0.warning('warn')
75.         log0.error('error')
76.         log0.critical('critical')
77.         
78.         log1.debug('debug')
79.         log1.info('info')
80.         log1.warn('warn')
81.         log1.error('error')
82.         log1.critical('critical')        
83.     test()
84.     print 'end___'