send file_uuids to websocket

2025-03-20 00:03:34 +05:30
parent 5fba3b42c5
commit 1cb3a1f197
2 changed files with 18 additions and 3 deletions
@@ -5,6 +5,7 @@ from channels.generic.websocket import WebsocketConsumer
 class MediaConsumer(WebsocketConsumer):
    def connect(self):
        self.accept()
+        self.file_uuid = [] # List to store file uuids
        self.send(text_data=json.dumps({
            "message": "Connected to WebSocket server!"
        }))
@@ -12,11 +13,17 @@ class MediaConsumer(WebsocketConsumer):
    def receive(self, text_data):
        data = json.loads(text_data)
        message = data.get("message", "")
-        
-        if message == "ping":
+
+        # If message contains a file UUID, store it
+        if "file_uuid" in data:
+            uuid = data["file_uuid"]
+            self.file_uuid.append(uuid)
+            self.send(text_data=json.dumps({"response": f"UUID {uuid} stored."}))
+        elif message == "ping":
            self.send(text_data=json.dumps({"response": "pong"}))
        else:
            self.send(text_data=json.dumps({"response": f"Received: {message}"}))
            
    def disconnect(self, close_code):
-        print("Disconnected from Websocket")
+        print("Disconnected from Websocket")
+        print("Stored file UUIDs:", self.file_uuid)