Python-AES/ProgressBar.py at master · pcaro90/Python-AES · GitHub

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
#!/usr/bin/env python
# -*- coding: utf-8 -*-

# ---------------------------------------------------
# Copyright (c) 2013 Pablo Caro. All Rights Reserved.
# Pablo Caro <me@pcaro.es> - http://pcaro.es/
# ProgressBar.py
# ---------------------------------------------------

import sys


class ProgressBar:
    def __init__(self, min=0, max=100, width=60, charset='[=]'):
        self.min = min
        self.max = max
        self.width = width
        self.current = min
        self.percent = 0.0
        self.int_percent = 0

        if len(charset) != 3:
            charset = '[=]'
        self.charset = charset

        self.bar = ''
        self.used = -1
        self.int_percent_change = False

    def update(self, current):
        self.current = current
        self.percent = (float(self.current-self.min)/(self.max-self.min))*100.0
        int_percent = int(self.percent)

        if int_percent != self.int_percent:
            self.int_percent_change = True
        self.int_percent = int_percent

        self.__generate_bar__()

        if self.int_percent_change:
            self.int_percent_change = False
            return True
        else:
            return False

    def show(self):
        sys.stdout.write(str(self))
        sys.stdout.flush()

    def __str__(self):
        return self.bar

    def __generate_bar__(self):
        self.used = int((float(self.current-self.min)/(self.max-self.min)) *
                        (self.width-6))

        center = self.charset[1] * self.used
        self.bar = (self.charset[0] + center + self.charset[2]
                    + " " + str(self.int_percent) + '%' + '\r')


def main():
    pass

if __name__ == '__main__':
    main()