vdstatus/plugins/ping.py

import os
import random
import time
import threading


class PluginThread(threading.Thread):
    def __init__(self, section, config, thread_id):
        threading.Thread.__init__(self)
        self.threadID = thread_id
        self.hosts = config.get(section, 'hosts').split(',')
        self.title = config.get(section, 'title')
        self.timeout = config.get(section, 'timeout', fallback='150')
        self.status = dict()
        self.freq = config.getint(section, 'freq', fallback=5)
        self.format_status('n/a')
        self.hide = False
        self.should_stop = False

    def format_status(self, state):
        self.status['full_text'] = self.title + ': ' + state
        if state == 'on':
            self.status['urgent'] = False
            self.hide = True
        else:
            self.status['urgent'] = True

    def main(self):
        random.shuffle(self.hosts)
        for host in self.hosts:
            fping = 'fping -qc1t' + self.timeout + ' ' + host + ' &>/dev/null'
            response = os.system(fping)
            if response == 0:
                self.format_status('on')
                break
            self.format_status('off')

    def sleep(self):
        seconds = 0
        while seconds < self.freq:
            time.sleep(1)
            seconds += 1
        del seconds

    def stop(self):
        self.should_stop = True

    def run(self):
        while True:
            if self.should_stop is False:
                self.main()
                self.sleep()
            else:
                break