ovdashboard/api/ovdashboard_api/dav_file.py

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