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
|
#!/usr/bin/python
import urwid
import sys
import StringIO
import traceback
ps1 = ">>> "
ps2 = "... "
class AssertAlways(object):
def __getattr__(self, name):
assert 0, name # FIXME
class ShellEnvironment(object):
def __init__(self):
self.globals = {}
def run(self, statement):
"""
Run the python commands in statement and return:
(output text, error information)
"""
err = None
try:
code = compile(statement, '<interactive>', 'single')
except:
err = sys.exc_info()
if err:
return None, err
#save_stdin = sys.stdin
#sys.stdin = AssertAlways()
save_stdout = sys.stdout
sys.stdout = StringIO.StringIO()
output = None
try:
try:
eval(code, self.globals)
except:
err = sys.exc_info()
output = sys.stdout.getvalue()
finally:
sys.stdout = save_stdout
#sys.stdin = save_stdin
return output, err
class ShellWindow(urwid.WidgetWrap):
def __init__(self):
self.lines = []
self.lines.append(urwid.Text("Python "+sys.version))
self.lb = urwid.ListBox(self.lines)
self.new_edit()
self.shenv = ShellEnvironment()
self.__super.__init__(self.lb)
def new_edit(self):
self.edit = ShellEdit(self)
self.lines.append(self.edit)
self.lb.set_focus(len(self.lines)-1)
def execute_edit(self):
et = self.edit.get_edit_text()
self.lines[-1] = urwid.Text([('prompt', ps1),et])
result, err = self.shenv.run(et)
if result:
if result[-1:] == '\n': result = result[:-1]
self.lines.append(urwid.Text(result))
if err:
trace = traceback.format_exception_only(*err[:2])
trace = "".join(trace)
if trace[-1] == '\n': trace = trace[:-1]
self.lines.append(urwid.Text(('error', trace)))
self.new_edit()
class ShellEdit(urwid.Edit):
def __init__(self, parent):
self.parent = parent
self.__super.__init__(('prompt', ps1))
def keypress(self, size, k):
k = self.__super.keypress(size, k)
if k == 'enter':
self.parent.execute_edit()
def main():
view = urwid.AttrWrap(ShellWindow(), 'body')
palette = [
('body', 'light gray', 'black'),
('prompt', 'yellow', 'black'),
('error', 'light red', 'black')
]
urwid.MainLoop(view).run()
if __name__ == "__main__":
main()
|
