d3fc648c08
Dynamically generate the accessors for a theme's attributes. NB: This might be even nicer coming from a JSON rather than inside the code.
58 lines
1.8 KiB
Python
58 lines
1.8 KiB
Python
import os
|
|
import io
|
|
import json
|
|
|
|
import core.event
|
|
|
|
THEME_BASE_DIR=os.path.dirname(os.path.realpath(__file__))
|
|
PATHS=[
|
|
'.',
|
|
os.path.join(THEME_BASE_DIR, '../themes'),
|
|
os.path.expanduser('~/.config/bumblebee-status/themes'),
|
|
]
|
|
|
|
class Theme(object):
|
|
def __init__(self, name='default', iconset=None, raw_data=None):
|
|
self.name = name
|
|
self.__widget_count = 0
|
|
if raw_data:
|
|
self.__data = raw_data
|
|
else:
|
|
self.__data = self.load(name)
|
|
core.event.register('start', self.__start)
|
|
core.event.register('next-widget', self.__next_widget)
|
|
|
|
for attr, default in [
|
|
('fg', None), ('bg', None),
|
|
('default-separators', True),
|
|
('separator-block-width', 0),
|
|
('separator', None)
|
|
]:
|
|
setattr(self, attr.replace('-', '_'), lambda widget=None, default=default, attr=attr: self.__get(widget, attr, default))
|
|
|
|
def load(self, name):
|
|
for path in PATHS:
|
|
theme_file = os.path.join(path, '{}.json'.format(name))
|
|
if os.path.isfile(theme_file):
|
|
with io.open(theme_file, encoding='utf-8') as data:
|
|
return json.load(data)
|
|
raise RuntimeError('unable to find theme {}'.format(name))
|
|
|
|
def __start(self):
|
|
self.__widget_count = 0
|
|
|
|
def __next_widget(self):
|
|
self.__widget_count = self.__widget_count + 1
|
|
|
|
def __get(self, widget, key, default=None):
|
|
value = default
|
|
|
|
for option in ['defaults', 'cycle']:
|
|
if option in self.__data:
|
|
tmp = self.__data[option]
|
|
if isinstance(tmp, list):
|
|
tmp = tmp[self.__widget_count % len(tmp)]
|
|
value = tmp.get(key, value)
|
|
return value
|
|
|
|
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
|