✨ Files get deleted, if quiz gets deleted

classquiz/storage/local_storage.py

@@ -1,17 +1,28 @@
 import io
 import os
 import aiofiles
+import aiofiles.os
 
 
 class LocalStorage:
     def __init__(self, base_path: str):
         self.base_path = base_path
 
-    async def get_file(self, file_name: str) -> io.BytesIO:
-        async with aiofiles.open(file=os.path.join(self.base_path, file_name), mode='rb') as f:
-            return io.BytesIO(await f.read())
+    async def get_file(self, file_name: str) -> io.BytesIO | 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())
+        except FileNotFoundError:
+            return None
 
     async def write_file(self, file_name: str, data: bytes) -> None:
         async with aiofiles.open(file=os.path.join(self.base_path, file_name), mode='wb') as f:
             await f.write(data)
 
+    async def delete_file(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