From 21977244fd28d66a46ee74d45d37544501bdc3f8 Mon Sep 17 00:00:00 2001
From: hkc <hatkidchan@gmail.com>
Date: Wed, 22 May 2024 14:37:27 +0300
Subject: [PATCH] Honestly, I don't even *need* AsyncDownloader

---
 make.py | 59 ++-------------------------------------------------------
 1 file changed, 2 insertions(+), 57 deletions(-)

diff --git a/make.py b/make.py
index 5b6d350..10c5d6f 100644
--- a/make.py
+++ b/make.py
@@ -1,11 +1,10 @@
 #!/usr/bin/env python3
 import json
 from pathlib import Path
-from typing import Optional, Union
+from typing import Union
 import asyncio
 from httpx import AsyncClient
-from asyncio.queues import Queue
-from rich.progress import TaskID, Progress
+from rich.progress import Progress
 from zipfile import ZipFile
 from PIL import Image
 
@@ -43,60 +42,6 @@ meta_url: str = (
 meta_filename = INPUT_PATH / f"meta-{meta_url.split('/')[-1]}"
 
 
-class AsyncDownloader:
-    def __init__(
-        self, files: Optional[list[tuple[str, Path]]] = None, maxfiles: int = 8
-    ):
-        self._queue: Queue[tuple[str, Path]] = Queue(maxfiles)
-        self._client = AsyncClient()
-        self._progress: Optional[Progress] = None
-        self._tasks: list[TaskID] = []
-
-        for url, path in files or []:
-            self._queue.put_nowait((url, path))
-
-    def add_file(self, url: str, path: Union[Path, str]):
-        self._queue.put_nowait((url, Path(path)))
-
-    async def __aenter__(self):
-        await self._client.__aenter__()
-        self._progress = Progress()
-        self._progress.__enter__()
-        for i in range(self._queue.maxsize):
-            self._tasks.append(
-                self._progress.add_task(f"Downloader {i + 1}", start=False)
-            )
-        return self
-
-    async def _worker(self, task: TaskID):
-        if not self._progress:
-            return
-        while not self._queue.empty():
-            url, path = await self._queue.get()
-            with path.open("wb") as fout:
-                async with self._client.stream("GET", url) as req:
-                    size = int(req.headers.get("Content-Length", 0))
-                    self._progress.start_task(task)
-                    self._progress.update(
-                        task,
-                        total=size,
-                        completed=0,
-                        description=f"{path.name}",
-                    )
-                    async for chunk in req.aiter_bytes(8192):
-                        self._progress.advance(task, fout.write(chunk))
-            self._progress.stop_task(task)
-
-    async def run(self):
-        await asyncio.gather(*[self._worker(task) for task in self._tasks])
-
-    async def __aexit__(self, a, b, c):
-        if self._progress is None:
-            raise ValueError("how did that happen?")
-        self._progress.__exit__(a, b, c)
-        await self._client.__aexit__()
-
-
 async def download_if_missing(url: str, path: Union[Path, str]):
     path = Path(path)
     if path.exists():