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
|
# -*- coding: utf-8 -*-
import urwid
from urwid_utils.colors import BASIC_COLORS, STYLES
from urwid_utils.util import is_valid_identifier
from urwid.display_common import _parse_color_256
try:
from urwid.display_common import _parse_color_true
URWID_HAS_TRUE_COLOR=True
except ImportError:
URWID_HAS_TRUE_COLOR=False
COLORS_ALLOWED_MAP = {
16: lambda val: val in BASIC_COLORS,
256: lambda val: _parse_color_256(val) is not None,
1<<24: lambda val: URWID_HAS_TRUE_COLOR and _parse_color_true(val) is not None
}
class PaletteEntry(list):
attrs = [
'name',
'foreground',
'background',
'mono',
'foreground_high',
'background_high',
]
def __init__(self, *args, **kwargs):
list.__init__(self, [None]*len(self.attrs))
for index, value in enumerate(args):
key = self.attrs[index]
kwargs[key] = value
for name, value in kwargs.items():
self.__setattr__(name=name, value=value)
def __repr__(self):
rep = []
class_name = self.__class__.__name__
attrs = []
for index, attr_name in enumerate(self.attrs):
value = self[index]
attrs.append('{0}={1}'.format(attr_name, repr(value)))
rep.append('<')
rep.append(class_name)
rep.append('(')
rep.append(', '.join(attrs))
rep.append(')>')
return ''.join(rep)
def _key(self):
return tuple(self[:len(self.attrs)])
def __hash__(self):
return hash(self._key())
def allowed(self, value, colors=None):
color_allowed = lambda val: any([
COLORS_ALLOWED_MAP[k](val)
for k in COLORS_ALLOWED_MAP.keys()
if not colors or k <= colors
])
return any([(
val is None
or val in [v for n,v in STYLES]
or color_allowed(val)
) for val in value.split(',')])
def __setattr__(self, name, value):
if name != 'name' and not self.allowed(value):
raise ValueError('"{0}": value not allowed'.format(value))
try:
index = self.attrs.index(name)
list.__setitem__(self, index, value)
return
except ValueError:
pass
list.__setattr__(self, name, value)
def __getattr__(self, name):
try:
index = self.attrs.index(name)
return self[index]
except ValueError:
pass
raise AttributeError('"{0}": unknown attribute'.format(name))
class Palette(list):
def __init__(self, name=None, **entries):
self.name = name
for name, entry in entries.items():
entry.name = name
list.__init__(self, list(entries.values()))
def __setattr__(self, name, value):
if isinstance(value, list):
if not is_valid_identifier(name):
raise AttributeError('"{0}" is not a legal python identifier.'.format(name))
for index, entry in enumerate(self):
if entry[0] == name:
self[index] = value
break
else:
value.name = name # Only here do we need to set the PaletteEntry()'s name
self.append(value)
else:
list.__setattr__(self, name, value)
def __getattr__(self, name):
for entry in self:
if entry[0] == name:
return entry
raise AttributeError('"{0}": unknown attribute'.format(name))
def __repr__(self):
rep = []
class_name = self.__class__.__name__
rep.append('<')
rep.append(class_name)
rep.append('(\n')
rep.append(' name={0},\n'.format(repr(self.name)))
rep.extend([' {0},\n'.format(repr(e)) for e in self])
rep.append(')>')
return ''.join(rep)
|
