Fix URL as direct camera source #522

AlexxIT · Jul 11, 2023 · e51aac0 · e51aac0
1 parent 8836726
commit e51aac0
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/custom_components/webrtc/__init__.py b/custom_components/webrtc/__init__.py
@@ -165,12 +165,21 @@ async def ws_connect(hass: HomeAssistantType, params) -> str:
             params={"name": entity, "src": src},
             timeout=3,
         )
-        if r.ok:
-            src = entity
+        assert r.ok, f"Can't update URL for {entity}"
+        src = entity
 
     elif src := params.get("url"):
         if "{{" in src or "{%" in src:
             src = Template(src, hass).async_render()
+
+        session = async_get_clientsession(hass)
+        r = await session.patch(
+            urljoin(go_url, "api/streams"),
+            params={"name": src, "src": src},
+            timeout=3,
+        )
+        assert r.ok, f"Can't update URL for {src}"
+
     else:
         raise Exception("Missing url or entity")