1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
|
# -*- python -*-
#-----------------------------------------------------------------------
# XALT: A tool that tracks users jobs and environments on a cluster.
# Copyright (C) 2013-2014 University of Texas at Austin
# Copyright (C) 2013-2014 University of Tennessee
#
# This library is free software; you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as
# published by the Free Software Foundation; either version 2.1 of
# the License, or (at your option) any later version.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library; if not, write to the Free
# Software Foundation, Inc., 59 Temple Place, Suite 330,
# Boston, MA 02111-1307 USA
#-----------------------------------------------------------------------
from __future__ import print_function, division
import os, sys, re
def getTerminalSize():
"""
Finds the terminal size if possible otherwise it assumes 25, 80
@returns: lines the number of rows
@returns: cols the number of columns
"""
import os, struct
def ioctl_GWINSZ(fd):
import fcntl, termios
return struct.unpack("hh", fcntl.ioctl(fd, termios.TIOCGWINSZ, "1234"))
# try stdin, stdout, stderr
if (not sys.stdout.isatty()):
return (25, 80)
fd = 1
try:
return ioctl_GWINSZ(fd)
except:
pass
# try os.ctermid()
try:
fd = os.open(os.ctermid(), os.O_RDONLY)
try:
return ioctl_GWINSZ(fd)
finally:
os.close(fd)
except:
pass
# try `stty size`
try:
return tuple(int(x) for x in os.popen("stty size", "r").read().split())
except:
pass
# try environment variables
try:
return tuple(int(os.getenv(var)) for var in ("LINES", "COLUMNS"))
except:
pass
# i give up. return default.
return (25, 80)
class ProgressBar(object):
""" A progress bar display class """
def __init__(self, termWidth=None, barWidth=None, maxVal=None, ttyOnly=False, fd=sys.stderr):
"""
Ctor that figures out the range of the progress bar.
@param termWidth: The terminal width
@param barWidth: The number of character that make up the bar
@param maxVal: The maximum value (required)
@param ttyOnly: If true then only print progress bar element if connected to a terminal.
@param fd: The output stream.
"""
if (not maxVal):
ValueError('Must specify maxVal')
self.__maxVal = maxVal
if (not barWidth):
barWidth = termWidth
if (not barWidth):
r, barWidth = getTerminalSize()
if (barWidth < 40):
barWidth = 40
barWidth = min(barWidth, self.__maxVal)
self.__active = True
if (ttyOnly and not sys.stdout.isatty()):
self.__active = False
self.__fd = fd
self.__barWidth = barWidth
self.__unit = 100/self.__barWidth
self.__fence = self.__unit
self.__mark = 10
self.__count = -1
self.__symbolT = [ '+', '+', '+', '+', '|', '+', '+', '+', '+', '|' ]
def update(self, i):
"""
Update progress bar.
@param i: input value
"""
if (not self.__active):
return
j = 100*i//self.__maxVal
k = 100*(i+1)//self.__maxVal
#print("i: ", i, "j: ",j,"k:",k, "fence: ", self.__fence,"mark:",self.__mark)
if (j >= self.__fence):
symbol = "-"
if (( j <= self.__mark and k >= self.__mark) or
( j == k and j == self.__mark) or
(self.__fence > self.__mark)):
self.__count += 1
self.__mark += 10
symbol = self.__symbolT[self.__count]
self.__fd.write(symbol)
self.__fence += self.__unit
def fini(self):
""" Finish progress bar output. """
if (not self.__active):
return
self.__fd.write("\n")
def main():
"""
Test program for progress bar.
"""
num = 200
pbar = ProgressBar(maxVal=num, ttyOnly=False)
for i in xrange(num):
pbar.update(i+1)
pbar.fini()
if ( __name__ == '__main__'): main()
|
