~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/trace.py

  • Committer: Joe Julian
  • Date: 2010-01-10 02:25:31 UTC
  • mto: (4634.119.7 2.0)
  • mto: This revision was merged to the branch mainline in revision 4959.
  • Revision ID: joe@julianfamily.org-20100110022531-wqk61rsagz8xsiga
Added MANIFEST.in to allow bdist_rpm to have all the required include files and tools. bdist_rpm will still fail to build correctly on some distributions due to a disttools bug http://bugs.python.org/issue644744

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2010 Canonical Ltd
 
1
# Copyright (C) 2005, 2006, 2007, 2008 Canonical Ltd
2
2
#
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
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
 
"""Messages and logging.
 
17
"""Messages and logging for bazaar-ng.
18
18
 
19
19
Messages are supplied by callers as a string-formatting template, plus values
20
20
to be inserted into it.  The actual %-formatting is deferred to the log
33
33
 
34
34
Output to stderr depends on the mode chosen by the user.  By default, messages
35
35
of info and above are sent out, which results in progress messages such as the
36
 
list of files processed by add and commit.  In debug mode, stderr gets debug messages too.
 
36
list of files processed by add and commit.  In quiet mode, only warnings and
 
37
above are shown.  In debug mode, stderr gets debug messages too.
37
38
 
38
39
Errors that terminate an operation are generally passed back as exceptions;
39
40
others may be just emitted as messages.
58
59
import logging
59
60
import os
60
61
import sys
 
62
import re
61
63
import time
62
64
 
63
65
from bzrlib.lazy_import import lazy_import
70
72
 
71
73
import bzrlib
72
74
 
73
 
from bzrlib.symbol_versioning import (
74
 
    deprecated_function,
75
 
    deprecated_in,
76
 
    )
77
 
 
78
75
lazy_import(globals(), """
79
76
from bzrlib import (
80
77
    debug,
82
79
    osutils,
83
80
    plugin,
84
81
    symbol_versioning,
85
 
    ui,
86
82
    )
87
83
""")
88
84
 
111
107
 
112
108
 
113
109
def note(*args, **kwargs):
114
 
    """Output a note to the user.
115
 
 
116
 
    Takes the same parameters as logging.info.
117
 
 
118
 
    :return: None
119
 
    """
120
110
    # FIXME note always emits utf-8, regardless of the terminal encoding
121
111
    #
122
112
    # FIXME: clearing the ui and then going through the abstract logging
133
123
    _bzr_logger.warning(*args, **kwargs)
134
124
 
135
125
 
136
 
@deprecated_function(deprecated_in((2, 1, 0)))
137
 
def info(*args, **kwargs):
138
 
    """Deprecated: use trace.note instead."""
139
 
    note(*args, **kwargs)
140
 
 
141
 
 
142
 
@deprecated_function(deprecated_in((2, 1, 0)))
143
 
def log_error(*args, **kwargs):
144
 
    """Deprecated: use bzrlib.trace.show_error instead"""
145
 
    _bzr_logger.error(*args, **kwargs)
146
 
 
147
 
 
148
 
@deprecated_function(deprecated_in((2, 1, 0)))
149
 
def error(*args, **kwargs):
150
 
    """Deprecated: use bzrlib.trace.show_error instead"""
151
 
    _bzr_logger.error(*args, **kwargs)
152
 
 
153
 
 
154
 
def show_error(*args, **kwargs):
155
 
    """Show an error message to the user.
156
 
 
157
 
    Don't use this for exceptions, use report_exception instead.
158
 
    """
159
 
    _bzr_logger.error(*args, **kwargs)
160
 
 
 
126
# configure convenient aliases for output routines
 
127
#
 
128
# TODO: deprecate them, have one name for each.
 
129
info = note
 
130
log_error = _bzr_logger.error
 
131
error =     _bzr_logger.error
 
132
 
 
133
 
 
134
_last_mutter_flush_time = None
161
135
 
162
136
def mutter(fmt, *args):
 
137
    global _last_mutter_flush_time
163
138
    if _trace_file is None:
164
139
        return
165
 
    # XXX: Don't check this every time; instead anyone who closes the file
166
 
    # ought to deregister it.  We can tolerate None.
167
140
    if (getattr(_trace_file, 'closed', None) is not None) and _trace_file.closed:
168
141
        return
169
142
 
186
159
    timestamp = '%0.3f  ' % (now - _bzr_log_start_time,)
187
160
    out = timestamp + out + '\n'
188
161
    _trace_file.write(out)
189
 
    # there's no explicit flushing; the file is typically line buffered.
 
162
    # We flush if we haven't flushed for a few seconds. We don't want to flush
 
163
    # on every mutter, but when a command takes a while, it can be nice to see
 
164
    # updates in the debug log.
 
165
    if (_last_mutter_flush_time is None
 
166
        or (now - _last_mutter_flush_time) > 2.0):
 
167
        flush = getattr(_trace_file, 'flush', None)
 
168
        if flush is not None:
 
169
            flush()
 
170
        _last_mutter_flush_time = now
190
171
 
191
172
 
192
173
def mutter_callsite(stacklevel, fmt, *args):
244
225
    This sets the global _bzr_log_filename.
245
226
    """
246
227
    global _bzr_log_filename
247
 
 
248
 
    def _open_or_create_log_file(filename):
249
 
        """Open existing log file, or create with ownership and permissions
250
 
 
251
 
        It inherits the ownership and permissions (masked by umask) from
252
 
        the containing directory to cope better with being run under sudo
253
 
        with $HOME still set to the user's homedir.
254
 
        """
255
 
        flags = os.O_WRONLY | os.O_APPEND | osutils.O_TEXT
256
 
        while True:
257
 
            try:
258
 
                fd = os.open(filename, flags)
259
 
                break
260
 
            except OSError, e:
261
 
                if e.errno != errno.ENOENT:
262
 
                    raise
263
 
            try:
264
 
                fd = os.open(filename, flags | os.O_CREAT | os.O_EXCL, 0666)
265
 
            except OSError, e:
266
 
                if e.errno != errno.EEXIST:
267
 
                    raise
268
 
            else:
269
 
                osutils.copy_ownership_from_path(filename)
270
 
                break
271
 
        return os.fdopen(fd, 'at', 0) # unbuffered
272
 
 
273
 
 
274
228
    _bzr_log_filename = _get_bzr_log_filename()
275
229
    _rollover_trace_maybe(_bzr_log_filename)
276
230
    try:
277
 
        bzr_log_file = _open_or_create_log_file(_bzr_log_filename)
 
231
        bzr_log_file = open(_bzr_log_filename, 'at', 1) # line buffered
 
232
        # bzr_log_file.tell() on windows always return 0 until some writing done
278
233
        bzr_log_file.write('\n')
279
234
        if bzr_log_file.tell() <= 2:
280
235
            bzr_log_file.write("this is a debug log for diagnosing/reporting problems in bzr\n")
281
236
            bzr_log_file.write("you can delete or truncate this file, or include sections in\n")
282
237
            bzr_log_file.write("bug reports to https://bugs.launchpad.net/bzr/+filebug\n\n")
283
 
 
284
238
        return bzr_log_file
285
 
 
286
 
    except EnvironmentError, e:
 
239
    except IOError, e:
287
240
        # If we are failing to open the log, then most likely logging has not
288
241
        # been set up yet. So we just write to stderr rather than using
289
242
        # 'warning()'. If we using warning(), users get the unhelpful 'no
306
259
    logging.getLogger("bzr").
307
260
 
308
261
    Output can be redirected away by calling _push_log_file.
309
 
 
310
 
    :return: A memento from push_log_file for restoring the log state.
311
262
    """
 
263
    # Do this before we open the log file, so we prevent
 
264
    # get_terminal_encoding() from mutter()ing multiple times
 
265
    term_encoding = osutils.get_terminal_encoding()
312
266
    start_time = osutils.format_local_date(_bzr_log_start_time,
313
267
                                           timezone='local')
314
268
    # create encoded wrapper around stderr
315
269
    bzr_log_file = _open_bzr_log()
316
270
    if bzr_log_file is not None:
317
271
        bzr_log_file.write(start_time.encode('utf-8') + '\n')
318
 
    memento = push_log_file(bzr_log_file,
 
272
    push_log_file(bzr_log_file,
319
273
        r'[%(process)5d] %(asctime)s.%(msecs)03d %(levelname)s: %(message)s',
320
274
        r'%Y-%m-%d %H:%M:%S')
321
275
    # after hooking output into bzr_log, we also need to attach a stderr
322
276
    # handler, writing only at level info and with encoding
323
 
    term_encoding = osutils.get_terminal_encoding()
324
277
    writer_factory = codecs.getwriter(term_encoding)
325
278
    encoded_stderr = writer_factory(sys.stderr, errors='replace')
326
279
    stderr_handler = logging.StreamHandler(encoded_stderr)
327
280
    stderr_handler.setLevel(logging.INFO)
328
281
    logging.getLogger('bzr').addHandler(stderr_handler)
329
 
    return memento
330
282
 
331
283
 
332
284
def push_log_file(to_file, log_format=None, date_format=None):
368
320
def pop_log_file((magic, old_handlers, new_handler, old_trace_file, new_trace_file)):
369
321
    """Undo changes to logging/tracing done by _push_log_file.
370
322
 
371
 
    This flushes, but does not close the trace file (so that anything that was
372
 
    in it is output.
 
323
    This flushes, but does not close the trace file.
373
324
 
374
325
    Takes the memento returned from _push_log_file."""
375
326
    global _trace_file
376
327
    _trace_file = old_trace_file
377
328
    bzr_logger = logging.getLogger('bzr')
378
329
    bzr_logger.removeHandler(new_handler)
379
 
    # must be closed, otherwise logging will try to close it at exit, and the
 
330
    # must be closed, otherwise logging will try to close it atexit, and the
380
331
    # file will likely already be closed underneath.
381
332
    new_handler.close()
382
333
    bzr_logger.handlers = old_handlers
383
 
    if new_trace_file is not None:
384
 
        new_trace_file.flush()
 
334
    new_trace_file.flush()
385
335
 
386
336
 
387
337
def log_exception_quietly():
402
352
    global _verbosity_level
403
353
    _verbosity_level = level
404
354
    _update_logging_level(level < 0)
405
 
    ui.ui_factory.be_quiet(level < 0)
406
355
 
407
356
 
408
357
def get_verbosity_level():
414
363
 
415
364
 
416
365
def be_quiet(quiet=True):
 
366
    # Perhaps this could be deprecated now ...
417
367
    if quiet:
418
368
        set_verbosity_level(-1)
419
369
    else:
477
427
 
478
428
    :return: The appropriate exit code for this error.
479
429
    """
 
430
    exc_type, exc_object, exc_tb = exc_info
480
431
    # Log the full traceback to ~/.bzr.log
481
432
    log_exception_quietly()
482
 
    if 'error' in debug.debug_flags:
483
 
        print_exception(exc_info, err_file)
484
 
        return errors.EXIT_ERROR
485
 
    exc_type, exc_object, exc_tb = exc_info
486
433
    if (isinstance(exc_object, IOError)
487
434
        and getattr(exc_object, 'errno', None) == errno.EPIPE):
488
435
        err_file.write("bzr: broken pipe\n")
501
448
    elif not getattr(exc_object, 'internal_error', True):
502
449
        report_user_error(exc_info, err_file)
503
450
        return errors.EXIT_ERROR
504
 
    elif isinstance(exc_object, (OSError, IOError)) or (
505
 
        # GZ 2010-05-20: Like (exc_type is pywintypes.error) but avoid import
506
 
        exc_type.__name__ == "error" and exc_type.__module__ == "pywintypes"):
 
451
    elif isinstance(exc_object, (OSError, IOError)):
507
452
        # Might be nice to catch all of these and show them as something more
508
453
        # specific, but there are too many cases at the moment.
509
454
        report_user_error(exc_info, err_file)
531
476
    :param advice: Extra advice to the user to be printed following the
532
477
        exception.
533
478
    """
 
479
    if 'error' in debug.debug_flags:
 
480
        print_exception(exc_info, err_file)
 
481
        return
534
482
    err_file.write("bzr: ERROR: %s\n" % (exc_info[1],))
535
483
    if advice:
536
484
        err_file.write("%s\n" % (advice,))
540
488
    """Report an exception that probably indicates a bug in bzr"""
541
489
    from bzrlib.crash import report_bug
542
490
    report_bug(exc_info, err_file)
543
 
 
544
 
 
545
 
def _flush_stdout_stderr():
546
 
    # called from the bzrlib library finalizer returned by bzrlib.initialize()
547
 
    try:
548
 
        sys.stdout.flush()
549
 
        sys.stderr.flush()
550
 
    except IOError, e:
551
 
        import errno
552
 
        if e.errno in [errno.EINVAL, errno.EPIPE]:
553
 
            pass
554
 
        else:
555
 
            raise
556
 
 
557
 
 
558
 
def _flush_trace():
559
 
    # called from the bzrlib library finalizer returned by bzrlib.initialize()
560
 
    global _trace_file
561
 
    if _trace_file:
562
 
        _trace_file.flush()
563
 
 
564
 
 
565
 
class Config(object):
566
 
    """Configuration of message tracing in bzrlib.
567
 
 
568
 
    This implements the context manager protocol and should manage any global
569
 
    variables still used. The default config used is DefaultConfig, but
570
 
    embedded uses of bzrlib may wish to use a custom manager.
571
 
    """
572
 
 
573
 
    def __enter__(self):
574
 
        return self # This is bound to the 'as' clause in a with statement.
575
 
 
576
 
    def __exit__(self, exc_type, exc_val, exc_tb):
577
 
        return False # propogate exceptions.
578
 
 
579
 
 
580
 
class DefaultConfig(Config):
581
 
    """A default configuration for tracing of messages in bzrlib.
582
 
 
583
 
    This implements the context manager protocol.
584
 
    """
585
 
 
586
 
    def __enter__(self):
587
 
        self._original_filename = _bzr_log_filename
588
 
        self._original_state = enable_default_logging()
589
 
        return self # This is bound to the 'as' clause in a with statement.
590
 
 
591
 
    def __exit__(self, exc_type, exc_val, exc_tb):
592
 
        pop_log_file(self._original_state)
593
 
        global _bzr_log_filename
594
 
        _bzr_log_filename = self._original_filename
595
 
        return False # propogate exceptions.