51 lines
1.3 KiB
Python
51 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"})))
|