From e1c052e4a13f467dccebba43823211a4993693bc Mon Sep 17 00:00:00 2001
From: "Dustin C. Hatch" <dustin.hatch@firemon.com>
Date: Tue, 3 Sep 2019 09:11:27 -0500
Subject: [PATCH] Add Zabbix py3status module

---
 i3status            |   4 ++
 py3status/zabbix.py | 137 ++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 141 insertions(+)
 create mode 100644 py3status/zabbix.py

diff --git a/i3status b/i3status
index f83d39e..dc6226c 100644
--- a/i3status
+++ b/i3status
@@ -4,6 +4,7 @@ general {
 	interval = 5
 }
 
+order += "zabbix"
 order += "color uname"
 order += "color uptime"
 order += "cpu_temperature 1"
@@ -11,6 +12,9 @@ order += "cpu_temperature 0"
 order += "load"
 order += "tztime local"
 
+zabbix {
+	url = "https://zabbix.securepassage.com/zabbix/"
+}
 color uname {
 	color = "#444444"
 	uname {
diff --git a/py3status/zabbix.py b/py3status/zabbix.py
new file mode 100644
index 0000000..2a741c2
--- /dev/null
+++ b/py3status/zabbix.py
@@ -0,0 +1,137 @@
+import os
+import requests
+import sys
+import traceback
+import urllib.parse
+import webbrowser
+
+
+class ZabbixError(Exception):
+
+    def __init__(self, *, code=None, data=None, message=None):
+        self.code = code
+        self.data = data
+        self.message = message
+
+    def __str__(self):
+        return '[{}] {}'.format(self.code, self.message)
+
+
+class Zabbix:
+
+    METHOD_PATHS = {
+        'zabbix.status': 'jsrpc.php?output=json-rpc',
+    }
+    DEFAULT_PATH = 'api_jsonrpc.php'
+
+    def __init__(self, url):
+        self.baseurl = url.rstrip('/') + '/'
+        self.auth = None
+
+    def get_triggers(self):
+        return self._call('trigger.get',
+            monitored=True,
+            min_severity=3,
+            filter={
+                'value': 1,
+            },
+        )
+
+    def zabbix_status(self):
+        return self._call('zabbix.status')
+
+    def login(self, user='guest', password=''):
+        self.auth = self._call('user.login', user=user, password=password)
+
+    def logout(self):
+        return self._call('user.logout')
+
+    def _call(self, method, **kwargs):
+        url = urllib.parse.urljoin(
+            self.baseurl,
+            self.METHOD_PATHS.get(method, self.DEFAULT_PATH),
+        )
+        data = {
+            'jsonrpc': '2.0',
+            'method': method,
+            'params': kwargs,
+            'id': 1,
+        }
+        if self.auth is not None:
+            data['auth'] = self.auth
+        r = requests.post(url, json=data, headers={
+            'Content-Type': 'application/json',
+            'Accept': 'application/json',
+        })
+        r.raise_for_status()
+        response = r.json()
+        if 'error' in response:
+            raise ZabbixError(**response['error'])
+        else:
+            return response['result']
+
+
+class Py3status:
+
+    def __init__(self):
+        self.user = 'guest'
+        self.password = ''
+        self.format = 'ZBX: {summary}'
+        self.__zabbix = None
+
+    @property
+    def _zabbix(self):
+        if not self.__zabbix:
+            self.__zabbix = Zabbix(self.url)
+        if not self.__zabbix.auth:
+            self.__zabbix.login(self.user, self.password)
+        return self.__zabbix
+
+    def kill(self):
+        if self.__zabbix:
+            self.__zabbix.logout()
+
+    def zbx_triggers(self):
+        try:
+            if not self._zabbix.zabbix_status()['result']:
+                summary = 'DOWN'
+                color = self.py3.COLOR_BAD
+            elif self._zabbix.get_triggers():
+                summary = 'PROBLEM'
+                color = self.py3.COLOR_BAD
+            else:
+                summary = 'OK'
+                color = self.py3.COLOR_GOOD
+        except:
+            traceback.print_exc()
+            summary = 'ERR'
+            color = self.py3.COLOR_BAD
+        return {
+            'full_text': self.py3.safe_format(self.format, {
+                'summary': summary,
+            }),
+            'color': color,
+        }
+
+    def on_click(self, event):
+        if event['button'] != 1:
+            return
+        if os.fork():
+            return
+        with open(os.devnull, 'r+') as f:
+            os.dup2(f.fileno(), sys.stdout.fileno())
+        webbrowser.open(self.url)
+
+
+def main():
+    z = Zabbix('https://zabbix.securepassage.com/zabbix/')
+    #z = Zabbix('http://localhost:8080')
+    z.login()
+    try:
+        print(z.get_triggers())
+    finally:
+        z.logout()
+
+
+if __name__ == '__main__':
+    main()