break apart the ever-growing owrx/source.py

owrx/sdr.py

@@ -0,0 +1,90 @@
+from owrx.config import PropertyManager
+from owrx.feature import FeatureDetector, UnknownFeatureException
+import sys
+
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+class SdrService(object):
+    sdrProps = None
+    sources = {}
+    lastPort = None
+
+    @staticmethod
+    def getNextPort():
+        pm = PropertyManager.getSharedInstance()
+        (start, end) = pm["iq_port_range"]
+        if SdrService.lastPort is None:
+            SdrService.lastPort = start
+        else:
+            SdrService.lastPort += 1
+            if SdrService.lastPort > end:
+                raise IndexError("no more available ports to start more sdrs")
+        return SdrService.lastPort
+
+    @staticmethod
+    def loadProps():
+        if SdrService.sdrProps is None:
+            pm = PropertyManager.getSharedInstance()
+            featureDetector = FeatureDetector()
+
+            def loadIntoPropertyManager(dict: dict):
+                propertyManager = PropertyManager()
+                for (name, value) in dict.items():
+                    propertyManager[name] = value
+                return propertyManager
+
+            def sdrTypeAvailable(value):
+                try:
+                    if not featureDetector.is_available(value["type"]):
+                        logger.error(
+                            'The RTL source type "{0}" is not available. please check requirements.'.format(
+                                value["type"]
+                            )
+                        )
+                        return False
+                    return True
+                except UnknownFeatureException:
+                    logger.error(
+                        'The RTL source type "{0}" is invalid. Please check your configuration'.format(value["type"])
+                    )
+                    return False
+
+            # transform all dictionary items into PropertyManager object, filtering out unavailable ones
+            SdrService.sdrProps = {
+                name: loadIntoPropertyManager(value) for (name, value) in pm["sdrs"].items() if sdrTypeAvailable(value)
+            }
+            logger.info(
+                "SDR sources loaded. Availables SDRs: {0}".format(
+                    ", ".join(map(lambda x: x["name"], SdrService.sdrProps.values()))
+                )
+            )
+
+    @staticmethod
+    def getSource(id=None):
+        SdrService.loadProps()
+        sources = SdrService.getSources()
+        if not sources:
+            return None
+        if id is None:
+            # TODO: configure default sdr in config? right now it will pick the first one off the list.
+            id = list(sources.keys())[0]
+
+        if not id in sources:
+            return None
+        return sources[id]
+
+    @staticmethod
+    def getSources():
+        SdrService.loadProps()
+        for id in SdrService.sdrProps.keys():
+            if not id in SdrService.sources:
+                props = SdrService.sdrProps[id]
+                sdrType = props["type"]
+                className = "".join(x for x in sdrType.title() if x.isalnum()) + "Source"
+                module = __import__("owrx.source.{0}".format(sdrType), fromlist=[className])
+                cls = getattr(module, className)
+                SdrService.sources[id] = cls(id, props, SdrService.getNextPort())
+        return {key: s for key, s in SdrService.sources.items() if not s.isFailed()}