bumblebee-status/modules/contrib/docker_ps.py
2020-04-13 13:38:05 +02:00

43 lines
1.1 KiB
Python

# -*- coding: utf-8 -*-
"""Displays the number of docker containers running
Requires the following python packages:
* docker
"""
import docker
from requests.exceptions import ConnectionError
import core.module
import core.widget
import core.decorators
class Module(core.module.Module):
@core.decorators.every(seconds=5)
def __init__(self, config):
super().__init__(config, core.widget.Widget(self.docker_info))
self.__info = ''
def state(self, widget):
state = []
if self.__info == 'OK - 0':
state.append('warning')
elif self.__info in ['n/a', 'daemon off']:
state.append('critical')
return state
def docker_info(self, widget):
try:
cli = docker.DockerClient(base_url='unix://var/run/docker.sock')
cli.ping()
except ConnectionError:
self.__info = 'daemon off'
except Exception:
self.__info = 'n/a'
self.__info = 'OK - {}'.format(len(cli.containers.list(filters={'status': 'running'})))
return self.__info
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4