format

Author: Gawor270

examples/selective_subscription/__init__.py

@@ -1 +1 @@
-"""Selective subscription demo package."""
+"""Selective subscription demo package."""

examples/selective_subscription/selective_subscription/__init__.py

@@ -1 +1 @@
-"""Selective subscription demo package."""
+"""Selective subscription demo package."""

examples/selective_subscription/selective_subscription/app.py

@@ -11,7 +11,9 @@
 from .room_service import RoomService
 
 room_service = RoomService()
-templates = Jinja2Templates(directory=str(Path(__file__).resolve().parent.parent / "templates"))
+templates = Jinja2Templates(
+    directory=str(Path(__file__).resolve().parent.parent / "templates")
+)
 
 
 async def create_peer(request: Request) -> Response:
@@ -22,21 +24,23 @@ async def create_peer(request: Request) -> Response:
 
         if not room_name or not peer_name:
             return JSONResponse(
-                {"error": "room_name and peer_name are required"},
-                status_code=400
+                {"error": "room_name and peer_name are required"}, status_code=400
             )
 
         peer, token = room_service.create_peer()
 
-        return JSONResponse({
-            "peer_id": peer.id,
-            "token": token,
-            "room_name": room_name,
-            "peer_name": peer_name
-        })
+        return JSONResponse(
+            {
+                "peer_id": peer.id,
+                "token": token,
+                "room_name": room_name,
+                "peer_name": peer_name,
+            }
+        )
     except Exception as e:
         return JSONResponse({"error": str(e)}, status_code=500)
 
+
 async def subscribe_peer(request: Request) -> Response:
     try:
         body = await request.json()
@@ -45,8 +49,7 @@ async def subscribe_peer(request: Request) -> Response:
 
         if not peer_id or not target_peer_id:
             return JSONResponse(
-                {"error": "peer_id and target_peer_id are required"},
-                status_code=400
+                {"error": "peer_id and target_peer_id are required"}, status_code=400
             )
 
         room_service.subscibe_peer(peer_id, target_peer_id)
@@ -63,15 +66,15 @@ async def subscribe_tracks(request: Request) -> Response:
 
         if not peer_id or not track_ids:
             return JSONResponse(
-                {"error": "peer_id and track_ids are required"},
-                status_code=400
+                {"error": "peer_id and track_ids are required"}, status_code=400
             )
 
         room_service.subscribe_tracks(peer_id, track_ids)
         return JSONResponse({"status": "subscribed"})
     except Exception as e:
         return JSONResponse({"error": str(e)}, status_code=500)
 
+
 async def serve_index(request: Request) -> Response:
     return templates.TemplateResponse("index.html", {"request": request})
 
@@ -96,4 +99,4 @@ async def serve_index(request: Request) -> Response:
 app = Starlette(
     routes=routes,
     middleware=middleware,
-)
+)

examples/selective_subscription/selective_subscription/config.py

@@ -4,4 +4,4 @@
 FISHJAM_TOKEN = os.environ["FISHJAM_MANAGEMENT_TOKEN"]
 FISHJAM_URL = os.getenv("FISHJAM_URL", "http://localhost:5002")
 HOST = os.getenv("HOST", "localhost")
-PORT = int(os.getenv("PORT", "8000"))
+PORT = int(os.getenv("PORT", "8000"))

examples/selective_subscription/selective_subscription/notification_handler.py

@@ -46,4 +46,3 @@ async def handle_track_removed(notification: ServerMessageTrackRemoved):
 
     async def start(self) -> None:
         await self._notifier.connect()
-

examples/selective_subscription/selective_subscription/room_service.py

@@ -9,10 +9,9 @@
 class RoomService:
     def __init__(self):
         self.fishjam = FishjamClient(FISHJAM_ID, FISHJAM_TOKEN)
-        self.room = self.fishjam.create_room(RoomOptions(
-            max_peers=10,
-            room_type="conference"
-        ))
+        self.room = self.fishjam.create_room(
+            RoomOptions(max_peers=10, room_type="conference")
+        )
 
     def get_or_create_room(self) -> Room:
         if self.room:
@@ -42,6 +41,7 @@ def get_peer_session(self, peer_id: str):
         room = self.get_or_create_room()
         for p in room.peers:
             if p.id == peer_id:
+
                 class _Session:
                     def __init__(self):
                         self.subscribed_peers: set[str] = set()

fishjam/api/_fishjam_client.py

@@ -262,9 +262,9 @@ def create_livestream_streamer_token(self, room_id: str) -> str:
         )
 
         return response.token
-    
+
     def subscribe_peer(self, room_id: str, peer_id: str, target_peer_id: str):
-        "" "Subscribe a peer to all tracks of another peer."""
+        "" "Subscribe a peer to all tracks of another peer." ""
 
         response = self._request(
             room_subscribe_peer,
@@ -274,15 +274,15 @@ def subscribe_peer(self, room_id: str, peer_id: str, target_peer_id: str):
         )
 
         return response
-    
+
     def subscribe_tracks(self, room_id: str, peer_id: str, track_ids: list[str]):
-        "" "Subscribe a peer to specific tracks of another peer."""
+        "" "Subscribe a peer to specific tracks of another peer." ""
 
         response = self._request(
             room_subscribe_tracks,
             room_id=room_id,
             id=peer_id,
-            body=SubscribeTracksBody(track_ids=track_ids)
+            body=SubscribeTracksBody(track_ids=track_ids),
         )
 
         return response