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
|
"""
@package core.gthread
@brief Threading
Classes:
- gthread::gThread
(C) 2013-2014 by the GRASS Development Team
This program is free software under the GNU General Public License
(>=v2). Read the file COPYING that comes with GRASS for details.
@author Stepan Turek <stepan.turek seznam.cz> (mentor: Martin Landa)
"""
import threading
import time
import wx
from wx.lib.newevent import NewEvent
import sys
import queue as Queue
from core.gconsole import EVT_CMD_DONE, wxCmdDone
wxThdTerminate, EVT_THD_TERMINATE = NewEvent()
class gThread(threading.Thread, wx.EvtHandler):
"""Thread for various backends
terminating thread:
https://www.geeksforgeeks.org/python-different-ways-to-kill-a-thread/
"""
requestId = 0
def __init__(self, requestQ=None, resultQ=None, **kwds):
wx.EvtHandler.__init__(self)
self.terminate = False
self._terminate_evt = None
threading.Thread.__init__(self, **kwds)
if requestQ is None:
self.requestQ = Queue.Queue()
else:
self.requestQ = requestQ
if resultQ is None:
self.resultQ = Queue.Queue()
else:
self.resultQ = resultQ
self.daemon = True
self.Bind(EVT_CMD_DONE, self.OnDone)
self.Bind(EVT_THD_TERMINATE, self.OnTerminate)
self.start()
def Run(self, *args, **kwds):
"""Run command in queue
:param args: unnamed command arguments
:param kwds: named command arguments, keyword 'callable'
represents function to be run, keyword 'ondone'
represents function to be called after the
callable is done
:return: request id in queue
"""
gThread.requestId += 1
self.requestQ.put((gThread.requestId, args, kwds))
return gThread.requestId
def GetId(self):
"""Get id for next command"""
return gThread.requestId + 1
def SetId(self, id):
"""Set starting id"""
gThread.requestId = id
def run(self):
variables = {
"callable": None,
"ondone": None,
"userdata": None,
"onterminate": None,
}
while True:
requestId, args, kwds = self.requestQ.get()
for key in ("callable", "ondone", "userdata", "onterminate"):
if key in kwds:
variables[key] = kwds[key]
del kwds[key]
requestTime = time.time()
ret = None
exception = None
time.sleep(0.01)
self._terminate_evt = wxThdTerminate(
onterminate=variables["onterminate"],
kwds=kwds,
args=args,
pid=requestId,
)
if self.terminate:
return
ret = variables["callable"](*args, **kwds)
if self.terminate:
return
# except Exception as e:
# exception = e;
self.resultQ.put((requestId, ret))
event = wxCmdDone(
ondone=variables["ondone"],
kwds=kwds,
args=args, # TODO expand args to kwds
ret=ret,
exception=exception,
userdata=variables["userdata"],
pid=requestId,
)
# send event
wx.PostEvent(self, event)
def OnDone(self, event):
if event.ondone:
event.ondone(event)
def Terminate(self, terminate=True):
"""Abort command(s)"""
self.terminate = terminate
def start(self):
self.__run_backup = self.run
self.run = self.__run
threading.Thread.start(self)
def __run(self):
sys.settrace(self.globaltrace)
self.__run_backup()
self.run = self.__run_backup
def globaltrace(self, frame, event, arg):
if event == "call":
return self.localtrace
else:
return None
def localtrace(self, frame, event, arg):
if self.terminate:
if event == "line":
# Send event
wx.PostEvent(self, self._terminate_evt)
raise SystemExit()
return self.localtrace
def OnTerminate(self, event):
if event.onterminate:
event.onterminate(event)
|
