77 lines
2.9 KiB
Python
77 lines
2.9 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
@Time : 2019/4/15
|
||
@Author : LXW
|
||
@Site :
|
||
@File : LoggerUtil.py
|
||
@Software: PyCharm
|
||
@Description: 日志工具，默认保存文件在当前文件夹下
|
||
"""
|
||
import logging
|
||
from logging import handlers
|
||
from functools import wraps
|
||
import traceback
|
||
import os
|
||
|
||
|
||
class Logger(object):
|
||
"""
|
||
日志封装类：
|
||
1. 作用是同时打印和写日志文件。
|
||
2. 业务日志如果是需要告警或者邮件，则走warning。
|
||
3. 运行出错则走error
|
||
4. record用来切片函数，记录函数开始和结束。
|
||
"""
|
||
level_relations = {
|
||
'debug': logging.DEBUG,
|
||
'info': logging.INFO,
|
||
'warning': logging.WARNING,
|
||
'error': logging.ERROR,
|
||
'crit': logging.CRITICAL
|
||
} # 日志级别关系映射
|
||
|
||
def __init__(self, filename, level='info', when='D', backCount=3,
|
||
fmt='%(asctime)s - %(pathname)s[line:%(lineno)d] - %(levelname)s: %(message)s'):
|
||
if not os.path.exists(os.path.dirname(filename)):
|
||
os.makedirs(os.path.dirname(filename))
|
||
self.logger = logging.getLogger(filename)
|
||
format_str = logging.Formatter(fmt) # 设置日志格式
|
||
self.logger.setLevel(self.level_relations.get(level)) # 设置日志级别
|
||
sh = logging.StreamHandler() # 往屏幕上输出
|
||
sh.setFormatter(format_str) # 设置屏幕上显示的格式
|
||
th = handlers.TimedRotatingFileHandler(filename=filename, when=when, backupCount=backCount,
|
||
encoding='utf-8') # 往文件里写入#指定间隔时间自动生成文件的处理器
|
||
# 实例化TimedRotatingFileHandler
|
||
# interval是时间间隔，backupCount是备份文件的个数，如果超过这个个数，就会自动删除，when是间隔的时间单位，单位有以下几种：
|
||
# S 秒
|
||
# M 分
|
||
# H 小时、
|
||
# D 天、
|
||
# W 每星期（interval==0时代表星期一）
|
||
# midnight 每天凌晨
|
||
th.setFormatter(format_str) # 设置文件里写入的格式
|
||
self.logger.addHandler(sh) # 把对象加到logger里
|
||
self.logger.addHandler(th)
|
||
|
||
def record(self, arg):
|
||
def _log(func):
|
||
@wraps(func)
|
||
def wrapper(*args, **kwargs):
|
||
self.logger.info(arg + '_开始----' + func.__name__)
|
||
try:
|
||
ret = func(*args, **kwargs)
|
||
except Exception as e:
|
||
desc = "%s_traceback.format_exc(): %s" % (e, traceback.format_exc())
|
||
self.logger.error(desc)
|
||
self.logger.info(arg + '_中断----' + func.__name__)
|
||
# exit(1)
|
||
return False
|
||
else:
|
||
self.logger.info(arg + '_结束----' + func.__name__)
|
||
return ret
|
||
|
||
return wrapper
|
||
|
||
return _log
|