diff options
Diffstat (limited to 'sugar/logger.py')
| -rw-r--r-- | sugar/logger.py | 118 |
1 files changed, 59 insertions, 59 deletions
diff --git a/sugar/logger.py b/sugar/logger.py index b4b2e0c..5eb8039 100644 --- a/sugar/logger.py +++ b/sugar/logger.py @@ -29,82 +29,82 @@ STDOUT_LEVEL = 1000 STDERR_LEVEL = 2000 class LogWriter: - def __init__(self, module_id): - self._module_id = module_id - - logs_dir = _get_logs_dir() - log_path = os.path.join(logs_dir, module_id + '.log') - self._log_file = open(log_path, 'w') - - def write_record(self, record): - self.write(record.levelno, record.msg) - - def write(self, level, msg): - if level == logging.ERROR: - level_txt = 'ERROR' - elif level == logging.WARNING: - level_txt = 'WARNING' - elif level == logging.DEBUG: - level_txt = 'DEBUG' - elif level == logging.INFO: - level_txt = 'INFO' - elif level == STDERR_LEVEL: - level_txt = 'STDERR' - elif level == STDOUT_LEVEL: - level_txt = 'STDOUT' - - fmt = "%s - %s\n" % (level_txt, msg) - fmt = fmt.encode("utf8") - self._log_file.write(fmt) - self._log_file.flush() + def __init__(self, module_id): + self._module_id = module_id + + logs_dir = _get_logs_dir() + log_path = os.path.join(logs_dir, module_id + '.log') + self._log_file = open(log_path, 'w') + + def write_record(self, record): + self.write(record.levelno, record.msg) + + def write(self, level, msg): + if level == logging.ERROR: + level_txt = 'ERROR' + elif level == logging.WARNING: + level_txt = 'WARNING' + elif level == logging.DEBUG: + level_txt = 'DEBUG' + elif level == logging.INFO: + level_txt = 'INFO' + elif level == STDERR_LEVEL: + level_txt = 'STDERR' + elif level == STDOUT_LEVEL: + level_txt = 'STDOUT' + + fmt = "%s - %s\n" % (level_txt, msg) + fmt = fmt.encode("utf8") + self._log_file.write(fmt) + self._log_file.flush() class Handler(logging.Handler): - def __init__(self, writer): - logging.Handler.__init__(self) + def __init__(self, writer): + logging.Handler.__init__(self) - self._writer = writer + self._writer = writer - def emit(self, record): - self._writer.write_record(record) + def emit(self, record): + self._writer.write_record(record) class StdoutCatcher: - def write(self, txt): - _log_writer.write(STDOUT_LEVEL, txt) - sys.__stdout__.write(txt) + def write(self, txt): + _log_writer.write(STDOUT_LEVEL, txt) + sys.__stdout__.write(txt) class StderrCatcher: - def write(self, txt): - _log_writer.write(STDERR_LEVEL, txt) - sys.__stderr__.write(txt) + def write(self, txt): + _log_writer.write(STDERR_LEVEL, txt) + sys.__stderr__.write(txt) def __exception_handler(typ, exc, tb): - trace = StringIO() - traceback.print_exception(typ, exc, tb, None, trace) - print >> sys.stderr, trace.getvalue() + trace = StringIO() + traceback.print_exception(typ, exc, tb, None, trace) + print >> sys.stderr, trace.getvalue() - _log_writer.write(logging.ERROR, trace.getvalue()) + _log_writer.write(logging.ERROR, trace.getvalue()) def _get_logs_dir(): - logs_dir = os.path.join(env.get_profile_path(), 'logs') - if not os.path.isdir(logs_dir): - os.makedirs(logs_dir) - return logs_dir + logs_dir = os.path.join(env.get_profile_path(), 'logs') + if not os.path.isdir(logs_dir): + os.makedirs(logs_dir) + return logs_dir def start(module_id): - log_writer = LogWriter(module_id) + log_writer = LogWriter(module_id) - root_logger = logging.getLogger('') - root_logger.setLevel(logging.DEBUG) - root_logger.addHandler(Handler(log_writer)) + root_logger = logging.getLogger('') + root_logger.setLevel(logging.DEBUG) + root_logger.addHandler(Handler(log_writer)) - sys.stdout = StdoutCatcher() - sys.stderr = StderrCatcher() + sys.stdout = StdoutCatcher() + sys.stderr = StderrCatcher() - global _log_writer - _log_writer = log_writer - sys.excepthook = __exception_handler + global _log_writer + _log_writer = log_writer + sys.excepthook = __exception_handler def cleanup(): - logs_dir = _get_logs_dir() - for f in os.listdir(logs_dir): - os.remove(os.path.join(logs_dir, f)) + logs_dir = _get_logs_dir() + for f in os.listdir(logs_dir): + os.remove(os.path.join(logs_dir, f)) |