| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172 |
- import functools
- 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):
- async def method(self, *args, **kwargs):
- cb = functools.partial(getattr(self._file, attr_name), *args, **kwargs)
- return await 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
|
