statusmanager.py

from dbmanager import PMDbManager
import threading
import re
import time
from subprocess import getstatusoutput as gso


class PMPingThread:
    def __init__(self, server):
        self.server = server
        self.status = "Pending"
        self.cycle_pause = 2
        self.stopped = False
        self.thread = threading.Thread(target=self.ping_loop)
        self.thread.start()

    def ping_loop(self):
        while True:
            if self.stopped:
                break
            ping_status_output = gso('ping -W 2 -c 1 ' + self.server)
            if ping_status_output[0] == 0:
                host_state = re.search('time=(\d+\.*\d+\s)ms', ping_status_output[1])
                if host_state is not None:
                    time.sleep(self.cycle_pause)
                    self.status = '{} ms'.format(host_state.group(1))
            elif ping_status_output[0] == 1:
                    self.status = 'offline'
            elif ping_status_output[0] == 2:
                time.sleep(self.cycle_pause)
                if ping_status_output[1].find('unknown host'):
                    self.status = 'unk host'
                elif ping_status_output[1].find('connect:'):
                    self.status = 'no net'


    def stop(self):
        self.cycle_pause = 0
        self.stopped = True


class PMStatusManager:
    def __init__(self, db_manager: PMDbManager):
        self.db = db_manager
        self.storage = {}
        self.start()

    def start(self):
        """
        Starts all ping threads
        """
        for i in self.db.get_servers():
            s = (i[0], PMPingThread(i[1]))
            if i[2] not in self.storage:

                self.storage[i[2]] = [s]
            else:
                self.storage[i[2]].append(s)

    def get_group(self, group_id):
        """
        :param group_id:
        :return: list of tuples (server_name, server_status) owned by group
        """
        return self.storage[int(group_id)]

    def stop(self):
        """
        Stops all ping threads
        """
        for group in self.storage:
            for server in self.storage[group]:
                server[1].stop()
        self.storage = {}

    def restart(self):
        self.stop()
        self.start()