| 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 | |