l.behm/bumblebee-status
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

[modules/http_status] quotes

Browse source
This commit is contained in:
tobi-wan-kenobi 2020-04-19 14:39:21 +02:00
parent f4d382eb19
commit 967ec5892b
1 changed files with 9 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
modules/contrib/http_status.py
View file

@ -16,22 +16,22 @@ import bumblebee.output
import bumblebee.engine
class Module(bumblebee.engine.Module):
UNK = "UNK"
UNK = 'UNK'
def __init__(self, engine, config):
widget = bumblebee.output.Widget(full_text=self.output)
super(Module, self).__init__(engine, config, widget)
self._label = self.parameter("label")
self._target = self.parameter("target")
self._expect = self.parameter("expect", "200")
self._label = self.parameter('label')
self._target = self.parameter('target')
self._expect = self.parameter('expect', '200')
self._status = self.getStatus()
self._output = self.getOutput()
def labelize(self, s):
if self._label is None:
return s
return "{}: {}".format(self._label, s)
return '{}: {}'.format(self._label, s)
def getStatus(self):
try:
@ -47,8 +47,8 @@ class Module(bumblebee.engine.Module):
if self._status == self._expect:
return self.labelize(self._status)
else:
reason = " != {}".format(self._expect)
return self.labelize("{}{}".format(self._status, reason))
reason = ' != {}'.format(self._expect)
return self.labelize('{}{}'.format(self._status, reason))
def output(self, widget):
return self._output
@ -59,9 +59,9 @@ class Module(bumblebee.engine.Module):
def state(self, widget):
if self._status == self.UNK:
return "warning"
return 'warning'
if self._status != self._expect:
return "critical"
return 'critical'
return self._output