blob: 39c3628b18585a6767c0af74910e4d06b2d1861b (
plain)
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
|
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
# Text progress bar library, like curl or scp.
import sys
import datetime
import time
class ProgressBar(object):
def __init__(self, label, limit, label_width=12):
self.label = label
self.limit = limit
self.label_width = label_width
self.cur = 0
self.t0 = datetime.datetime.now()
self.fullwidth = None
self.barlen = 64 - self.label_width
self.fmt = (
"\r%-" + str(label_width) + "s %3d%% %-" + str(self.barlen) + "s| %6.1fs"
)
def update(self, value):
self.cur = value
pct = int(100.0 * self.cur / self.limit)
barlen = int(1.0 * self.barlen * self.cur / self.limit) - 1
bar = "=" * barlen + ">"
dt = datetime.datetime.now() - self.t0
dt = dt.seconds + dt.microseconds * 1e-6
line = self.fmt % (self.label[: self.label_width], pct, bar, dt)
self.fullwidth = len(line)
sys.stdout.write(line)
sys.stdout.flush()
# Clear the current bar and leave the cursor at the start of the line.
def clear(self):
if self.fullwidth:
sys.stdout.write("\r" + " " * self.fullwidth + "\r")
self.fullwidth = None
def finish(self):
self.update(self.limit)
sys.stdout.write("\n")
if __name__ == "__main__":
pb = ProgressBar("test", 12)
for i in range(12):
pb.update(i)
time.sleep(0.5)
pb.finish()
|
