python-swidget 1.4.10__tar.gz → 1.4.11__tar.gz

{python_swidget-1.4.10 → python_swidget-1.4.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: python-swidget
-Version: 1.4.10
+Version: 1.4.11
 Summary: Python API for Swidget smart devices
 Home-page: https://github.com/swidget/python-swidget
 License: GPL-3.0-or-later

{python_swidget-1.4.10 → python_swidget-1.4.11}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "python-swidget"
-version = "1.4.10"
+version = "1.4.11"
 description = "Python API for Swidget smart devices"
 license = "GPL-3.0-or-later"
 authors = ["Swidget"]

{python_swidget-1.4.10 → python_swidget-1.4.11}/swidget/discovery.py

@@ -17,7 +17,7 @@ from .swidgetswitch import SwidgetSwitch
 from .swidgettimerswitch import SwidgetTimerSwitch
 
 RESPONSE_SEC = 5
-SWIDGET_ST = "urn:swidget:pico:1"
+SWIDGET_STS = ("urn:swidget:pico:1", "urn:swidget:video:1")
 # Generous timeout: TLS handshakes on the device's MCU can take several
 # seconds on first connection.
 DETECT_TIMEOUT_SEC = 10
@@ -91,7 +91,7 @@ class SwidgetProtocol(ssdp.SimpleServiceDiscoveryProtocol):
         headers = {h[0]: h[1] for h in response.headers}
         mac_address = headers["USN"].split("-")[-1]
         ip_address = urlparse(headers["LOCATION"]).hostname
-        if headers["ST"] == SWIDGET_ST:
+        if headers["ST"] in SWIDGET_STS:
             device_type = headers["SERVER"].split(" ")[1].split("+")[0]
             insert_type = headers["SERVER"].split(" ")[1].split("+")[1].split("/")[0]
             friendly_name = headers["SERVER"].split("/")[2].strip('"')
@@ -116,17 +116,19 @@ async def discover_devices(timeout=RESPONSE_SEC):
         SwidgetProtocol, family=socket.AF_INET
     )
 
-    # Send out an M-SEARCH request, requesting Swidget service types.
-    search_request = ssdp.SSDPRequest(
-        "M-SEARCH",
-        headers={
-            "HOST": "239.255.255.250:1900",
-            "MAN": '"ssdp:discover"',
-            "MX": timeout,
-            "ST": SWIDGET_ST,
-        },
-    )
-    search_request.sendto(transport, (SwidgetProtocol.MULTICAST_ADDRESS, 1900))
+    # SSDP M-SEARCH matches one ST per request, so emit one packet per
+    # supported service type. Devices reply only to their matching ST.
+    for st in SWIDGET_STS:
+        search_request = ssdp.SSDPRequest(
+            "M-SEARCH",
+            headers={
+                "HOST": "239.255.255.250:1900",
+                "MAN": '"ssdp:discover"',
+                "MX": timeout,
+                "ST": st,
+            },
+        )
+        search_request.sendto(transport, (SwidgetProtocol.MULTICAST_ADDRESS, 1900))
     await asyncio.sleep(timeout)
     return devices