bumblebee-status/modules/contrib/http_status.py

# pylint: disable=C0111,R0903

"""Display HTTP status code

Parameters:
    * http__status.label: Prefix label (optional)
    * http__status.target: Target to retrieve the HTTP status from
    * http__status.expect: Expected HTTP status
"""

from requests import head

import psutil

import core.module
import core.widget
import core.decorators

class Module(core.module.Module):
    UNK = 'UNK'

    @core.decorators.every(seconds=30)
    def __init__(self, config):
        super().__init__(config, core.widget.Widget(self.output))

        self.__label = self.parameter('label')
        self.__target = self.parameter('target')
        self.__expect = self.parameter('expect', '200')

    def labelize(self, s):
        if self.__label is None:
            return s
        return '{}: {}'.format(self.__label, s)

    def getStatus(self):
        try:
            res = head(self.__target)
        except Exception as e:
            print(e)
            return self.UNK
        else:
            status = str(res.status_code)
            return status

    def getOutput(self):
        if self.__status == self.__expect:
            return self.labelize(self.__status)
        else:
            reason = ' != {}'.format(self.__expect)
            return self.labelize('{}{}'.format(self.__status, reason))

    def output(self, widget):
        return self.__output

    def update(self):
        self.__status = self.getStatus()
        self.__output = self.getOutput()

    def state(self, widget):
        if self.__status == self.UNK:
            return 'warning'
        if self.__status != self.__expect:
            return 'critical'
        return self.__output

# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4