129 lines
4.1 KiB
Python
129 lines
4.1 KiB
Python
# pylint: disable=C0111,R0903
|
|
|
|
'''Displays the current color temperature of redshift
|
|
|
|
Requires the following executable:
|
|
* redshift
|
|
|
|
Parameters:
|
|
* redshift.location : location provider, either of 'geoclue2' (default), \
|
|
'ipinfo' (requires the requests package), or 'manual'
|
|
* redshift.lat : latitude if location is set to 'manual'
|
|
* redshift.lon : longitude if location is set to 'manual'
|
|
'''
|
|
|
|
import threading
|
|
try:
|
|
import requests
|
|
except ImportError:
|
|
pass
|
|
|
|
import bumblebee.input
|
|
import bumblebee.output
|
|
import bumblebee.engine
|
|
|
|
|
|
def is_terminated():
|
|
for thread in threading.enumerate():
|
|
if thread.name == 'MainThread' and not thread.is_alive():
|
|
return True
|
|
return False
|
|
|
|
|
|
def get_redshift_value(widget, location, lat, lon):
|
|
while True:
|
|
if is_terminated():
|
|
return
|
|
widget.get('condition').acquire()
|
|
while True:
|
|
try:
|
|
widget.get('condition').wait(1)
|
|
except RuntimeError:
|
|
continue
|
|
break
|
|
widget.get('condition').release()
|
|
|
|
command = ['redshift', '-p', '-l']
|
|
if location == 'manual':
|
|
command.append(lat + ':' + lon)
|
|
else:
|
|
command.append('geoclue2')
|
|
|
|
try:
|
|
res = bumblebee.util.execute(' '.join(command))
|
|
except Exception:
|
|
res = ''
|
|
widget.set('temp', 'n/a')
|
|
widget.set('transition', None)
|
|
widget.set('state', 'day')
|
|
for line in res.split('\n'):
|
|
line = line.lower()
|
|
if 'temperature' in line:
|
|
widget.set('temp', line.split(' ')[2])
|
|
if 'period' in line:
|
|
state = line.split(' ')[1]
|
|
if 'day' in state:
|
|
widget.set('state', 'day')
|
|
elif 'night' in state:
|
|
widget.set('state', 'night')
|
|
else:
|
|
widget.set('state', 'transition')
|
|
widget.set('transition', ' '.join(line.split(' ')[2:]))
|
|
|
|
|
|
class Module(bumblebee.engine.Module):
|
|
def __init__(self, engine, config):
|
|
widget = bumblebee.output.Widget(full_text=self.text)
|
|
super(Module, self).__init__(engine, config, widget)
|
|
|
|
self._location = self.parameter('location', 'geoclue2')
|
|
self._lat = self.parameter('lat', None)
|
|
self._lon = self.parameter('lon', None)
|
|
|
|
# Even if location method is set to manual, if we have no lat or lon,
|
|
# fall back to the geoclue2 method.
|
|
if self._location == 'manual' and (self._lat is None
|
|
or self._lon is None):
|
|
self._location == 'geoclue2'
|
|
|
|
if self._location == 'ipinfo':
|
|
try:
|
|
location_url = 'http://ipinfo.io/json'
|
|
location = requests.get(location_url).json()
|
|
self._lat, self._lon = location['loc'].split(',')
|
|
self._lat = str(float(self._lat))
|
|
self._lon = str(float(self._lon))
|
|
self._location = 'manual'
|
|
except Exception:
|
|
# Fall back to geoclue2.
|
|
self._location = 'geoclue2'
|
|
|
|
self._text = ''
|
|
self._condition = threading.Condition()
|
|
widget.set('condition', self._condition)
|
|
self._thread = threading.Thread(target=get_redshift_value,
|
|
args=(widget, self._location,
|
|
self._lat, self._lon))
|
|
self._thread.start()
|
|
self._condition.acquire()
|
|
self._condition.notify()
|
|
self._condition.release()
|
|
|
|
def text(self, widget):
|
|
return '{}'.format(self._text)
|
|
|
|
def update(self, widgets):
|
|
widget = widgets[0]
|
|
self._condition.acquire()
|
|
self._condition.notify()
|
|
self._condition.release()
|
|
temp = widget.get('temp', 'n/a')
|
|
self._text = temp
|
|
transition = widget.get('transition', None)
|
|
if transition:
|
|
self._text = '{} {}'.format(temp, transition)
|
|
|
|
def state(self, widget):
|
|
return widget.get('state', None)
|
|
|
|
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
|