import logging from dataclasses import dataclass from io import BytesIO from typing import Any from webdav3.client import Resource from . import get_ttl_hash, run_in_executor, timed_alru_cache, webdav_resource _logger = logging.getLogger(__name__) @timed_alru_cache(maxsize=20) async def _get_buffer( remote_path: Any, ) -> BytesIO: @run_in_executor def _get_buffer_inner(resource: Resource) -> BytesIO: _logger.info(f"updating {resource}") print(f"updating {resource}") buffer = BytesIO() resource.write_to(buffer) return buffer resource = webdav_resource(remote_path) return await _get_buffer_inner(resource) @dataclass(frozen=True) class DavFile: remote_path: str @property async def __buffer(self) -> BytesIO: return await _get_buffer( ttl_hash=get_ttl_hash(), remote_path=self.remote_path, ) @property async def bytes(self) -> bytes: buffer = await self.__buffer buffer.seek(0) return buffer.read() @property async def string(self) -> str: bytes = await self.bytes return bytes.decode(encoding="utf-8")