l.behm/bumblebee-status
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

[modules/pihole] quotes

Browse source
This commit is contained in:
tobi-wan-kenobi 2020-04-12 14:05:55 +02:00
parent 5f2455ac79
commit cf16cecfec
1 changed files with 15 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
modules/contrib/pihole.py
View file

@ -17,12 +17,12 @@ class Module(bumblebee.engine.Module):
bumblebee.output.Widget(full_text=self.pihole_status) bumblebee.output.Widget(full_text=self.pihole_status)
) )
buttons = {"LEFT_CLICK":bumblebee.input.LEFT_MOUSE} buttons = {'LEFT_CLICK':bumblebee.input.LEFT_MOUSE}
self._pihole_address = self.parameter("address", "") self._pihole_address = self.parameter('address', '')
self._pihole_pw_hash = self.parameter("pwhash", "") self._pihole_pw_hash = self.parameter('pwhash', '')
self._pihole_status = None self._pihole_status = None
self._ads_blocked_today = "-" self._ads_blocked_today = '-'
self.update_pihole_status() self.update_pihole_status()
engine.input.register_callback(self, button=bumblebee.input.LEFT_MOUSE, engine.input.register_callback(self, button=bumblebee.input.LEFT_MOUSE,
@ -30,14 +30,14 @@ class Module(bumblebee.engine.Module):
def pihole_status(self, widget): def pihole_status(self, widget):
if self._pihole_status is None: if self._pihole_status is None:
return "pi-hole unknown" return 'pi-hole unknown'
return "pi-hole " + ("up/" + self._ads_blocked_today if self._pihole_status else "down") return 'pi-hole ' + ('up/' + self._ads_blocked_today if self._pihole_status else 'down')
def update_pihole_status(self): def update_pihole_status(self):
try: try:
data = requests.get(self._pihole_address + "/admin/api.php?summary").json() data = requests.get(self._pihole_address + '/admin/api.php?summary').json()
self._pihole_status = True if data["status"] == "enabled" else False self._pihole_status = True if data['status'] == 'enabled' else False
self._ads_blocked_today = data["ads_blocked_today"] self._ads_blocked_today = data['ads_blocked_today']
except: except:
self._pihole_status = None self._pihole_status = None
@ -46,13 +46,13 @@ class Module(bumblebee.engine.Module):
try: try:
req = None req = None
if self._pihole_status: if self._pihole_status:
req = requests.get(self._pihole_address + "/admin/api.php?disable&auth=" + self._pihole_pw_hash) req = requests.get(self._pihole_address + '/admin/api.php?disable&auth=' + self._pihole_pw_hash)
else: else:
req = requests.get(self._pihole_address + "/admin/api.php?enable&auth=" + self._pihole_pw_hash) req = requests.get(self._pihole_address + '/admin/api.php?enable&auth=' + self._pihole_pw_hash)
if req is not None: if req is not None:
if req.status_code == 200: if req.status_code == 200:
status = req.json()["status"] status = req.json()['status']
self._pihole_status = False if status == "disabled" else True self._pihole_status = False if status == 'disabled' else True
except: except:
pass pass
@ -64,5 +64,5 @@ class Module(bumblebee.engine.Module):
if self._pihole_status is None: if self._pihole_status is None:
return [] return []
elif self._pihole_status: elif self._pihole_status:
return ["enabled"] return ['enabled']
return ["disabled", "warning"] return ['disabled', 'warning']