Update satip_facts.py, fix syntax for hardware autoinstall checks

library/satip_facts.py

@@ -5,23 +5,26 @@ module: hardware_facts
 short_description: "check if at least one SAT>IP server responds on the network"
 description:
      - This script sends a multicast message and awaits responses by Sat>IP servers.
-       Returns the boolean variable 'satip_detected'
+       Returns a list of detected SAT>IP servers with their name and capabilites.
 '''
 EXAMPLES = '''
 - name: "detect SAT>IP Server on the network"
   action: satip_facts
 
 - debug:
-    var: satip_detected
+    var: satip_devices
 '''
 
 import json
 import socket
 import sys
 import time
-
+import xml.etree.ElementTree as ET
+import requests
+from contextlib import contextmanager
 from ansible.module_utils.basic import *
 
+SSDP_BIND = "0.0.0.0"
 SSDP_ADDR = "239.255.255.250"
 SSDP_PORT = 1900
 # SSDP_MX = max delay for server response
@@ -37,26 +40,79 @@ ssdpRequest = "\r\n".join((
     "ST: %s" % (SSDP_ST),
     "\r\n"))
 
-def main():
-    module = AnsibleModule(argument_spec={}, supports_check_mode=True,)
-    sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
-    # according to Sat>IP Specification 1.2.2, p. 20
-    # a client should send three requests within 100 ms with a ttl of 2
+@contextmanager
+def socket_manager(*args, **kwargs):
+    """provide a context manager for socket"""
+    sock = socket.socket(*args, **kwargs)
+    sock.setblocking(False)
     sock.setsockopt(socket.IPPROTO_IP, socket.IP_MULTICAST_TTL, 2)
-    sock.settimeout(SSDP_MX + 0.5)
-    for _ in range(3):
-        sock.sendto(ssdpRequest, (SSDP_ADDR, SSDP_PORT))
-        time.sleep(0.03)
+    sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
     try:
-        response = sock.recv(1000)
-        if response and "SERVER:" in response:
-            got_response = True
-        else:
-            raise ValueError('No satip server detected')
-    except (socket.timeout, ValueError):
-        got_response = False
+        sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEPORT, 1)
+    except socket.error:
+        pass
+    sock.settimeout(SSDP_MX + 0.5)
+    sock.bind((SSDP_BIND, SSDP_PORT))
+    try:
+        yield sock
+    finally:
+        sock.close()
 
-    module.exit_json(changed=False, ansible_facts={'satip_detected': got_response})
 
-if __name__ == '__main__':  
+def parse_satip_xml(data):
+    """ Parse SAT>IP XML data.
+    Args:
+        data (str): XML input data..
+    Returns:
+        dict: Parsed SAT>IP device name and frontend information.
+    """
+    result = {'name': '', 'frontends': {}}
+    if data:
+        root = ET.fromstring(data)
+        name = root.find('.//*/{urn:schemas-upnp-org:device-1-0}friendlyName')
+        result['name'] = name.text
+        satipcap = root.find('.//*/{urn:ses-com:satip}X_SATIPCAP')
+        if satipcap is None:
+            raise ValueError("Invalid SAT>IP device description")
+        caps = {}
+        for system in satipcap.text.split(","):
+            cap = system.split("-")
+            if cap:
+                count = int(cap[1])
+                if cap[0] in caps:
+                    count = count + caps[cap[0]]
+                caps[cap[0]] = count
+        result['frontends'] = caps
+    return result
+
+
+def main():
+    description_urls = []
+    device_list = []
+    module = AnsibleModule(argument_spec={}, supports_check_mode=True,)
+    with socket_manager(socket.AF_INET, socket.SOCK_DGRAM) as sock:
+        # according to Sat>IP Specification 1.2.2, p. 20
+        # a client should send three requests within 100 ms with a ttl of 2
+
+        for _ in range(3):
+            sock.sendto(ssdpRequest, (SSDP_ADDR, SSDP_PORT))
+            time.sleep(0.03)
+        try:
+            response = sock.recv(1024)
+            if response and "SERVER:" in response:
+                for line in response.splitlines():
+                    if "LOCATION" in line:
+                        url = line.split()[-1].strip()
+                        if url not in description_urls:
+                            description_urls.append(url)
+                            info = requests.get(url, timeout=2)
+                            device_list.append(parse_satip_xml(info.text))
+            else:
+                raise ValueError('No satip server detected')
+        except (socket.timeout, ValueError, ET.ParseError):
+            pass
+
+    module.exit_json(changed=False, ansible_facts={'satip_devices': device_list})
+
+if __name__ == '__main__':
     main()