✨ Added streaming up- and downloads, fixed file sizes and added a (not working) dashboard

classquiz/storage/local_storage.py

@@ -2,32 +2,45 @@
 #  License, v. 2.0. If a copy of the MPL was not distributed with this
 #  file, You can obtain one at https://mozilla.org/MPL/2.0/.
 
-import io
 import os
+from typing import BinaryIO, Generator
 
 import aiofiles
 import aiofiles.os
 
+_DEFAULT_CHUNK_SIZE = 32768  # bytes; arbitrary
+
+
+async def aioshutil_copyfileobj(async_fsrc, async_fdst, *, chunksize: int = _DEFAULT_CHUNK_SIZE) -> None:
+    while (chunk := await async_fsrc.read(chunksize)) != b"":
+        await async_fdst.write(chunk)
+
 
 class LocalStorage:
     def __init__(self, base_path: str):
         self.base_path = base_path
 
-    async def get_file(self, file_name: str) -> io.BytesIO | None:
+    async def download(self, file_name: str) -> Generator | None:
         try:
             async with aiofiles.open(file=os.path.join(self.base_path, file_name), mode="rb") as f:
-                return io.BytesIO(await f.read())
+                yield f.read()
         except FileNotFoundError:
-            return None
+            yield None
 
-    async def write_file(self, file_name: str, data: bytes) -> None:
+    async def upload(self, file_name: str, data: BinaryIO) -> None:
         async with aiofiles.open(file=os.path.join(self.base_path, file_name), mode="wb") as f:
-            await f.write(data)
+            await aioshutil_copyfileobj(data, f)
 
-    async def delete_file(self, file_names: [str]) -> None:
+    async def delete(self, file_names: [str]) -> None:
         for i in file_names:
             try:
                 await aiofiles.os.remove(os.path.join(self.base_path, i))
             except FileNotFoundError:
                 pass
         return None
+
+    def size(self, file_name: str) -> int | None:
+        try:
+            return os.stat(os.path.join(self.base_path, file_name))
+        except FileNotFoundError:
+            return None