/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/progress.py

  • Committer: Robert Collins
  • Date: 2010-05-06 11:08:10 UTC
  • mto: This revision was merged to the branch mainline in revision 5223.
  • Revision ID: robertc@robertcollins.net-20100506110810-h3j07fh5gmw54s25
Cleaner matcher matching revised unlocking protocol.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006, 2008, 2009 Canonical Ltd
 
1
# Copyright (C) 2005-2010 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
27
27
import os
28
28
 
29
29
 
30
 
from bzrlib import (
31
 
    errors,
32
 
    )
33
 
from bzrlib.trace import mutter
34
30
from bzrlib.symbol_versioning import (
35
 
    deprecated_function,
36
31
    deprecated_in,
 
32
    deprecated_method,
37
33
    )
38
34
 
39
35
 
40
 
# XXX: deprecated; can be removed when the ProgressBar factory is removed
41
36
def _supports_progress(f):
42
 
    """Detect if we can use pretty progress bars on the output stream f.
 
37
    """Detect if we can use pretty progress bars on file F.
43
38
 
44
39
    If this returns true we expect that a human may be looking at that
45
40
    output, and that we can repaint a line to update it.
 
41
 
 
42
    This doesn't check the policy for whether we *should* use them.
46
43
    """
47
44
    isatty = getattr(f, 'isatty', None)
48
45
    if isatty is None:
49
46
        return False
50
47
    if not isatty():
51
48
        return False
 
49
    # The following case also handles Win32 - on that platform $TERM is
 
50
    # typically never set, so the case None is treated as a smart terminal,
 
51
    # not dumb.  <https://bugs.launchpad.net/bugs/334808>  win32 files do have
 
52
    # isatty methods that return true.
52
53
    if os.environ.get('TERM') == 'dumb':
53
54
        # e.g. emacs compile window
54
55
        return False
64
65
    Code updating the task may also set fields as hints about how to display
65
66
    it: show_pct, show_spinner, show_eta, show_count, show_bar.  UIs
66
67
    will not necessarily respect all these fields.
 
68
    
 
69
    :ivar update_latency: The interval (in seconds) at which the PB should be
 
70
        updated.  Setting this to zero suggests every update should be shown
 
71
        synchronously.
 
72
 
 
73
    :ivar show_transport_activity: If true (default), transport activity
 
74
        will be shown when this task is drawn.  Disable it if you're sure 
 
75
        that only irrelevant or uninteresting transport activity can occur
 
76
        during this task.
67
77
    """
68
78
 
69
 
    def __init__(self, parent_task=None, ui_factory=None):
 
79
    def __init__(self, parent_task=None, ui_factory=None, progress_view=None):
70
80
        """Construct a new progress task.
71
81
 
 
82
        :param parent_task: Enclosing ProgressTask or None.
 
83
 
 
84
        :param progress_view: ProgressView to display this ProgressTask.
 
85
 
 
86
        :param ui_factory: The UI factory that will display updates; 
 
87
            deprecated in favor of passing progress_view directly.
 
88
 
72
89
        Normally you should not call this directly but rather through
73
90
        `ui_factory.nested_progress_bar`.
74
91
        """
77
94
        self.total_cnt = None
78
95
        self.current_cnt = None
79
96
        self.msg = ''
 
97
        # TODO: deprecate passing ui_factory
80
98
        self.ui_factory = ui_factory
 
99
        self.progress_view = progress_view
81
100
        self.show_pct = False
82
101
        self.show_spinner = True
83
102
        self.show_eta = False,
84
103
        self.show_count = True
85
104
        self.show_bar = True
 
105
        self.update_latency = 0.1
 
106
        self.show_transport_activity = True
86
107
 
87
108
    def __repr__(self):
88
109
        return '%s(%r/%r, msg=%r)' % (
96
117
        self.current_cnt = current_cnt
97
118
        if total_cnt:
98
119
            self.total_cnt = total_cnt
99
 
        self.ui_factory._progress_updated(self)
 
120
        if self.progress_view:
 
121
            self.progress_view.show_progress(self)
 
122
        else:
 
123
            self.ui_factory._progress_updated(self)
100
124
 
101
125
    def tick(self):
102
126
        self.update(self.msg)
103
127
 
104
128
    def finished(self):
105
 
        self.ui_factory._progress_finished(self)
 
129
        if self.progress_view:
 
130
            self.progress_view.task_finished(self)
 
131
        else:
 
132
            self.ui_factory._progress_finished(self)
106
133
 
107
134
    def make_sub_task(self):
108
 
        return ProgressTask(self, self.ui_factory)
 
135
        return ProgressTask(self, ui_factory=self.ui_factory,
 
136
            progress_view=self.progress_view)
109
137
 
110
138
    def _overall_completion_fraction(self, child_fraction=0.0):
111
139
        """Return fractional completion of this task and its parents
124
152
                own_fraction = 0.0
125
153
            return self._parent_task._overall_completion_fraction(own_fraction)
126
154
 
 
155
    @deprecated_method(deprecated_in((2, 1, 0)))
127
156
    def note(self, fmt_string, *args):
128
 
        """Record a note without disrupting the progress bar."""
129
 
        # XXX: shouldn't be here; put it in mutter or the ui instead
 
157
        """Record a note without disrupting the progress bar.
 
158
        
 
159
        Deprecated: use ui_factory.note() instead or bzrlib.trace.  Note that
 
160
        ui_factory.note takes just one string as the argument, not a format
 
161
        string and arguments.
 
162
        """
130
163
        if args:
131
164
            self.ui_factory.note(fmt_string % args)
132
165
        else:
133
166
            self.ui_factory.note(fmt_string)
134
167
 
135
168
    def clear(self):
136
 
        # XXX: shouldn't be here; put it in mutter or the ui instead
137
 
        self.ui_factory.clear_term()
138
 
 
139
 
 
140
 
@deprecated_function(deprecated_in((1, 16, 0)))
141
 
def ProgressBar(to_file=None, **kwargs):
142
 
    """Abstract factory"""
143
 
    if to_file is None:
144
 
        to_file = sys.stderr
145
 
    requested_bar_type = os.environ.get('BZR_PROGRESS_BAR')
146
 
    # An value of '' or not set reverts to standard processing
147
 
    if requested_bar_type in (None, ''):
148
 
        if _supports_progress(to_file):
149
 
            return TTYProgressBar(to_file=to_file, **kwargs)
 
169
        # TODO: deprecate this method; the model object shouldn't be concerned
 
170
        # with whether it's shown or not.  Most callers use this because they
 
171
        # want to write some different non-progress output to the screen, but
 
172
        # they should probably instead use a stream that's synchronized with
 
173
        # the progress output.  It may be there is a model-level use for
 
174
        # saying "this task's not active at the moment" but I don't see it. --
 
175
        # mbp 20090623
 
176
        if self.progress_view:
 
177
            self.progress_view.clear()
150
178
        else:
151
 
            return DummyProgress(to_file=to_file, **kwargs)
152
 
    else:
153
 
        # Minor sanitation to prevent spurious errors
154
 
        requested_bar_type = requested_bar_type.lower().strip()
155
 
        # TODO: jam 20060710 Arguably we shouldn't raise an exception
156
 
        #       but should instead just disable progress bars if we
157
 
        #       don't recognize the type
158
 
        if requested_bar_type not in _progress_bar_types:
159
 
            raise errors.InvalidProgressBarType(requested_bar_type,
160
 
                                                _progress_bar_types.keys())
161
 
        return _progress_bar_types[requested_bar_type](to_file=to_file, **kwargs)
162
 
 
163
 
 
 
179
            self.ui_factory.clear_term()
 
180
 
 
181
 
 
182
# NOTE: This is also deprecated; you should provide a ProgressView instead.
164
183
class _BaseProgressBar(object):
165
184
 
166
185
    def __init__(self,
207
226
        self.to_messages_file.write(fmt_string % args)
208
227
        self.to_messages_file.write('\n')
209
228
 
210
 
    @deprecated_function(deprecated_in((1, 16, 0)))
211
 
    def child_progress(self, **kwargs):
212
 
        return ChildProgress(**kwargs)
213
 
 
214
 
 
215
 
class DummyProgress(_BaseProgressBar):
 
229
 
 
230
class DummyProgress(object):
216
231
    """Progress-bar standin that does nothing.
217
232
 
218
 
    This can be used as the default argument for methods that
219
 
    take an optional progress indicator."""
 
233
    This was previously often constructed by application code if no progress
 
234
    bar was explicitly passed in.  That's no longer recommended: instead, just
 
235
    create a progress task from the ui_factory.  This class can be used in
 
236
    test code that needs to fake a progress task for some reason.
 
237
    """
220
238
 
221
239
    def tick(self):
222
240
        pass
237
255
        return DummyProgress(**kwargs)
238
256
 
239
257
 
240
 
class DotsProgressBar(_BaseProgressBar):
241
 
 
242
 
    @deprecated_function(deprecated_in((1, 16, 0)))
243
 
    def __init__(self, **kwargs):
244
 
        _BaseProgressBar.__init__(self, **kwargs)
245
 
        self.last_msg = None
246
 
        self.need_nl = False
247
 
 
248
 
    def tick(self):
249
 
        self.update()
250
 
 
251
 
    def update(self, msg=None, current_cnt=None, total_cnt=None):
252
 
        if msg and msg != self.last_msg:
253
 
            if self.need_nl:
254
 
                self.to_file.write('\n')
255
 
            self.to_file.write(msg + ': ')
256
 
            self.last_msg = msg
257
 
        self.need_nl = True
258
 
        self.to_file.write('.')
259
 
 
260
 
    def clear(self):
261
 
        if self.need_nl:
262
 
            self.to_file.write('\n')
263
 
        self.need_nl = False
264
 
 
265
 
    def child_update(self, message, current, total):
266
 
        self.tick()
267
 
 
268
 
 
269
 
class TTYProgressBar(_BaseProgressBar):
270
 
    """Progress bar display object.
271
 
 
272
 
    Several options are available to control the display.  These can
273
 
    be passed as parameters to the constructor or assigned at any time:
274
 
 
275
 
    show_pct
276
 
        Show percentage complete.
277
 
    show_spinner
278
 
        Show rotating baton.  This ticks over on every update even
279
 
        if the values don't change.
280
 
    show_eta
281
 
        Show predicted time-to-completion.
282
 
    show_bar
283
 
        Show bar graph.
284
 
    show_count
285
 
        Show numerical counts.
286
 
 
287
 
    The output file should be in line-buffered or unbuffered mode.
288
 
    """
289
 
    SPIN_CHARS = r'/-\|'
290
 
 
291
 
    @deprecated_function(deprecated_in((1, 16, 0)))
292
 
    def __init__(self, **kwargs):
293
 
        from bzrlib.osutils import terminal_width
294
 
        _BaseProgressBar.__init__(self, **kwargs)
295
 
        self.spin_pos = 0
296
 
        self.width = terminal_width()
297
 
        self.last_updates = []
298
 
        self._max_last_updates = 10
299
 
        self.child_fraction = 0
300
 
        self._have_output = False
301
 
 
302
 
    def throttle(self, old_msg):
303
 
        """Return True if the bar was updated too recently"""
304
 
        # time.time consistently takes 40/4000 ms = 0.01 ms.
305
 
        # time.clock() is faster, but gives us CPU time, not wall-clock time
306
 
        now = time.time()
307
 
        if self.start_time is not None and (now - self.start_time) < 1:
308
 
            return True
309
 
        if old_msg != self.last_msg:
310
 
            return False
311
 
        interval = now - self.last_update
312
 
        # if interval > 0
313
 
        if interval < self.MIN_PAUSE:
314
 
            return True
315
 
 
316
 
        self.last_updates.append(now - self.last_update)
317
 
        # Don't let the queue grow without bound
318
 
        self.last_updates = self.last_updates[-self._max_last_updates:]
319
 
        self.last_update = now
320
 
        return False
321
 
 
322
 
    def tick(self):
323
 
        self.update(self.last_msg, self.last_cnt, self.last_total,
324
 
                    self.child_fraction)
325
 
 
326
 
    def child_update(self, message, current, total):
327
 
        if current is not None and total != 0:
328
 
            child_fraction = float(current) / total
329
 
            if self.last_cnt is None:
330
 
                pass
331
 
            elif self.last_cnt + child_fraction <= self.last_total:
332
 
                self.child_fraction = child_fraction
333
 
        if self.last_msg is None:
334
 
            self.last_msg = ''
335
 
        self.tick()
336
 
 
337
 
    def update(self, msg, current_cnt=None, total_cnt=None,
338
 
            child_fraction=0):
339
 
        """Update and redraw progress bar.
340
 
        """
341
 
        if msg is None:
342
 
            msg = self.last_msg
343
 
 
344
 
        if total_cnt is None:
345
 
            total_cnt = self.last_total
346
 
 
347
 
        if current_cnt < 0:
348
 
            current_cnt = 0
349
 
 
350
 
        if current_cnt > total_cnt:
351
 
            total_cnt = current_cnt
352
 
 
353
 
        ## # optional corner case optimisation
354
 
        ## # currently does not seem to fire so costs more than saved.
355
 
        ## # trivial optimal case:
356
 
        ## # NB if callers are doing a clear and restore with
357
 
        ## # the saved values, this will prevent that:
358
 
        ## # in that case add a restore method that calls
359
 
        ## # _do_update or some such
360
 
        ## if (self.last_msg == msg and
361
 
        ##     self.last_cnt == current_cnt and
362
 
        ##     self.last_total == total_cnt and
363
 
        ##     self.child_fraction == child_fraction):
364
 
        ##     return
365
 
 
366
 
        if msg is None:
367
 
            msg = ''
368
 
 
369
 
        old_msg = self.last_msg
370
 
        # save these for the tick() function
371
 
        self.last_msg = msg
372
 
        self.last_cnt = current_cnt
373
 
        self.last_total = total_cnt
374
 
        self.child_fraction = child_fraction
375
 
 
376
 
        # each function call takes 20ms/4000 = 0.005 ms,
377
 
        # but multiple that by 4000 calls -> starts to cost.
378
 
        # so anything to make this function call faster
379
 
        # will improve base 'diff' time by up to 0.1 seconds.
380
 
        if self.throttle(old_msg):
381
 
            return
382
 
 
383
 
        if self.show_eta and self.start_time and self.last_total:
384
 
            eta = get_eta(self.start_time, self.last_cnt + self.child_fraction,
385
 
                    self.last_total, last_updates = self.last_updates)
386
 
            eta_str = " " + str_tdelta(eta)
387
 
        else:
388
 
            eta_str = ""
389
 
 
390
 
        if self.show_spinner:
391
 
            spin_str = self.SPIN_CHARS[self.spin_pos % 4] + ' '
392
 
        else:
393
 
            spin_str = ''
394
 
 
395
 
        # always update this; it's also used for the bar
396
 
        self.spin_pos += 1
397
 
 
398
 
        if self.show_pct and self.last_total and self.last_cnt:
399
 
            pct = 100.0 * ((self.last_cnt + self.child_fraction) / self.last_total)
400
 
            pct_str = ' (%5.1f%%)' % pct
401
 
        else:
402
 
            pct_str = ''
403
 
 
404
 
        if not self.show_count:
405
 
            count_str = ''
406
 
        elif self.last_cnt is None:
407
 
            count_str = ''
408
 
        elif self.last_total is None:
409
 
            count_str = ' %i' % (self.last_cnt)
410
 
        else:
411
 
            # make both fields the same size
412
 
            t = '%i' % (self.last_total)
413
 
            c = '%*i' % (len(t), self.last_cnt)
414
 
            count_str = ' ' + c + '/' + t
415
 
 
416
 
        if self.show_bar:
417
 
            # progress bar, if present, soaks up all remaining space
418
 
            cols = self.width - 1 - len(self.last_msg) - len(spin_str) - len(pct_str) \
419
 
                   - len(eta_str) - len(count_str) - 3
420
 
 
421
 
            if self.last_total:
422
 
                # number of markers highlighted in bar
423
 
                markers = int(round(float(cols) *
424
 
                              (self.last_cnt + self.child_fraction) / self.last_total))
425
 
                bar_str = '[' + ('=' * markers).ljust(cols) + '] '
426
 
            elif False:
427
 
                # don't know total, so can't show completion.
428
 
                # so just show an expanded spinning thingy
429
 
                m = self.spin_pos % cols
430
 
                ms = (' ' * m + '*').ljust(cols)
431
 
 
432
 
                bar_str = '[' + ms + '] '
433
 
            else:
434
 
                bar_str = ''
435
 
        else:
436
 
            bar_str = ''
437
 
 
438
 
        m = spin_str + bar_str + self.last_msg + count_str \
439
 
            + pct_str + eta_str
440
 
        self.to_file.write('\r%-*.*s' % (self.width - 1, self.width - 1, m))
441
 
        self._have_output = True
442
 
        #self.to_file.flush()
443
 
 
444
 
    def clear(self):
445
 
        if self._have_output:
446
 
            self.to_file.write('\r%s\r' % (' ' * (self.width - 1)))
447
 
        self._have_output = False
448
 
        #self.to_file.flush()
449
 
 
450
 
 
451
 
class ChildProgress(_BaseProgressBar):
452
 
    """A progress indicator that pushes its data to the parent"""
453
 
 
454
 
    @deprecated_function(deprecated_in((1, 16, 0)))
455
 
    def __init__(self, _stack, **kwargs):
456
 
        _BaseProgressBar.__init__(self, _stack=_stack, **kwargs)
457
 
        self.parent = _stack.top()
458
 
        self.current = None
459
 
        self.total = None
460
 
        self.child_fraction = 0
461
 
        self.message = None
462
 
 
463
 
    def update(self, msg, current_cnt=None, total_cnt=None):
464
 
        self.current = current_cnt
465
 
        if total_cnt is not None:
466
 
            self.total = total_cnt
467
 
        self.message = msg
468
 
        self.child_fraction = 0
469
 
        self.tick()
470
 
 
471
 
    def child_update(self, message, current, total):
472
 
        if current is None or total == 0:
473
 
            self.child_fraction = 0
474
 
        else:
475
 
            self.child_fraction = float(current) / total
476
 
        self.tick()
477
 
 
478
 
    def tick(self):
479
 
        if self.current is None:
480
 
            count = None
481
 
        else:
482
 
            count = self.current+self.child_fraction
483
 
            if count > self.total:
484
 
                if __debug__:
485
 
                    mutter('clamping count of %d to %d' % (count, self.total))
486
 
                count = self.total
487
 
        self.parent.child_update(self.message, count, self.total)
488
 
 
489
 
    def clear(self):
490
 
        pass
491
 
 
492
 
    def note(self, *args, **kwargs):
493
 
        self.parent.note(*args, **kwargs)
494
 
 
495
 
 
496
258
def str_tdelta(delt):
497
259
    if delt is None:
498
260
        return "-:--:--"
549
311
        else:
550
312
            self.cur_phase += 1
551
313
        self.pb.update(self.message, self.cur_phase, self.total)
552
 
 
553
 
 
554
 
_progress_bar_types = {}
555
 
_progress_bar_types['dummy'] = DummyProgress
556
 
_progress_bar_types['none'] = DummyProgress
557
 
_progress_bar_types['tty'] = TTYProgressBar
558
 
_progress_bar_types['dots'] = DotsProgressBar