mirror of
https://github.com/arabianq/yndx-disk.git
synced 2026-04-27 22:21:23 +00:00
59 lines
2.6 KiB
Python
59 lines
2.6 KiB
Python
import asyncio
|
|
from aiohttp import ClientSession
|
|
|
|
from yndx_disk.classes import File, Directory
|
|
from yndx_disk.clients.async_client import AsyncDiskClient
|
|
|
|
|
|
class DiskClient(AsyncDiskClient):
|
|
def __init__(self, token: str, auto_update_info: bool = True, session: ClientSession = None):
|
|
if not session:
|
|
self._loop = asyncio.new_event_loop()
|
|
asyncio.set_event_loop(self._loop)
|
|
|
|
session = ClientSession(loop=self._loop)
|
|
|
|
super().__init__(token, auto_update_info, session=session)
|
|
|
|
def _cleanup(self) -> None:
|
|
self._loop.run_until_complete(self.session.close())
|
|
|
|
def update_disk_info(self) -> None:
|
|
return self._loop.run_until_complete(super().update_disk_info())
|
|
|
|
def get_object(self, path: str) -> File | Directory:
|
|
return self._loop.run_until_complete(super().get_object(path))
|
|
|
|
def listdir(self, path: str = "/", limit: int = 100, offset: int = 0) -> list[File | Directory]:
|
|
return self._loop.run_until_complete(super().listdir(path, limit, offset))
|
|
|
|
def delete(self, path: str = "", permanently: bool = False) -> None:
|
|
return self._loop.run_until_complete(super().delete(path, permanently))
|
|
|
|
def move(self, source_path: str, destination_path: str, overwrite: bool = False) -> None:
|
|
return self._loop.run_until_complete(super().move(source_path, destination_path, overwrite))
|
|
|
|
def copy(self, source_path: str, destination_path: str, overwrite: bool = False) -> None:
|
|
return self._loop.run_until_complete(super().copy(source_path, destination_path, overwrite))
|
|
|
|
def publish(self, path: str, return_public_url: bool = False) -> str | None:
|
|
return self._loop.run_until_complete(super().publish(path, return_public_url))
|
|
|
|
def unpublish(self, path: str):
|
|
return self._loop.run_until_complete(super().unpublish(path))
|
|
|
|
def upload_file(self, file_path: str, path: str, overwrite: bool = False, chunk_size: int = 1024) -> None:
|
|
return self._loop.run_until_complete(super().upload_file(file_path, path, overwrite, chunk_size))
|
|
|
|
def get_url(self, path: str = "/") -> str:
|
|
return self._loop.run_until_complete(super().get_url(path))
|
|
|
|
def listdir_trash(self, path: str = "/", limit: int = 100, offset: int = 0) -> list[File | Directory]:
|
|
return self._loop.run_until_complete(super().listdir_trash(path, limit, offset))
|
|
|
|
def delete_trash(self, path: str = ""):
|
|
return self._loop.run_until_complete(super().delete_trash(path))
|
|
|
|
def restore_trash(self, path: str, new_name: str = "", overwrite: bool = False):
|
|
return self._loop.run_until_complete(super().restore_trash(path, new_name, overwrite))
|