71 lines
1.4 KiB
Python
Executable File
71 lines
1.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import json
|
|
import os
|
|
import subprocess
|
|
|
|
from block import *
|
|
|
|
import requests
|
|
|
|
ICON = '\U000f056e'
|
|
MONDASH_URL = 'https://mondash.org/{dashid}'
|
|
STATUS_COLOR = {
|
|
'OK': COLOR_SUCCESS,
|
|
'Warning': COLOR_WARNING,
|
|
'Critical': COLOR_DANGER,
|
|
'Unknown': COLOR_SECONDARY,
|
|
}
|
|
|
|
|
|
class Mondash(Block):
|
|
def dashid(self):
|
|
id = self.instance()
|
|
|
|
if id.startswith('$env_'):
|
|
id = os.environ[id[5:]]
|
|
|
|
return id
|
|
|
|
def execute(self):
|
|
if self.button_is(BTN_LEFT):
|
|
subprocess.check_call([
|
|
'xdg-open',
|
|
MONDASH_URL.format(dashid=self.dashid()),
|
|
])
|
|
|
|
result = []
|
|
|
|
for status, count in self.get_counts().items():
|
|
if count == 0:
|
|
continue
|
|
|
|
color = STATUS_COLOR[status]
|
|
|
|
result.append(self.color_text(str(count), color))
|
|
|
|
return ' / '.join(result)
|
|
|
|
def get_counts(self):
|
|
counts = {'OK': 0, 'Warning': 0, 'Critical': 0, 'Unknown': 0}
|
|
|
|
url = MONDASH_URL.format(dashid=self.dashid()) + '.json'
|
|
data = requests.get(url).json()
|
|
|
|
if not 'metrics' in data:
|
|
raise Exception('No metrics block found in result data')
|
|
|
|
for metric in data['metrics']:
|
|
counts[metric['status']] += 1
|
|
|
|
return counts
|
|
|
|
|
|
def main():
|
|
block = Mondash(ICON)
|
|
block.render()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|