Commit | Line | Data |
---|---|---|
8ee66edd FC |
1 | from threading import Lock |
2 | ||
3 | ||
4 | class CallbackMux: | |
5 | # this is a sort of "multiplexer" i.e. it manages callbacks from threads | |
6 | # and redirect them to the main thread (this prevents deadlocks) | |
7 | ||
8 | def __init__(self): | |
9 | self.lock = Lock() | |
10 | self.callbacks = [] | |
11 | taskMgr.add(self.process_callbacks, 'processing callbacks') | |
12 | ||
13 | def add_cb(self, func, args=None): | |
14 | args = args or [] | |
15 | with self.lock: self.callbacks += [(func, args)] | |
16 | ||
17 | def process_callbacks(self, task): | |
18 | with self.lock: | |
19 | callbacks = self.callbacks[:] | |
20 | self.callbacks = [] | |
21 | for func, args in callbacks: func(*args) | |
22 | return task.cont |