75 lines
1.9 KiB
Python
75 lines
1.9 KiB
Python
import functools
|
|
from types import coroutine
|
|
|
|
|
|
def delegate_to_executor(*attrs):
|
|
def cls_builder(cls):
|
|
for attr_name in attrs:
|
|
setattr(cls, attr_name, _make_delegate_method(attr_name))
|
|
return cls
|
|
|
|
return cls_builder
|
|
|
|
|
|
def proxy_method_directly(*attrs):
|
|
def cls_builder(cls):
|
|
for attr_name in attrs:
|
|
setattr(cls, attr_name, _make_proxy_method(attr_name))
|
|
return cls
|
|
|
|
return cls_builder
|
|
|
|
|
|
def proxy_property_directly(*attrs):
|
|
def cls_builder(cls):
|
|
for attr_name in attrs:
|
|
setattr(cls, attr_name, _make_proxy_property(attr_name))
|
|
return cls
|
|
|
|
return cls_builder
|
|
|
|
|
|
def cond_delegate_to_executor(*attrs):
|
|
def cls_builder(cls):
|
|
for attr_name in attrs:
|
|
setattr(cls, attr_name, _make_cond_delegate_method(attr_name))
|
|
return cls
|
|
|
|
return cls_builder
|
|
|
|
|
|
def _make_delegate_method(attr_name):
|
|
@coroutine
|
|
def method(self, *args, **kwargs):
|
|
cb = functools.partial(getattr(self._file, attr_name), *args, **kwargs)
|
|
return (yield from self._loop.run_in_executor(self._executor, cb))
|
|
|
|
return method
|
|
|
|
|
|
def _make_proxy_method(attr_name):
|
|
def method(self, *args, **kwargs):
|
|
return getattr(self._file, attr_name)(*args, **kwargs)
|
|
|
|
return method
|
|
|
|
|
|
def _make_proxy_property(attr_name):
|
|
def proxy_property(self):
|
|
return getattr(self._file, attr_name)
|
|
|
|
return property(proxy_property)
|
|
|
|
|
|
def _make_cond_delegate_method(attr_name):
|
|
"""For spooled temp files, delegate only if rolled to file object"""
|
|
|
|
async def method(self, *args, **kwargs):
|
|
if self._file._rolled:
|
|
cb = functools.partial(getattr(self._file, attr_name), *args, **kwargs)
|
|
return await self._loop.run_in_executor(self._executor, cb)
|
|
else:
|
|
return getattr(self._file, attr_name)(*args, **kwargs)
|
|
|
|
return method
|