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
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
|
# -*- test-case-name: twisted.conch.test.test_text -*-
# Copyright (c) Twisted Matrix Laboratories.
# See LICENSE for details.
"""
Character attribute manipulation API
This module provides a domain-specific language (using Python syntax)
for the creation of text with additional display attributes associated
with it. It is intended as an alternative to manually building up
strings containing ECMA 48 character attribute control codes. It
currently supports foreground and background colors (black, red,
green, yellow, blue, magenta, cyan, and white), intensity selection,
underlining, blinking and reverse video. Character set selection
support is planned.
Character attributes are specified by using two Python operations:
attribute lookup and indexing. For example, the string \"Hello
world\" with red foreground and all other attributes set to their
defaults, assuming the name twisted.conch.insults.text.attributes has
been imported and bound to the name \"A\" (with the statement C{from
twisted.conch.insults.text import attributes as A}, for example) one
uses this expression::
| A.fg.red[\"Hello world\"]
Other foreground colors are set by substituting their name for
\"red\". To set both a foreground and a background color, this
expression is used::
| A.fg.red[A.bg.green[\"Hello world\"]]
Note that either A.bg.green can be nested within A.fg.red or vice
versa. Also note that multiple items can be nested within a single
index operation by separating them with commas::
| A.bg.green[A.fg.red[\"Hello\"], " ", A.fg.blue[\"world\"]]
Other character attributes are set in a similar fashion. To specify a
blinking version of the previous expression::
| A.blink[A.bg.green[A.fg.red[\"Hello\"], " ", A.fg.blue[\"world\"]]]
C{A.reverseVideo}, C{A.underline}, and C{A.bold} are also valid.
A third operation is actually supported: unary negation. This turns
off an attribute when an enclosing expression would otherwise have
caused it to be on. For example::
| A.underline[A.fg.red[\"Hello\", -A.underline[\" world\"]]]
@author: Jp Calderone
"""
from twisted.conch.insults import helper, insults
class _Attribute(object):
def __init__(self):
self.children = []
def __getitem__(self, item):
assert isinstance(item, (list, tuple, _Attribute, str))
if isinstance(item, (list, tuple)):
self.children.extend(item)
else:
self.children.append(item)
return self
def serialize(self, write, attrs=None):
if attrs is None:
attrs = helper.CharacterAttribute()
for ch in self.children:
if isinstance(ch, _Attribute):
ch.serialize(write, attrs.copy())
else:
write(attrs.toVT102())
write(ch)
class _NormalAttr(_Attribute):
def serialize(self, write, attrs):
attrs.__init__()
super(_NormalAttr, self).serialize(write, attrs)
class _OtherAttr(_Attribute):
def __init__(self, attrname, attrvalue):
self.attrname = attrname
self.attrvalue = attrvalue
self.children = []
def __neg__(self):
result = _OtherAttr(self.attrname, not self.attrvalue)
result.children.extend(self.children)
return result
def serialize(self, write, attrs):
attrs = attrs.wantOne(**{self.attrname: self.attrvalue})
super(_OtherAttr, self).serialize(write, attrs)
class _ColorAttr(_Attribute):
def __init__(self, color, ground):
self.color = color
self.ground = ground
self.children = []
def serialize(self, write, attrs):
attrs = attrs.wantOne(**{self.ground: self.color})
super(_ColorAttr, self).serialize(write, attrs)
class _ForegroundColorAttr(_ColorAttr):
def __init__(self, color):
super(_ForegroundColorAttr, self).__init__(color, 'foreground')
class _BackgroundColorAttr(_ColorAttr):
def __init__(self, color):
super(_BackgroundColorAttr, self).__init__(color, 'background')
class CharacterAttributes(object):
class _ColorAttribute(object):
def __init__(self, ground):
self.ground = ground
attrs = {
'black': helper.BLACK,
'red': helper.RED,
'green': helper.GREEN,
'yellow': helper.YELLOW,
'blue': helper.BLUE,
'magenta': helper.MAGENTA,
'cyan': helper.CYAN,
'white': helper.WHITE}
def __getattr__(self, name):
try:
return self.ground(self.attrs[name])
except KeyError:
raise AttributeError(name)
fg = _ColorAttribute(_ForegroundColorAttr)
bg = _ColorAttribute(_BackgroundColorAttr)
attrs = {
'bold': insults.BOLD,
'blink': insults.BLINK,
'underline': insults.UNDERLINE,
'reverseVideo': insults.REVERSE_VIDEO}
def __getattr__(self, name):
if name == 'normal':
return _NormalAttr()
if name in self.attrs:
return _OtherAttr(name, True)
raise AttributeError(name)
def flatten(output, attrs):
"""Serialize a sequence of characters with attribute information
The resulting string can be interpreted by VT102-compatible
terminals so that the contained characters are displayed and, for
those attributes which the terminal supports, have the attributes
specified in the input.
For example, if your terminal is VT102 compatible, you might run
this for a colorful variation on the \"hello world\" theme::
| from twisted.conch.insults.text import flatten, attributes as A
| from twisted.conch.insults.helper import CharacterAttribute
| print flatten(
| A.normal[A.bold[A.fg.red['He'], A.fg.green['ll'], A.fg.magenta['o'], ' ',
| A.fg.yellow['Wo'], A.fg.blue['rl'], A.fg.cyan['d!']]],
| CharacterAttribute())
@param output: Object returned by accessing attributes of the
module-level attributes object.
@param attrs: A L{twisted.conch.insults.helper.CharacterAttribute}
instance
@return: A VT102-friendly string
"""
L = []
output.serialize(L.append, attrs)
return ''.join(L)
attributes = CharacterAttributes()
__all__ = ['attributes', 'flatten']
|
