68 lines
2.5 KiB
Python
68 lines
2.5 KiB
Python
# pylint: disable=C0111,R0903
|
|
|
|
"""Displays the pi-hole status (up/down) together with the number of ads that were blocked today
|
|
Parameters:
|
|
* pihole.address : pi-hole address (e.q: http://192.168.1.3)
|
|
* pihole.pwhash : pi-hole webinterface password hash (can be obtained from the /etc/pihole/SetupVars.conf file)
|
|
"""
|
|
|
|
import requests
|
|
|
|
import core.module
|
|
import core.widget
|
|
import core.input
|
|
|
|
class Module(core.module.Module):
|
|
@core.decorators.every(minutes=1)
|
|
def __init__(self, config):
|
|
super().__init__(config, core.widget.Widget(self.pihole_status))
|
|
|
|
self._pihole_address = self.parameter('address', '')
|
|
self._pihole_pw_hash = self.parameter('pwhash', '')
|
|
self._pihole_status = None
|
|
self._ads_blocked_today = '-'
|
|
self.update_pihole_status()
|
|
|
|
core.input.register(self, button=core.input.LEFT_MOUSE,
|
|
cmd=self.toggle_pihole_status)
|
|
|
|
def pihole_status(self, widget):
|
|
if self._pihole_status is None:
|
|
return 'pi-hole unknown'
|
|
return 'pi-hole {}'.format('up {} blocked'.format(self._ads_blocked_today) if self._pihole_status else 'down')
|
|
|
|
def update_pihole_status(self):
|
|
try:
|
|
data = requests.get(self._pihole_address + '/admin/api.php?summary').json()
|
|
self._pihole_status = True if data['status'] == 'enabled' else False
|
|
self._ads_blocked_today = data['ads_blocked_today']
|
|
except Exception as e:
|
|
self._pihole_status = None
|
|
|
|
def toggle_pihole_status(self, widget):
|
|
if self._pihole_status is not None:
|
|
try:
|
|
req = None
|
|
if self._pihole_status:
|
|
req = requests.get(self._pihole_address + '/admin/api.php?disable&auth=' + self._pihole_pw_hash)
|
|
else:
|
|
req = requests.get(self._pihole_address + '/admin/api.php?enable&auth=' + self._pihole_pw_hash)
|
|
if req is not None:
|
|
if req.status_code == 200:
|
|
status = req.json()['status']
|
|
self._pihole_status = False if status == 'disabled' else True
|
|
except:
|
|
pass
|
|
|
|
|
|
def update(self):
|
|
self.update_pihole_status()
|
|
|
|
def state(self, widget):
|
|
if self._pihole_status is None:
|
|
return []
|
|
elif self._pihole_status:
|
|
return ['enabled']
|
|
return ['disabled', 'warning']
|
|
|
|
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
|