[module] improvement of docker_ps by adding states

Added state to the module to indicate when no containers are running, vs daemon off, container running
This commit is contained in:
Martin 2020-03-17 23:06:43 +01:00
parent 929f4751c3
commit 1e2f1243f0

View file

@ -21,14 +21,25 @@ import bumblebee.engine
class Module(bumblebee.engine.Module): class Module(bumblebee.engine.Module):
def __init__(self, engine, config): def __init__(self, engine, config):
widget = bumblebee.output.Widget(full_text=self.status) widgets = bumblebee.output.Widget(full_text=self.status)
super(Module, self).__init__(engine, config, widget) super(Module, self).__init__(engine, config, widgets)
self._status = self.status self._status = self.status
self._state = self.state
def update(self, widgets): def update(self, widgets):
self._status = self.status self._status = self.status
self._state = self.state
def status(self, _): def state(self, widget):
state = []
status = self.status(widget)
if status == "OK - 0":
state.append("warning")
elif status in ["n/a", "Daemon off"]:
state.append("critical")
return state
def status(self, widget):
try: try:
cli = docker.DockerClient(base_url='unix://var/run/docker.sock') cli = docker.DockerClient(base_url='unix://var/run/docker.sock')
cli.ping() cli.ping()