1
# Copyright (C) 2005-2011 Canonical Ltd
1
# Copyright (C) 2005-2010 Canonical Ltd
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
121
122
# FIXME: clearing the ui and then going through the abstract logging
122
123
# framework is whack; we should probably have a logging Handler that
123
124
# deals with terminal output if needed.
124
ui.ui_factory.clear_term()
126
bzrlib.ui.ui_factory.clear_term()
125
127
_bzr_logger.info(*args, **kwargs)
128
130
def warning(*args, **kwargs):
129
ui.ui_factory.clear_term()
132
bzrlib.ui.ui_factory.clear_term()
130
133
_bzr_logger.warning(*args, **kwargs)
303
306
logging.getLogger("bzr").
305
308
Output can be redirected away by calling _push_log_file.
307
:return: A memento from push_log_file for restoring the log state.
310
# Do this before we open the log file, so we prevent
311
# get_terminal_encoding() from mutter()ing multiple times
312
term_encoding = osutils.get_terminal_encoding()
309
313
start_time = osutils.format_local_date(_bzr_log_start_time,
310
314
timezone='local')
311
315
# create encoded wrapper around stderr
312
316
bzr_log_file = _open_bzr_log()
313
317
if bzr_log_file is not None:
314
318
bzr_log_file.write(start_time.encode('utf-8') + '\n')
315
memento = push_log_file(bzr_log_file,
319
push_log_file(bzr_log_file,
316
320
r'[%(process)5d] %(asctime)s.%(msecs)03d %(levelname)s: %(message)s',
317
321
r'%Y-%m-%d %H:%M:%S')
318
322
# after hooking output into bzr_log, we also need to attach a stderr
319
323
# handler, writing only at level info and with encoding
320
term_encoding = osutils.get_terminal_encoding()
321
324
writer_factory = codecs.getwriter(term_encoding)
322
325
encoded_stderr = writer_factory(sys.stderr, errors='replace')
323
326
stderr_handler = logging.StreamHandler(encoded_stderr)
324
327
stderr_handler.setLevel(logging.INFO)
325
328
logging.getLogger('bzr').addHandler(stderr_handler)
329
331
def push_log_file(to_file, log_format=None, date_format=None):
365
367
def pop_log_file((magic, old_handlers, new_handler, old_trace_file, new_trace_file)):
366
368
"""Undo changes to logging/tracing done by _push_log_file.
368
This flushes, but does not close the trace file (so that anything that was
370
This flushes, but does not close the trace file.
371
372
Takes the memento returned from _push_log_file."""
372
373
global _trace_file
373
374
_trace_file = old_trace_file
374
375
bzr_logger = logging.getLogger('bzr')
375
376
bzr_logger.removeHandler(new_handler)
376
# must be closed, otherwise logging will try to close it at exit, and the
377
# must be closed, otherwise logging will try to close it atexit, and the
377
378
# file will likely already be closed underneath.
378
379
new_handler.close()
379
380
bzr_logger.handlers = old_handlers
380
if new_trace_file is not None:
381
new_trace_file.flush()
381
new_trace_file.flush()
384
384
def log_exception_quietly():
469
def _dump_memory_usage(err_file):
472
fd, name = tempfile.mkstemp(prefix="bzr_memdump", suffix=".json")
473
dump_file = os.fdopen(fd, 'w')
474
from meliae import scanner
475
scanner.dump_gc_objects(dump_file)
476
err_file.write("Memory dumped to %s\n" % name)
478
err_file.write("Dumping memory requires meliae module.\n")
479
log_exception_quietly()
481
err_file.write("Exception while dumping memory.\n")
482
log_exception_quietly()
484
if dump_file is not None:
490
def _qualified_exception_name(eclass, unqualified_bzrlib_errors=False):
491
"""Give name of error class including module for non-builtin exceptions
493
If `unqualified_bzrlib_errors` is True, errors specific to bzrlib will
494
also omit the module prefix.
496
class_name = eclass.__name__
497
module_name = eclass.__module__
498
if module_name in ("exceptions", "__main__") or (
499
unqualified_bzrlib_errors and module_name == "bzrlib.errors"):
501
return "%s.%s" % (module_name, class_name)
504
470
def report_exception(exc_info, err_file):
505
471
"""Report an exception to err_file (typically stderr) and to .bzr.log.
523
489
return errors.EXIT_ERROR
524
490
elif isinstance(exc_object, MemoryError):
525
491
err_file.write("bzr: out of memory\n")
526
if 'mem_dump' in debug.debug_flags:
527
_dump_memory_usage(err_file)
529
err_file.write("Use -Dmem_dump to dump memory to a file.\n")
530
492
return errors.EXIT_ERROR
531
493
elif isinstance(exc_object, ImportError) \
532
494
and str(exc_object).startswith("No module named "):
593
555
def _flush_trace():
594
# called from the bzrlib library finalizer returned by bzrlib.initialize()
556
# run from atexit hook
595
557
global _trace_file
597
559
_trace_file.flush()
600
class Config(object):
601
"""Configuration of message tracing in bzrlib.
603
This implements the context manager protocol and should manage any global
604
variables still used. The default config used is DefaultConfig, but
605
embedded uses of bzrlib may wish to use a custom manager.
609
return self # This is bound to the 'as' clause in a with statement.
611
def __exit__(self, exc_type, exc_val, exc_tb):
612
return False # propogate exceptions.
615
class DefaultConfig(Config):
616
"""A default configuration for tracing of messages in bzrlib.
618
This implements the context manager protocol.
622
self._original_filename = _bzr_log_filename
623
self._original_state = enable_default_logging()
624
return self # This is bound to the 'as' clause in a with statement.
626
def __exit__(self, exc_type, exc_val, exc_tb):
627
pop_log_file(self._original_state)
628
global _bzr_log_filename
629
_bzr_log_filename = self._original_filename
630
return False # propogate exceptions.