← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/progress.py

  • Committer: Sidnei da Silva
  • Date: 2009-07-03 15:06:42 UTC
  • mto: (4531.1.1 integration)
  • mto: This revision was merged to the branch mainline in revision 4532.
  • Revision ID: sidnei.da.silva@canonical.com-20090703150642-hjfra5waj5879cae
- Add top-level make target to build all installers using buildout and another to cleanup

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/progress.py
1
 
# Copyright (C) 2005 Aaron Bentley <aaron.bentley@utoronto.ca>
2
 
# Copyright (C) 2005, 2006 Canonical Ltd
 
1
# Copyright (C) 2005, 2006, 2008, 2009 Canonical Ltd
3
2
#
4
3
# This program is free software; you can redistribute it and/or modify
5
4
# it under the terms of the GNU General Public License as published by
13
12
#
14
13
# You should have received a copy of the GNU General Public License
15
14
# along with this program; if not, write to the Free Software
16
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
17
 
 
18
 
 
19
 
"""Simple text-mode progress indicator.
20
 
 
21
 
To display an indicator, create a ProgressBar object.  Call it,
22
 
passing Progress objects indicating the current state.  When done,
23
 
call clear().
24
 
 
25
 
Progress is suppressed when output is not sent to a terminal, so as
26
 
not to clutter log files.
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
16
 
 
17
 
 
18
"""Progress indicators.
 
19
 
 
20
The usual way to use this is via bzrlib.ui.ui_factory.nested_progress_bar which
 
21
will manage a conceptual stack of nested activities.
27
22
"""
28
23
 
29
 
# TODO: should be a global option e.g. --silent that disables progress
30
 
# indicators, preferably without needing to adjust all code that
31
 
# potentially calls them.
32
 
 
33
 
# TODO: If not on a tty perhaps just print '......' for the benefit of IDEs, etc
34
 
 
35
 
# TODO: Optionally show elapsed time instead/as well as ETA; nicer
36
 
# when the rate is unpredictable
37
24
 
38
25
import sys
39
26
import time
40
27
import os
41
 
 
42
 
from bzrlib.lazy_import import lazy_import
43
 
lazy_import(globals(), """
 
28
import warnings
 
29
 
 
30
 
44
31
from bzrlib import (
45
32
    errors,
 
33
    osutils,
 
34
    trace,
 
35
    ui,
46
36
    )
47
 
""")
48
 
 
49
37
from bzrlib.trace import mutter
 
38
from bzrlib.symbol_versioning import (
 
39
    deprecated_in,
 
40
    deprecated_method,
 
41
    )
50
42
 
51
43
 
52
44
def _supports_progress(f):
 
45
    """Detect if we can use pretty progress bars on the output stream f.
 
46
 
 
47
    If this returns true we expect that a human may be looking at that
 
48
    output, and that we can repaint a line to update it.
 
49
    """
53
50
    isatty = getattr(f, 'isatty', None)
54
51
    if isatty is None:
55
52
        return False
61
58
    return True
62
59
 
63
60
 
64
 
_progress_bar_types = {}
 
61
class ProgressTask(object):
 
62
    """Model component of a progress indicator.
 
63
 
 
64
    Most code that needs to indicate progress should update one of these,
 
65
    and it will in turn update the display, if one is present.
 
66
 
 
67
    Code updating the task may also set fields as hints about how to display
 
68
    it: show_pct, show_spinner, show_eta, show_count, show_bar.  UIs
 
69
    will not necessarily respect all these fields.
 
70
    """
 
71
 
 
72
    def __init__(self, parent_task=None, ui_factory=None):
 
73
        """Construct a new progress task.
 
74
 
 
75
        Normally you should not call this directly but rather through
 
76
        `ui_factory.nested_progress_bar`.
 
77
        """
 
78
        self._parent_task = parent_task
 
79
        self._last_update = 0
 
80
        self.total_cnt = None
 
81
        self.current_cnt = None
 
82
        self.msg = ''
 
83
        self.ui_factory = ui_factory
 
84
        self.show_pct = False
 
85
        self.show_spinner = True
 
86
        self.show_eta = False,
 
87
        self.show_count = True
 
88
        self.show_bar = True
 
89
 
 
90
    def __repr__(self):
 
91
        return '%s(%r/%r, msg=%r)' % (
 
92
            self.__class__.__name__,
 
93
            self.current_cnt,
 
94
            self.total_cnt,
 
95
            self.msg)
 
96
 
 
97
    def update(self, msg, current_cnt=None, total_cnt=None):
 
98
        self.msg = msg
 
99
        self.current_cnt = current_cnt
 
100
        if total_cnt:
 
101
            self.total_cnt = total_cnt
 
102
        self.ui_factory._progress_updated(self)
 
103
 
 
104
    def tick(self):
 
105
        self.update(self.msg)
 
106
 
 
107
    def finished(self):
 
108
        self.ui_factory._progress_finished(self)
 
109
 
 
110
    def make_sub_task(self):
 
111
        return ProgressTask(self, self.ui_factory)
 
112
 
 
113
    def _overall_completion_fraction(self, child_fraction=0.0):
 
114
        """Return fractional completion of this task and its parents
 
115
 
 
116
        Returns None if no completion can be computed."""
 
117
        if self.current_cnt is not None and self.total_cnt:
 
118
            own_fraction = (float(self.current_cnt) + child_fraction) / self.total_cnt
 
119
        else:
 
120
            # if this task has no estimation, it just passes on directly
 
121
            # whatever the child has measured...
 
122
            own_fraction = child_fraction
 
123
        if self._parent_task is None:
 
124
            return own_fraction
 
125
        else:
 
126
            if own_fraction is None:
 
127
                own_fraction = 0.0
 
128
            return self._parent_task._overall_completion_fraction(own_fraction)
 
129
 
 
130
    def note(self, fmt_string, *args):
 
131
        """Record a note without disrupting the progress bar."""
 
132
        # XXX: shouldn't be here; put it in mutter or the ui instead
 
133
        if args:
 
134
            self.ui_factory.note(fmt_string % args)
 
135
        else:
 
136
            self.ui_factory.note(fmt_string)
 
137
 
 
138
    def clear(self):
 
139
        # XXX: shouldn't be here; put it in mutter or the ui instead
 
140
        self.ui_factory.clear_term()
65
141
 
66
142
 
67
143
def ProgressBar(to_file=None, **kwargs):
74
150
        if _supports_progress(to_file):
75
151
            return TTYProgressBar(to_file=to_file, **kwargs)
76
152
        else:
77
 
            return DotsProgressBar(to_file=to_file, **kwargs)
 
153
            return DummyProgress(to_file=to_file, **kwargs)
78
154
    else:
79
155
        # Minor sanitation to prevent spurious errors
80
156
        requested_bar_type = requested_bar_type.lower().strip()
86
162
                                                _progress_bar_types.keys())
87
163
        return _progress_bar_types[requested_bar_type](to_file=to_file, **kwargs)
88
164
 
89
 
 
 
165
 
90
166
class ProgressBarStack(object):
91
 
    """A stack of progress bars."""
92
 
 
 
167
    """A stack of progress bars.
 
168
 
 
169
    This class is deprecated: instead, ask the ui factory for a new progress
 
170
    task and finish it when it's done.
 
171
    """
 
172
 
 
173
    @deprecated_method(deprecated_in((1, 12, 0)))
93
174
    def __init__(self,
94
175
                 to_file=None,
95
176
                 show_pct=False,
146
227
    def return_pb(self, bar):
147
228
        """Return bar after its been used."""
148
229
        if bar is not self._stack[-1]:
149
 
            raise errors.MissingProgressBarFinish()
150
 
        self._stack.pop()
151
 
 
152
 
 
 
230
            warnings.warn("%r is not currently active" % (bar,))
 
231
        else:
 
232
            self._stack.pop()
 
233
 
 
234
 
153
235
class _BaseProgressBar(object):
154
236
 
155
237
    def __init__(self,
188
270
    def finished(self):
189
271
        """Return this bar to its progress stack."""
190
272
        self.clear()
191
 
        assert self._stack is not None
192
273
        self._stack.return_pb(self)
193
274
 
194
275
    def note(self, fmt_string, *args, **kwargs):
206
287
 
207
288
    This can be used as the default argument for methods that
208
289
    take an optional progress indicator."""
 
290
 
209
291
    def tick(self):
210
292
        pass
211
293
 
217
299
 
218
300
    def clear(self):
219
301
        pass
220
 
        
 
302
 
221
303
    def note(self, fmt_string, *args, **kwargs):
222
304
        """See _BaseProgressBar.note()."""
223
305
 
225
307
        return DummyProgress(**kwargs)
226
308
 
227
309
 
228
 
_progress_bar_types['dummy'] = DummyProgress
229
 
_progress_bar_types['none'] = DummyProgress
230
 
 
231
 
 
232
310
class DotsProgressBar(_BaseProgressBar):
233
311
 
234
312
    def __init__(self, **kwargs):
235
313
        _BaseProgressBar.__init__(self, **kwargs)
236
314
        self.last_msg = None
237
315
        self.need_nl = False
238
 
        
 
316
 
239
317
    def tick(self):
240
318
        self.update()
241
 
        
 
319
 
242
320
    def update(self, msg=None, current_cnt=None, total_cnt=None):
243
321
        if msg and msg != self.last_msg:
244
322
            if self.need_nl:
247
325
            self.last_msg = msg
248
326
        self.need_nl = True
249
327
        self.to_file.write('.')
250
 
        
 
328
 
251
329
    def clear(self):
252
330
        if self.need_nl:
253
331
            self.to_file.write('\n')
254
332
        self.need_nl = False
255
 
        
 
333
 
256
334
    def child_update(self, message, current, total):
257
335
        self.tick()
258
336
 
259
337
 
260
 
_progress_bar_types['dots'] = DotsProgressBar
261
 
 
262
 
    
 
338
 
 
339
 
263
340
class TTYProgressBar(_BaseProgressBar):
264
341
    """Progress bar display object.
265
342
 
292
369
        self._max_last_updates = 10
293
370
        self.child_fraction = 0
294
371
        self._have_output = False
295
 
    
296
372
 
297
373
    def throttle(self, old_msg):
298
374
        """Return True if the bar was updated too recently"""
313
389
        self.last_updates = self.last_updates[-self._max_last_updates:]
314
390
        self.last_update = now
315
391
        return False
316
 
        
 
392
 
317
393
    def tick(self):
318
 
        self.update(self.last_msg, self.last_cnt, self.last_total, 
 
394
        self.update(self.last_msg, self.last_cnt, self.last_total,
319
395
                    self.child_fraction)
320
396
 
321
397
    def child_update(self, message, current, total):
325
401
                pass
326
402
            elif self.last_cnt + child_fraction <= self.last_total:
327
403
                self.child_fraction = child_fraction
328
 
            else:
329
 
                mutter('not updating child fraction')
330
404
        if self.last_msg is None:
331
405
            self.last_msg = ''
332
406
        self.tick()
333
407
 
334
 
    def update(self, msg, current_cnt=None, total_cnt=None, 
335
 
               child_fraction=0):
336
 
        """Update and redraw progress bar."""
 
408
    def update(self, msg, current_cnt=None, total_cnt=None,
 
409
            child_fraction=0):
 
410
        """Update and redraw progress bar.
 
411
        """
337
412
        if msg is None:
338
413
            msg = self.last_msg
339
414
 
342
417
 
343
418
        if current_cnt < 0:
344
419
            current_cnt = 0
345
 
            
 
420
 
346
421
        if current_cnt > total_cnt:
347
422
            total_cnt = current_cnt
348
 
        
349
 
        ## # optional corner case optimisation 
 
423
 
 
424
        ## # optional corner case optimisation
350
425
        ## # currently does not seem to fire so costs more than saved.
351
426
        ## # trivial optimal case:
352
427
        ## # NB if callers are doing a clear and restore with
359
434
        ##     self.child_fraction == child_fraction):
360
435
        ##     return
361
436
 
 
437
        if msg is None:
 
438
            msg = ''
 
439
 
362
440
        old_msg = self.last_msg
363
441
        # save these for the tick() function
364
442
        self.last_msg = msg
366
444
        self.last_total = total_cnt
367
445
        self.child_fraction = child_fraction
368
446
 
369
 
        # each function call takes 20ms/4000 = 0.005 ms, 
 
447
        # each function call takes 20ms/4000 = 0.005 ms,
370
448
        # but multiple that by 4000 calls -> starts to cost.
371
449
        # so anything to make this function call faster
372
450
        # will improve base 'diff' time by up to 0.1 seconds.
374
452
            return
375
453
 
376
454
        if self.show_eta and self.start_time and self.last_total:
377
 
            eta = get_eta(self.start_time, self.last_cnt + self.child_fraction, 
 
455
            eta = get_eta(self.start_time, self.last_cnt + self.child_fraction,
378
456
                    self.last_total, last_updates = self.last_updates)
379
457
            eta_str = " " + str_tdelta(eta)
380
458
        else:
381
459
            eta_str = ""
382
460
 
383
461
        if self.show_spinner:
384
 
            spin_str = self.SPIN_CHARS[self.spin_pos % 4] + ' '            
 
462
            spin_str = self.SPIN_CHARS[self.spin_pos % 4] + ' '
385
463
        else:
386
464
            spin_str = ''
387
465
 
404
482
            # make both fields the same size
405
483
            t = '%i' % (self.last_total)
406
484
            c = '%*i' % (len(t), self.last_cnt)
407
 
            count_str = ' ' + c + '/' + t 
 
485
            count_str = ' ' + c + '/' + t
408
486
 
409
487
        if self.show_bar:
410
488
            # progress bar, if present, soaks up all remaining space
413
491
 
414
492
            if self.last_total:
415
493
                # number of markers highlighted in bar
416
 
                markers = int(round(float(cols) * 
 
494
                markers = int(round(float(cols) *
417
495
                              (self.last_cnt + self.child_fraction) / self.last_total))
418
496
                bar_str = '[' + ('=' * markers).ljust(cols) + '] '
419
497
            elif False:
421
499
                # so just show an expanded spinning thingy
422
500
                m = self.spin_pos % cols
423
501
                ms = (' ' * m + '*').ljust(cols)
424
 
                
 
502
 
425
503
                bar_str = '[' + ms + '] '
426
504
            else:
427
505
                bar_str = ''
428
506
        else:
429
507
            bar_str = ''
430
508
 
431
 
        m = spin_str + bar_str + self.last_msg + count_str + pct_str + eta_str
 
509
        m = spin_str + bar_str + self.last_msg + count_str \
 
510
            + pct_str + eta_str
432
511
        self.to_file.write('\r%-*.*s' % (self.width - 1, self.width - 1, m))
433
512
        self._have_output = True
434
513
        #self.to_file.flush()
435
 
            
436
 
    def clear(self):        
 
514
 
 
515
    def clear(self):
437
516
        if self._have_output:
438
517
            self.to_file.write('\r%s\r' % (' ' * (self.width - 1)))
439
518
        self._have_output = False
440
 
        #self.to_file.flush()        
441
 
 
442
 
 
443
 
_progress_bar_types['tty'] = TTYProgressBar
 
519
        #self.to_file.flush()
 
520
 
 
521
 
444
522
 
445
523
 
446
524
class ChildProgress(_BaseProgressBar):
456
534
 
457
535
    def update(self, msg, current_cnt=None, total_cnt=None):
458
536
        self.current = current_cnt
459
 
        self.total = total_cnt
 
537
        if total_cnt is not None:
 
538
            self.total = total_cnt
460
539
        self.message = msg
461
540
        self.child_fraction = 0
462
541
        self.tick()
485
564
    def note(self, *args, **kwargs):
486
565
        self.parent.note(*args, **kwargs)
487
566
 
488
 
 
 
567
 
 
568
class InstrumentedProgress(TTYProgressBar):
 
569
    """TTYProgress variant that tracks outcomes"""
 
570
 
 
571
    def __init__(self, *args, **kwargs):
 
572
        self.always_throttled = True
 
573
        self.never_throttle = False
 
574
        TTYProgressBar.__init__(self, *args, **kwargs)
 
575
 
 
576
    def throttle(self, old_message):
 
577
        if self.never_throttle:
 
578
            result =  False
 
579
        else:
 
580
            result = TTYProgressBar.throttle(self, old_message)
 
581
        if result is False:
 
582
            self.always_throttled = False
 
583
 
 
584
 
489
585
def str_tdelta(delt):
490
586
    if delt is None:
491
587
        return "-:--:--"
512
608
 
513
609
    if elapsed < 2.0:                   # not enough time to estimate
514
610
        return None
515
 
    
 
611
 
516
612
    total_duration = float(elapsed) * float(total) / float(current)
517
613
 
518
 
    assert total_duration >= elapsed
519
 
 
520
614
    if last_updates and len(last_updates) >= n_recent:
521
615
        avg = sum(last_updates) / float(len(last_updates))
522
616
        time_left = avg * (total - current)
543
637
            self.cur_phase = 0
544
638
        else:
545
639
            self.cur_phase += 1
546
 
        assert self.cur_phase < self.total 
547
640
        self.pb.update(self.message, self.cur_phase, self.total)
548
641
 
549
642
 
550
 
def run_tests():
551
 
    import doctest
552
 
    result = doctest.testmod()
553
 
    if result[1] > 0:
554
 
        if result[0] == 0:
555
 
            print "All tests passed"
556
 
    else:
557
 
        print "No tests to run"
558
 
 
559
 
 
560
 
def demo():
561
 
    sleep = time.sleep
562
 
    
563
 
    print 'dumb-terminal test:'
564
 
    pb = DotsProgressBar()
565
 
    for i in range(100):
566
 
        pb.update('Leoparden', i, 99)
567
 
        sleep(0.1)
568
 
    sleep(1.5)
569
 
    pb.clear()
570
 
    sleep(1.5)
571
 
    
572
 
    print 'smart-terminal test:'
573
 
    pb = ProgressBar(show_pct=True, show_bar=True, show_spinner=False)
574
 
    for i in range(100):
575
 
        pb.update('Elephanten', i, 99)
576
 
        sleep(0.1)
577
 
    sleep(2)
578
 
    pb.clear()
579
 
    sleep(1)
580
 
 
581
 
    print 'done!'
582
 
 
583
 
if __name__ == "__main__":
584
 
    demo()
 
643
_progress_bar_types = {}
 
644
_progress_bar_types['dummy'] = DummyProgress
 
645
_progress_bar_types['none'] = DummyProgress
 
646
_progress_bar_types['tty'] = TTYProgressBar
 
647
_progress_bar_types['dots'] = DotsProgressBar