Import python-netdisco_2.2.0.orig.tar.gz

[dgit import orig python-netdisco_2.2.0.orig.tar.gz]

1 files changed, 25 insertions, 0 deletions

diff --git a/netdisco/discoverables/samsung_tv.py b/netdisco/discoverables/samsung_tv.py
new file mode 100644
index 0000000..4ad31bb
--- /dev/null
+++ b/netdisco/discoverables/samsung_tv.py
@@ -0,0 +1,25 @@
+"""Discover Samsung Smart TV services."""
+from . import SSDPDiscoverable
+from ..const import ATTR_NAME
+
+# For some models, Samsung forces a [TV] prefix to the user-specified name.
+FORCED_NAME_PREFIX = '[TV]'
+
+
+class Discoverable(SSDPDiscoverable):
+    """Add support for discovering Samsung Smart TV services."""
+
+    def get_entries(self):
+        """Get all the Samsung RemoteControlReceiver entries."""
+        return self.find_by_st(
+            "urn:samsung.com:device:RemoteControlReceiver:1")
+
+    def info_from_entry(self, entry):
+        """Get most important info, by default the description location."""
+        info = super().info_from_entry(entry)
+
+        # Strip the forced prefix, if present
+        if info[ATTR_NAME].startswith(FORCED_NAME_PREFIX):
+            info[ATTR_NAME] = info[ATTR_NAME][len(FORCED_NAME_PREFIX):].strip()
+
+        return info