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
|
#!/usr/bin/env python3
import urwid
from .sparkwidgets import *
class ProgressBar(urwid.WidgetWrap):
def __init__(self, width, maximum, value=0,
progress_color=None, remaining_color=None):
self.width = width
self.maximum = maximum
self.value = value
self.progress_color = progress_color or DEFAULT_BAR_COLOR
self.remaining_color = remaining_color or DEFAULT_LABEL_COLOR
self.placeholder = urwid.WidgetPlaceholder(urwid.Text(""))
self.update()
super().__init__(self.placeholder)
def pack(self, size, focus=False):
return (self.width, 1)
@property
def value_label(self):
label_text = str(self.value)
bar_len = self.spark_bar.bar_width(0)
attr1 = f"{DEFAULT_LABEL_COLOR}:{self.progress_color}"
content = [(attr1, label_text[:bar_len])]
if len(label_text) > bar_len-1:
attr2 = f"{DEFAULT_LABEL_COLOR}:{self.remaining_color}"
content.append((attr2, label_text[bar_len:]))
return urwid.Text(content)
@property
def maximum_label(self):
label_text = str(self.maximum)
bar_len = self.spark_bar.bar_width(1)
attr1 = f"{DEFAULT_LABEL_COLOR}:{self.remaining_color}"
content = []
if bar_len:
content.append((attr1, label_text[-bar_len:]))
if len(label_text) > bar_len:
attr2 = f"{DEFAULT_LABEL_COLOR}:{self.progress_color}"
content.insert(0, (attr2, label_text[:-bar_len or None]))
return urwid.Text(content)
def update(self):
value_label = None
maximum_label = None
self.spark_bar = SparkBarWidget(
[
SparkBarItem(self.value, bcolor=self.progress_color),
SparkBarItem(self.maximum-self.value, bcolor=self.remaining_color),
], width=self.width
)
overlay1 = urwid.Overlay(
urwid.Filler(self.value_label),
urwid.Filler(self.spark_bar),
"left",
len(self.value_label.get_text()[0]),
"top",
1
)
label_len = len(self.maximum_label.get_text()[0])
overlay2 = urwid.Overlay(
urwid.Filler(self.maximum_label),
overlay1,
"left",
label_len,
"top",
1,
left=self.width - label_len
)
self.placeholder.original_widget = urwid.BoxAdapter(overlay2, 1)
def set_value(self, value):
self.value = value
self.update()
@property
def items(self):
return self.spark_bar.items
__all__ = ["ProgressBar"]
|
