Putting names at the top of files is is not recommended. Collective wisdom for
[opensuse:osc.git] / osc / meter.py
1 #   This library is free software; you can redistribute it and/or
2 #   modify it under the terms of the GNU Lesser General Public
3 #   License as published by the Free Software Foundation; either
4 #   version 2.1 of the License, or (at your option) any later version.
5 #
6 #   This library is distributed in the hope that it will be useful,
7 #   but WITHOUT ANY WARRANTY; without even the implied warranty of
8 #   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
9 #   Lesser General Public License for more details.
10 #
11 #   You should have received a copy of the GNU Lesser General Public
12 #   License along with this library; if not, write to the 
13 #      Free Software Foundation, Inc., 
14 #      59 Temple Place, Suite 330, 
15 #      Boston, MA  02111-1307  USA
16
17 # this is basically a copy of python-urlgrabber's TextMeter class, 
18 # with support added for dynamical sizing according to screen size.
19 # it uses getScreenWidth() scrapped from smart.
20 # 2007-04-24, poeml
21
22
23 from urlgrabber.progress import BaseMeter, format_time, format_number
24 import sys, os
25     
26 def getScreenWidth():
27     import termios, struct, fcntl
28     s = struct.pack('HHHH', 0, 0, 0, 0)
29     try:
30         x = fcntl.ioctl(1, termios.TIOCGWINSZ, s)
31     except IOError:
32         return 80
33     return struct.unpack('HHHH', x)[1]
34
35
36 class TextMeter(BaseMeter):
37     def __init__(self, fo=sys.stderr):
38         BaseMeter.__init__(self)
39         self.fo = fo
40         try:
41             width = int(os.environ['COLUMNS'])
42         except (KeyError, ValueError):
43             width = getScreenWidth()
44
45
46         #self.unsized_templ = '\r%-60.60s    %5sB %s '
47         self.unsized_templ = '\r%%-%s.%ss    %%5sB %%s ' % (width *4/3, width*3/5)
48         #self.sized_templ = '\r%-45.45s %3i%% |%-15.15s| %5sB %8s '
49         self.bar_length = width/5
50         self.sized_templ = '\r%%-%s.%ss %%3i%%%% |%%-%s.%ss| %%5sB %%8s ' %(width*4/10, width*4/10, self.bar_length, self.bar_length)
51
52
53
54     def _do_update(self, amount_read, now=None):
55         etime = self.re.elapsed_time()
56         fetime = format_time(etime)
57         fread = format_number(amount_read)
58         #self.size = None
59         if self.text is not None:
60             text = self.text
61         else:
62             text = self.basename
63         if self.size is None:
64             out = self.unsized_templ % \
65                   (text, fread, fetime)
66         else:
67             rtime = self.re.remaining_time()
68             frtime = format_time(rtime)
69             frac = self.re.fraction_read()
70             bar = '='*int(self.bar_length * frac)
71
72             out = self.sized_templ % \
73                   (text, frac*100, bar, fread, frtime) + 'ETA '
74
75         self.fo.write(out)
76         self.fo.flush()
77
78     def _do_end(self, amount_read, now=None):
79         total_time = format_time(self.re.elapsed_time())
80         total_size = format_number(amount_read)
81         if self.text is not None:
82             text = self.text
83         else:
84             text = self.basename
85         if self.size is None:
86             out = self.unsized_templ % \
87                   (text, total_size, total_time)
88         else:
89             bar = '=' * self.bar_length
90             out = self.sized_templ % \
91                   (text, 100, bar, total_size, total_time) + '    '
92         self.fo.write(out + '\n')
93         self.fo.flush()