| 1234567891011121314151617181920212223242526 |
- from BTL.defer import Deferred, defer_to_thread
- class ThreadProxy(object):
- __slots__ = ('obj', 'local_queue_task', 'thread_queue_task')
- def __init__(self, obj, local_queue_task, thread_queue_task):
- self.obj = obj
- self.local_queue_task = local_queue_task
- self.thread_queue_task = thread_queue_task
- def __gen_call_wrapper__(self, f):
- def call_wrapper(*a, **kw):
- return defer_to_thread(self.local_queue_task, self.thread_queue_task,
- f, *a, **kw)
- return call_wrapper
- def __getattr__(self, attr):
- a = getattr(self.obj, attr)
- if callable(a):
- return self.__gen_call_wrapper__(a)
- return a
- def call_with_obj(self, _f, *a, **k):
- w = self.__gen_call_wrapper__(_f)
- return w(self.obj, *a, **k)
|
