← Back to branch summary

~abentley/bzrtools/bzrtools.dev

« back to all changes in this revision

Viewing changes to progress.py

  • Committer: abentley
  • Date: 2005-04-29 14:41:03 UTC
  • Revision ID: abentley@lappy-20050429144103-b0a1a78752650702
refactored out progress.py

Show diffs side-by-side

added added

removed removed

Lines of Context:
progress.py
1
 
# Copyright (C) 2005 Aaron Bentley
2
 
# <aaron.bentley@utoronto.ca>
3
 
#
4
 
#    This program is free software; you can redistribute it and/or modify
5
 
#    it under the terms of the GNU General Public License as published by
6
 
#    the Free Software Foundation; either version 2 of the License, or
7
 
#    (at your option) any later version.
8
 
#
9
 
#    This program is distributed in the hope that it will be useful,
10
 
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
11
 
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12
 
#    GNU General Public License for more details.
13
 
#
14
 
#    You should have received a copy of the GNU General Public License
15
 
#    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
1
import sys
19
 
import datetime
20
 
from bzrlib.progress import ProgressBar
21
 
 
22
2
class Progress(object):
23
3
    def __init__(self, units, current, total=None):
24
4
        self.units = units
25
5
        self.current = current
26
6
        self.total = total
27
 
 
28
 
    def _get_percent(self):
29
 
        if self.total is None:
30
 
            return None
31
 
        return 100.0 * self.current / self.total
32
 
 
33
 
    percent = property(_get_percent)
 
7
        self.percent = None
 
8
        if self.total is not None:
 
9
            self.percent = 100.0 * current / total
34
10
 
35
11
    def __str__(self):
36
12
        if self.total is not None:
39
15
        else:
40
16
            return "%i %s" (self.current, self.units) 
41
17
 
42
 
def show_progress(pi, prog):
43
 
    pi.update(prog.units, prog.current, prog.total)
44
 
        
 
18
 
 
19
def progress_bar(progress):
 
20
    fmt = " %i of %i %s (%.1f%%)"
 
21
    f = fmt % (progress.total, progress.total, progress.units, 100.0)
 
22
    max = len(f)
 
23
    cols = 77 - max
 
24
    markers = int (float(cols) * progress.current / progress.total)
 
25
    txt = fmt % (progress.current, progress.total, progress.units,
 
26
                 progress.percent)
 
27
    sys.stdout.write("\r[%s%s]%s" % ('='*markers, ' '*(cols-markers), txt))
 
28
 
45
29
def clear_progress_bar():
46
 
    sys.stderr.write('\r%s\r' % (' '*79))
 
30
    sys.stdout.write('\r%s\r' % (' '*79))
 
31