您的位置:首页 > 编程语言 > Python开发

新版python logging 封装,支持同时向console,file,socket输出

2012-11-27 20:29 525 查看
将python的logging封装更新了一下,目前支持同时向console,file,socket输出,同时在config_logging或者config_logging_plus的时候先清除根logger的所有handler,避免在某些情况下的重复输出。
具体代码如下:

# -*- coding: utf-8 -*-
'''
Modified on 2012-11-27
@summary: clear old root logger handlers when reconfig logging
@author: JerryKwan

Created on 2012-06-14 19:50
@summary:  logging config
@author: JerryKwan
'''

import logging

import logging.handlers

import os
import sys

LEVELS = {'NOSET': logging.NOTSET,
'DEBUG': logging.DEBUG,
'INFO': logging.INFO,
'WARNING': logging.WARNING,
'ERROR': logging.ERROR,
'CRITICAL': logging.CRITICAL}

#set up logging to file

#logging.basicConfig(level = logging.NOTSET,
#                    format = "%(asctime)s %(name)-12s %(levelname)-8s %(message)s"
#                    )

##                    filename = "./log.txt",

##                    filemode = "w")

# create logs file folder
def config_logging(file_name = "log.txt", log_level = "NOSET"):
'''
@summary: config logging to write logs to local file
@param file_name: name of log file
@param log_level: log level
'''
logs_dir = os.path.join(os.path.dirname(__file__), "logs")
if os.path.exists(logs_dir) and os.path.isdir(logs_dir):
pass
else:
os.makedirs(logs_dir)

# clear old root logger handlers
logging.getLogger("").handlers = []

file_name = os.path.join(logs_dir, file_name)
# define a rotating file handler
rotatingFileHandler = logging.handlers.RotatingFileHandler(filename =file_name,
maxBytes = 1024 * 1024 * 50,
backupCount = 5)
formatter = logging.Formatter("%(asctime)s %(name)-12s %(levelname)-8s %(message)s")
rotatingFileHandler.setFormatter(formatter)
logging.getLogger("").addHandler(rotatingFileHandler)

# define a handler whitch writes messages to sys
console = logging.StreamHandler()
# set a format which is simple for console use
formatter = logging.Formatter("%(name)-12s: %(levelname)-8s %(message)s")
# tell the handler to use this format
console.setFormatter(formatter)
# add the handler to the root logger
logging.getLogger("").addHandler(console)
# set initial log level
logger = logging.getLogger("")
level = LEVELS[log_level.upper()]
logger.setLevel(level)

def config_logging_plus(file_name = "log.txt", log_level = "NOSET", remote_address = ("127.0.0.1", 8888), write_console = False):
'''
@summary: config logging to write logs to remote service
@param host: hostname or ip address of the log server
@param port: port to be used for log server
@log_level: log level
'''
logs_dir = os.path.join(os.path.dirname(__file__), "logs")
if os.path.exists(logs_dir) and os.path.isdir(logs_dir):
pass
else:
os.makedirs(logs_dir)
# format file name
if file_name is None:
file_name = os.path.splitext(sys.argv[0])[0]
file_name = os.path.join(logs_dir, "%s_%s.log"%(file_name, os.getpid()))
else:
file_name = os.path.join(logs_dir, file_name)

# clear old root logger handlers
logging.getLogger("").handlers = []

# define a rotating file handler
rotatingFileHandler = logging.handlers.RotatingFileHandler(filename =file_name,
maxBytes = 1024 * 1024 * 50,
backupCount = 5)
formatter = logging.Formatter("%(asctime)s %(name)-12s %(levelname)-8s %(message)s")
rotatingFileHandler.setFormatter(formatter)
# add log handler
logging.getLogger("").addHandler(rotatingFileHandler)

if write_console is not None and  write_console == True:
# define a handler whitch writes messages to sys
console = logging.StreamHandler()
console.setLevel(logging.NOTSET)
# set a format which is simple for console use
formatter = logging.Formatter("%(name)-12s: %(levelname)-8s %(message)s")
# tell the handler to use this format
console.setFormatter(formatter)
# add log handler
logging.getLogger("").addHandler(console)

if remote_address is not None and hasattr(remote_address, "__iter__") and len(remote_address) > 1:
# define a socket handler
socketHandler = logging.handlers.SocketHandler(remote_address[0], remote_address[1])
formatter = logging.Formatter("%(asctime)s %(processName)s %(process)s %(name)-12s %(levelname)-8s %(message)s")
socketHandler.setFormatter(formatter)
# add log handler
logging.getLogger("").addHandler(socketHandler)

# set initial log level
logger = logging.getLogger("")
level = LEVELS[log_level.upper()]
logger.setLevel(level)


使用方法很简单,只需要需要config logging的时候config一下即可,示例代码如下:

import loggingconfig
loggingconfig.config_logging(file_name = "log.log", log_level = "NOSET")
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐