bootstrap-vz/bootstrapvz/remote/log.py
2015-04-16 22:15:19 +02:00

32 lines
813 B
Python

import logging
class LogForwarder(logging.Handler):
def __init__(self, level=logging.NOTSET):
self.server = None
super(LogForwarder, self).__init__(level)
def set_server(self, server):
self.server = server
def emit(self, record):
if self.server is not None:
if record.exc_info is not None:
import traceback
exc_type, exc_value, exc_traceback = record.exc_info
record.exc_info = traceback.print_exception(exc_type, exc_value, exc_traceback)
# TODO: Use serpent instead
import pickle
self.server.handle(pickle.dumps(record))
class LogServer(object):
def handle(self, pickled_record):
import pickle
record = pickle.loads(pickled_record)
log = logging.getLogger()
record.extra = getattr(record, 'extra', {})
record.extra['source'] = 'remote'
log.handle(record)