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
|
"""Panel to execute scripts and redirect their output for SPE"""
import os, re
import wx
import wx.stc as wx_stc
import wx.html as html
import _spe.info as info
def icon(x):
return os.path.join(info.INFO['skinLocation'],x)
FIND_ICON = icon('lookup.png').replace('\\','\\\\')
RUN_ICON = icon('run.png')
RE_LINK = re.compile('(\s*)(File "(.*))\n')
RE_LOCATION = re.compile('File "([^"]*)", line (\d+)')
KILL_ERROR = {
wx.KILL_OK : 'ok',
wx.KILL_BAD_SIGNAL : 'bad signal',
wx.KILL_ACCESS_DENIED : 'access denied',
wx.KILL_NO_PROCESS : 'no such process',
wx.KILL_ERROR : 'unspecified error',
}
class Output(html.HtmlWindow):
pid = -1
def __init__(self,parent=None,*args,**keyw):
html.HtmlWindow.__init__(self,parent,*args,**keyw)
self.SetFonts(normal_face='helvetica',fixed_face='courier',sizes=[8,9,10,12,16,20,22])
self.app = parent.app
#---execute
def _check_run(self,bool):
#assing method for check run tool button
child = self.app.childActive
if child.frame.menuBar:
child.frame.menuBar.check_run(bool)
else:
child.parentFrame.menuBar.check_run(bool)
def Execute(self, command, label = None, statustext = "Running script...",beep=False):
"""Executes a command of which the output will be redirected by OnIdle and OnEndProcess."""
if self.pid is -1:
if not label: label = command
#give feedback
self.AddText('<table bgcolor=#CCCCCC width=100%%><tr><td><TT><img src="%s"> %s</TT></td></tr></table><br>'%(RUN_ICON,label))
self.SetStatusText(statustext)
self.UpdateToolbar()
self.Raise()
#bind events
self.Bind(wx.EVT_IDLE,self.OnIdle)
self.Bind(wx.EVT_END_PROCESS,self.OnEndProcess)
#create process
self.process = wx.Process(self)
self.process.Redirect()
if info.WIN:
self.pid = wx.Execute(command, wx.EXEC_ASYNC | wx.EXEC_NOHIDE, self.process)
else:
self.pid = wx.Execute(command, wx.EXEC_ASYNC | wx.EXEC_MAKE_GROUP_LEADER, self.process)
self.inputstream = self.process.GetInputStream()
self.errorstream = self.process.GetErrorStream()
self.outputstream = self.process.GetOutputStream()
self.inputstream.Write = Write
self._check_run(True)
self.beep = beep
def Kill(self):
if self.pid != -1 and self.process and hasattr(self.process,'Kill'):
#result = self.process.Kill(self.pid,wx.SIGKILL,wx.KILL_CHILDREN)
#result = self.process.Kill(self.pid,wx.SIGTERM,wx.KILL_CHILDREN)
result = self.process.Kill(self.pid)#,wx.SIGNONE)#,wx.KILL_CHILDREN)
self.OnEndProcess(event=None)
message = 'Script stopped by user (%s).'%KILL_ERROR.get(result,'unknown error')
self.SetStatusText(message)
self.AddText(message,error=True)
def IsBusy(self):
"""Is the instance busy executing a command."""
return not (self.pid is -1)
#---user feedback
def AddText(self,text,error=False):
"""Add text and in case of error, colour red and provide links."""
text = text.replace('\r\n','\n').replace("<string>","<string>")
if error:
text = RE_LINK.sub(r"\g<1><img src='%s'> <a href='\g<2>'>\g<2></a><br>"%FIND_ICON,text)
text = '<font color=red>%s</font>'%text
text = text.replace('\n','<br>')
self.AppendToPage(text)
self.Scroll(0,self.GetVirtualSize()[1]/self.GetScrollPixelsPerUnit()[1])
def SetStatusText(self,text):
print text
def UpdateToolbar(self):
#print 'Updating toolbar...'
pass
#---event handlers
def OnIdle(self, event):
if self.inputstream.CanRead():
text = self.inputstream.read()
self.AddText(text)
if self.errorstream.CanRead():
text = self.errorstream.read()
self.AddText(text,error=True)
def OnEndProcess(self, event):
#unbind events
self.Unbind(wx.EVT_IDLE)
self.Unbind(wx.EVT_END_PROCESS)
#check for any leftover output.
self.OnIdle(event)
#destroy process
self.process.Destroy()
self.process = None
self.pid = -1
#give feedback
self.UpdateToolbar()
self._check_run(False)
if event:
message = "Script terminated."
self.SetStatusText(message)
self.AddText(message)
wx.Bell()
def OnLinkClicked(self,linkInfo):
match = RE_LOCATION.match(linkInfo.GetHref())
try:
fileName = match.group(1)
lineno = int(match.group(2))
self.OpenFile(fileName,lineno-1)
self.SetStatusText('Jumped to file "%s" (line %s).'%(fileName,lineno))
except Exception, message:
self.SetStatusText('SPE could not locate source file. (%s)'%message)
def OpenFile(self,fileName,lineno):
print fileName,lineno
def Write(self,*args,**keyw):
print args,keyw
wx.OutputStream.Write(self,*args,**keyw)
class Panel(Output):
def __init__(self,panel,*args,**keyw):
Output.__init__(self,parent=panel,id=-1,*args,**keyw)
self.app = panel.app
self.OpenFile = panel.openList
self.SetStatusText = panel.SetActiveStatusText
def Raise(self):
self.GetParent().SetSelection(3)#todo: 3 should be determined dynamic
class TestFrame(wx.Frame):
def __init__(self,parent=None,id=-1,*args,**keyw):
wx.Frame.__init__(self,parent=parent,id=id,*args,**keyw)
self.output = Output(parent=self,id=-1)
self.Show()
def test(self):
import time
for i in range (1):
self.output.Execute('python -c "%s/0"'%i)
## while self.output.IsBusy():
## print self.output.pid
## time.sleep(1)
if __name__ == '__main__':
app = wx.PySimpleApp()
frame = TestFrame()
app.SetTopWindow(frame)
wx.CallAfter(frame.test)
app.MainLoop()
|
