[core] restructure to allow PIP packaging
OK - so I have to admit I *hate* the fact that PIP seems to require a subdirectory named like the library. But since the PIP package is something really nifty to have (thanks to @tony again!!!), I updated the codebase to hopefully conform with what PIP expects. Testruns so far look promising...
This commit is contained in:
parent
1d25be2059
commit
320827d577
146 changed files with 2509 additions and 2 deletions
70
bumblebee_status/modules/contrib/http_status.py
Normal file
70
bumblebee_status/modules/contrib/http_status.py
Normal file
|
@ -0,0 +1,70 @@
|
|||
# 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
|
||||
|
||||
contributed by `valkheim <https://github.com/valkheim>`_ - many thanks!
|
||||
"""
|
||||
|
||||
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, theme):
|
||||
super().__init__(config, theme, 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
|
Loading…
Add table
Add a link
Reference in a new issue