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

[modules/spaceapi] Update to latest API

Browse source
This commit is contained in:
tobi-wan-kenobi 2020-04-25 16:35:00 +02:00
parent 081186d97d
commit 50449bcee4
1 changed files with 35 additions and 37 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

72
modules/contrib/spaceapi.py
View file

@ -26,15 +26,15 @@ Format Strings:
would write '%%state.open%Open%Closed%%' would write '%%state.open%Open%Closed%%'
""" """
import bumblebee.input
import bumblebee.output
import bumblebee.engine
import requests import requests
import threading import threading
import re import re
import json import json
import core.module
import core.widget
import core.input
import core.decorators
def formatStringBuilder(s, json): def formatStringBuilder(s, json):
""" """
@ -61,75 +61,73 @@ def formatStringBuilder(s, json):
return s return s
class Module(bumblebee.engine.Module): class Module(core.module.Module):
def __init__(self, engine, config): @core.decorators.every(minutes=15)
super(Module, self).__init__( def __init__(self, config):
engine, config, bumblebee.output.Widget(full_text=self.getState) super().__init__(config, core.widget.Widget(self.getState))
core.input.register(
self, button=core.input.LEFT_MOUSE, cmd=self.__forceReload
) )
engine.input.register_callback( self.__data = {}
self, button=bumblebee.input.LEFT_MOUSE, cmd=self.__forceReload self.__error = None
) self.__thread = None
self._data = {}
self._error = None
self._threadingCount = 0
# The URL representing the api endpoint # The URL representing the api endpoint
self._url = self.parameter('url', default='http://club.entropia.de/spaceapi') self.__url = self.parameter('url', default='http://club.entropia.de/spaceapi')
self._format = self.parameter( self._format = self.parameter(
'format', default=u'%%space%%: %%state.open%Open%Closed%%' 'format', default=u'%%space%%: %%state.open%Open%Closed%%'
) )
def state(self, widget): def state(self, widget):
try: try:
if self._error is not None: if self.__error is not None:
return ['critical'] return ['critical']
elif self._data['state.open']: elif self.__data['state.open']:
return ['warning'] return ['warning']
else: else:
return [] return []
except KeyError: except KeyError:
return ['critical'] return ['critical']
def update(self, widgets): def update(self):
if self._threadingCount == 0: if not self.__thread or self.__thread.is_alive() == False:
thread = threading.Thread(target=self.get_api_async, args=()) self.__thread = threading.Thread(target=self.get_api_async, args=())
thread.start() self.__thread.start()
self._threadingCount = (
0 if self._threadingCount > 300 else self._threadingCount + 1
)
def getState(self, widget): def getState(self, widget):
text = self._format text = self._format
if self._error is not None: if self.__error is not None:
text = self._error text = self.__error
else: else:
try: try:
text = formatStringBuilder(self._format, self._data) text = formatStringBuilder(self._format, self.__data)
except KeyError: except KeyError:
text = 'KeyError' text = 'KeyError'
return text return text
def get_api_async(self): def get_api_async(self):
try: try:
with requests.get(self._url, timeout=10) as request: with requests.get(self.__url, timeout=10) as request:
# Can't implement error handling for python2.7 if I use # Can't implement error handling for python2.7 if I use
# request.json() as it uses simplejson in newer versions # request.json() as it uses simplejson in newer versions
self._data = self.__flatten(json.loads(request.text)) self.__data = self.__flatten(json.loads(request.text))
self._error = None self.__error = None
except requests.exceptions.Timeout: except requests.exceptions.Timeout:
self._error = 'Timeout' self.__error = 'Timeout'
except requests.exceptions.HTTPError: except requests.exceptions.HTTPError:
self._error = 'HTTP Error' self.__error = 'HTTP Error'
except ValueError: except ValueError:
self._error = 'Not a JSON response' self.__error = 'Not a JSON response'
core.event.trigger('update', [ self.id ], redraw_only=True)
# left_mouse_button handler # left_mouse_button handler
def __forceReload(self, event): def __forceReload(self, event):
self._threadingCount += 300 if self.__thread:
self._error = 'RELOADING' self.__thread.raise_exception()
self.__error = 'RELOADING'
core.event.trigger('update', [ self.id ], redraw_only=True)
# Flattens the JSON structure recursively, e.g. ['space']['open'] # Flattens the JSON structure recursively, e.g. ['space']['open']
# becomes ['space.open'] # becomes ['space.open']