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

[modules/spaceapi] refactor, add comments, use interval function

Browse source 
Signed-off-by: Tobias Manske <tobias.manske@mailbox.org>
This commit is contained in:
Tobias Manske 2018-10-05 21:37:17 +02:00
parent 8fb0c2fa70
commit defdc7f077
No known key found for this signature in database
GPG key ID: 978D99F12D4E041F
1 changed files with 29 additions and 30 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

59
bumblebee/modules/spaceapi.py
View file

@ -4,14 +4,12 @@
Requires the following libraries: Requires the following libraries:
* requests * requests
* json
* time
Parameters: Parameters:
* spaceapi.url: String representation of the api endpoint * spaceapi.url: String representation of the api endpoint
* spaceapi.name: String overwriting the space name * spaceapi.name: String overwriting the space name
* spaceapi.prefix: Prefix for the space string * spaceapi.prefix: Prefix for the space string
* spaceapi.interval: time between updates * spaceapi.interval: time between updates in minutes
""" """
import bumblebee.input import bumblebee.input
@ -19,8 +17,6 @@ import bumblebee.output
import bumblebee.engine import bumblebee.engine
import requests import requests
import json
import time
class Module(bumblebee.engine.Module): class Module(bumblebee.engine.Module):
@ -28,48 +24,51 @@ class Module(bumblebee.engine.Module):
super(Module, self).__init__( super(Module, self).__init__(
engine, config, bumblebee.output.Widget(full_text=self.getState) engine, config, bumblebee.output.Widget(full_text=self.getState)
) )
self._state = False
# Represents the state of the hackerspace
self._open = False
# Set to true if there was an error calling the spaceapi
self._error = False self._error = False
# The URL representing the api endpoint
self._url = self.parameter("url", self._url = self.parameter("url",
default="http://club.entropia.de/spaceapi") default="http://club.entropia.de/spaceapi")
# Space Name, can be set manually in case of multiple widgets,
# so you're able to distinguish
self._name = self.parameter("name", default="") self._name = self.parameter("name", default="")
self._lastQuery = 0
self._sleeptime = self.parameter("interval", default=300) # Only execute every 5 minutes by default
self.interval(self.parameter("interval", default=5))
def getState(self, widget): def getState(self, widget):
string = self.parameter("prefix", default="") text = self.parameter("prefix", default="")
string += self._name + ": " text += self._name + ": "
if self._error: if self._error:
string += "ERROR" text += "ERROR"
elif self._state: elif self._open:
string += "Open" text += "Open"
else: else:
string += "Closed" text += "Closed"
return string return text
def state(self, widget): def state(self, widget):
if self._error: if self._error:
return ["critical"] return ["critical"]
elif self._state: elif self._open:
return ["warning"] return ["warning"]
else: else:
return [] return []
def update(self, widgets): def update(self, widgets):
unixtime = time.mktime(time.gmtime()) try:
# Only query again after interval has passed with requests.get(self._url) as u:
if self._lastQuery + self._sleeptime < int(unixtime): json = u.json()
self._lastQuery = int(unixtime) self._open = json["state"]["open"]
try: self._name = self.parameter("name", default=json["space"])
self._error = False
with requests.get(self._url) as u: except Exception:
data = u.json() # Displays ERROR status
self._state = data["state"]["open"] self._error = True
self._name = self.parameter("name", default=data["space"])
self._error = False
except Exception:
# Displays ERROR status
self._error = True
# Author: Tobias Manske <tobias.manske@mailbox.org> # Author: Tobias Manske <tobias.manske@mailbox.org>