Skip to content
Snippets Groups Projects
connection_tester.py 1.97 KiB
Newer Older
  • Learn to ignore specific revisions
  • import urllib
    import simplejson
    
    from modules.communication.liquidsoap.communicator import LiquidSoapCommunicator
    from libraries.database.broadcasts import ScheduleEntry
    from libraries.base.common import AuraCommon
    from libraries.exceptions.auraexceptions import LQConnectionError
    
    
    class ConnectionTester(AuraCommon):
    
        def __init__(self):
            super(ConnectionTester, self).__init__(logger=False)
    
        def get_connection_status(self):
            dbconn = self.test_db_conn()
            pvconn = self.test_pv_conn()
            lqsconn = self.test_lqs_conn()
            tankconn = self.test_tank_conn()
            redisconn = self.test_redis_conn()
    
            status = dict()
            status["db"] = dbconn
            status["pv"] = pvconn
            status["lqs"] = lqsconn
            status["tank"] = tankconn
            status["redis"] = redisconn
    
            return simplejson.dumps(status)
    
        def test_db_conn(self):
            try:
                ScheduleEntry.select_all()
            except:
                return False
    
            return True
    
        def test_lqs_conn(self):
            try:
                lsc = LiquidSoapCommunicator(self.config)
                s = lsc.get_mixer_status()
            except LQConnectionError:
                return False
    
            if s:
                return True
            return False
    
        def test_pv_conn(self):
            return self.test_url_connection(self.config.get("calendarurl"))
    
        def test_tank_conn(self):
            return self.test_url_connection(self.config.get("importerurl")+"1")
    
        def test_redis_conn(self):
            from modules.communication.redis.adapter import ClientRedisAdapter
            try:
                cra = ClientRedisAdapter()
                cra.publish("aura", "status")
            except:
                return False
    
            return True
    
        def test_url_connection(self, url):
            try:
                request = urllib.request.Request(url)
                response = urllib.request.urlopen(request)
                response.read()
            except Exception as e:
                print(e)
                return False
    
            return True