1e2f1243f0
Added state to the module to indicate when no containers are running, vs daemon off, container running
50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
"""Displays the number of docker containers running
|
|
|
|
Requires the following python packages:
|
|
* docker
|
|
|
|
"""
|
|
|
|
try:
|
|
import docker
|
|
except ImportError:
|
|
pass
|
|
|
|
from requests.exceptions import ConnectionError
|
|
|
|
import bumblebee.input
|
|
import bumblebee.output
|
|
import bumblebee.engine
|
|
|
|
|
|
class Module(bumblebee.engine.Module):
|
|
def __init__(self, engine, config):
|
|
widgets = bumblebee.output.Widget(full_text=self.status)
|
|
super(Module, self).__init__(engine, config, widgets)
|
|
self._status = self.status
|
|
self._state = self.state
|
|
|
|
def update(self, widgets):
|
|
self._status = self.status
|
|
self._state = self.state
|
|
|
|
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:
|
|
cli = docker.DockerClient(base_url='unix://var/run/docker.sock')
|
|
cli.ping()
|
|
except ConnectionError:
|
|
return "Daemon off"
|
|
except Exception:
|
|
return "n/a"
|
|
return "OK - {}".format(len(cli.containers.list(filters={'status': "running"})))
|