Remove ws connection after unsubscribe (#547)

to allow reconnection

app/modules/cdr/endpoints_cdr.py

@@ -1897,6 +1897,16 @@ async def create_curriculum_membership(
             status_code=403,
             detail="You can't remove a curriculum to another user.",
         )
+
+    wanted_curriculum = await cruds_cdr.get_curriculum_by_id(
+        db=db,
+        curriculum_id=curriculum_id,
+    )
+    if not wanted_curriculum:
+        raise HTTPException(
+            status_code=404,
+            detail="Invalid curriculum_id",
+        )
     db_user = await get_user_by_id(db=db, user_id=user_id)
     if not db_user:
         raise HTTPException(
@@ -1942,7 +1952,7 @@ async def create_curriculum_membership(
                 message=schemas_cdr.NewUserWSMessageModel(
                     data=schemas_cdr.CdrUser(
                         curriculum=schemas_cdr.CurriculumComplete(
-                            **curriculum.__dict__,
+                            **wanted_curriculum.__dict__,
                         ),
                         **db_user.__dict__,
                     ),

app/types/websocket.py

@@ -122,7 +122,8 @@ def _remove_task_from_listening_tasks_callback(
         Asyncio task callback to remove the task from the listening_tasks dict
         The method is called after the listening task is done or cancelled
         """
-        self.listening_tasks.pop(room_id)
+        self.listening_tasks.pop(room_id, None)
+        self.connections.pop(room_id, None)
         hyperion_error_logger.info(
             f"Websocket: unsubscribed broadcaster from channel {room_id} for worker {os.getpid()}",
         )