python 多进程按天打印日志

encoding: utf-8

import os
import time
import logging

“”"
重写FileHandler类，用于多线程中日志按时间间隔分割
“”"

class SafeFileHandler(logging.FileHandler):
def init(self, filename, mode, timemode=“day”):
“”"
Use the specified filename for streamed logging
:param filename:
:param mode:
:param timemode: 日志的时间分片方式(天/小时/分/秒)
“”"
encoding = None
logging.FileHandler.init(self, filename, mode=‘a’, encoding=None, delay=False)
self.mode = mode
self.encoding = encoding
if timemode == “day”:
self.suffix = “%Y-%m-%d”
elif timemode == “hour”:
self.suffix = “%Y-%m-%d_%H”
elif timemode == “min”:
self.suffix = “%Y-%m-%d_%H:%M”
elif timemode == “sec”:
self.suffix = “%Y-%m-%d_%H:%M:%S”

    self.suffix_time = ""

def emit(self, record):
    """
    Emit a record.

    Always check time
    """
    try:
        if self.check_base_filename(record):
            self.build_base_filename()
        logging.FileHandler.emit(self, record)
    except (KeyboardInterrupt, SystemExit):
        raise
    except:
        self.handleError(record)

def check_base_filename(self, record):
    """
    Determine if builder should occur.

    record is not used, as we are just comparing times,
    but it is needed so the method signatures are the same
    """
    time_tuple = time.localtime()

    if self.suffix_time != time.strftime(self.suffix, time_tuple) or not os.path.exists(
            self.baseFilename + '.' + self.suffix_time):
        return 1
    else:
        return 0

def build_base_filename(self):
    """
    do builder; in this case,
    old time stamp is removed from filename and
    a new time stamp is append to the filename
    """
    if self.stream:
        self.stream.close()
        self.stream = None

    # remove old suffix
    if self.suffix_time != "":
        index = self.baseFilename.find("." + self.suffix_time)
        if index == -1:
            index = self.baseFilename.rfind(".")
        self.baseFilename = self.baseFilename[:index]

    # add new suffix
    current_time_tuple = time.localtime()
    self.suffix_time = time.strftime(self.suffix, current_time_tuple)
    self.baseFilename = self.baseFilename + "." + self.suffix_time

    self.mode = 'a'
    if not self.delay:
        self.stream = self._open()