blob: 38a3d42399b4672209cc4abc7ee8a17bfa5395b6 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
# Code to implement asynchronous logging from a background thread
#
# Copyright (C) 2016,2017 Kevin O'Connor <kevin@koconnor.net>
#
# This file may be distributed under the terms of the GNU GPLv3 license.
import logging, logging.handlers, threading, Queue
# Class to forward all messages through a queue to a background thread
class QueueHandler(logging.Handler):
def __init__(self, queue):
logging.Handler.__init__(self)
self.queue = queue
def emit(self, record):
try:
self.format(record)
record.msg = record.message
record.args = None
record.exc_info = None
self.queue.put_nowait(record)
except Exception:
self.handleError(record)
# Class to poll a queue in a background thread and log each message
class QueueListener(object):
def __init__(self, handler):
self.handler = handler
self.queue = Queue.Queue()
self.thread = threading.Thread(target=self._bg_thread)
self.thread.start()
def _bg_thread(self):
while 1:
record = self.queue.get(True)
if record is None:
break
self.handler.handle(record)
def stop(self):
self.queue.put_nowait(None)
self.thread.join()
def setup_bg_logging(filename, debuglevel):
handler = logging.handlers.TimedRotatingFileHandler(
filename, when='midnight', backupCount=5)
ql = QueueListener(handler)
qh = QueueHandler(ql.queue)
root = logging.getLogger()
root.addHandler(qh)
root.setLevel(debuglevel)
return ql
|
