diff --git a/api/api/consumers.py b/api/api/consumers.py index 6d5f6ca..2b517c7 100644 --- a/api/api/consumers.py +++ b/api/api/consumers.py @@ -1,7 +1,12 @@ # api/consumers.py import json +import os +import shutil +from pathlib import Path from channels.generic.websocket import WebsocketConsumer +UPLOAD_DIR = Path("/tmp/freqsplit") + class MediaConsumer(WebsocketConsumer): def connect(self): self.accept() @@ -26,4 +31,11 @@ class MediaConsumer(WebsocketConsumer): def disconnect(self, close_code): print("Disconnected from Websocket") - print("Stored file UUIDs:", self.file_uuid) \ No newline at end of file + print("Stored file UUIDs:", self.file_uuid) + for file_uuid in self.file_uuid: + dir_path = os.path.join(UPLOAD_DIR, file_uuid); + try: + if os.path.exists(dir_path): + shutil.rmtree(dir_path) + except Exception as e: + print(f"Error: Failed to cleanup {dir_path}: {e}") \ No newline at end of file