119 lines
3.0 KiB
Python
119 lines
3.0 KiB
Python
import sys, os
|
|
import logging
|
|
import traceback
|
|
from cStringIO import StringIO
|
|
|
|
import dbus
|
|
import dbus.dbus_bindings
|
|
import gobject
|
|
|
|
__queue = None
|
|
|
|
CONSOLE_BUS_NAME = 'org.laptop.Sugar.Console'
|
|
CONSOLE_OBJECT_PATH = '/org/laptop/Sugar/Console'
|
|
CONSOLE_IFACE = 'org.laptop.Sugar.Console'
|
|
|
|
class MessageQueue:
|
|
def __init__(self, console, console_id):
|
|
self._idle_id = 0
|
|
self._console = console
|
|
self._console_id = console_id
|
|
self._levels = []
|
|
self._messages = []
|
|
self._bus = dbus.SessionBus()
|
|
|
|
if self._console == None:
|
|
con = self._bus._connection
|
|
if dbus.dbus_bindings.bus_name_has_owner(con, CONSOLE_BUS_NAME):
|
|
self.setup_console()
|
|
else:
|
|
self._bus.add_signal_receiver(
|
|
self.__name_owner_changed,
|
|
dbus_interface = "org.freedesktop.DBus",
|
|
signal_name = "NameOwnerChanged")
|
|
|
|
def setup_console(self):
|
|
proxy_obj = self._bus.get_object(CONSOLE_BUS_NAME,
|
|
CONSOLE_OBJECT_PATH)
|
|
self._console = dbus.Interface(proxy_obj, CONSOLE_IFACE)
|
|
self._queue_log()
|
|
|
|
def __name_owner_changed(self, service_name, old_name, new_name):
|
|
if service_name == CONSOLE_BUS_NAME:
|
|
if new_name != None:
|
|
self.setup_console()
|
|
else:
|
|
self._console = None
|
|
|
|
def _queue_log(self):
|
|
if self._idle_id == 0:
|
|
self._idle_id = gobject.idle_add(self._log)
|
|
|
|
def _log(self):
|
|
# Use stderr for now until dbus issues get sorted out
|
|
use_dbus = False
|
|
|
|
if use_dbus:
|
|
if self._console == None or len(self._messages) == 0:
|
|
self._idle_id = 0
|
|
return False
|
|
|
|
if isinstance(self._console, dbus.Interface):
|
|
self._console.log(self._console_id, self._levels,
|
|
self._messages, timeout = 1000)
|
|
else:
|
|
self._console.log(self._console_id, self._levels,
|
|
self._messages)
|
|
else:
|
|
for x in range(0, len(self._messages)):
|
|
level = self._levels[x]
|
|
msg = self._messages[x]
|
|
prog = os.path.basename(sys.argv[0])
|
|
fmt = "%s (%s): Level %s - %s\n" % (prog, os.getpid(), level, msg)
|
|
fmt = fmt.encode("utf8")
|
|
sys.stderr.write(fmt)
|
|
|
|
self._levels = []
|
|
self._messages = []
|
|
self._idle_id = 0
|
|
|
|
return False
|
|
|
|
def append_record(self, record):
|
|
self.append(record.levelno, record.msg)
|
|
|
|
def append(self, level, message):
|
|
self._levels.append(level)
|
|
self._messages.append(message)
|
|
self._queue_log()
|
|
|
|
class Handler(logging.Handler):
|
|
def __init__(self, queue):
|
|
logging.Handler.__init__(self)
|
|
|
|
self._queue = queue
|
|
|
|
def emit(self, record):
|
|
self._queue.append_record(record)
|
|
|
|
def __exception_handler(typ, exc, tb):
|
|
trace = StringIO()
|
|
traceback.print_exception(typ, exc, tb, None, trace)
|
|
print >> sys.stderr, trace.getvalue()
|
|
|
|
__queue.append(logging.ERROR, trace.getvalue())
|
|
|
|
def start(console_id, console = None):
|
|
queue = MessageQueue(console, console_id)
|
|
|
|
root_logger = logging.getLogger('')
|
|
root_logger.setLevel(logging.DEBUG)
|
|
root_logger.addHandler(Handler(queue))
|
|
fileh = logging.FileHandler('/tmp/sugar.log')
|
|
fileh.setFormatter(logging.Formatter("""[%(asctime)s] %(message)s"""))
|
|
root_logger.addHandler(fileh)
|
|
|
|
global __queue
|
|
__queue = queue
|
|
sys.excepthook = __exception_handler
|