7f91b8298f
Modules now get an output and a complete config object. This should make customization much easier in the future.
105 lines
3.2 KiB
Python
105 lines
3.2 KiB
Python
import os
|
|
import json
|
|
import glob
|
|
|
|
def getpath():
|
|
return os.path.dirname("{}/themes/".format(os.path.dirname(os.path.realpath(__file__))))
|
|
|
|
def themes():
|
|
d = getpath()
|
|
return [ os.path.basename(f).replace(".json", "") for f in glob.iglob("{}/*.json".format(d)) ]
|
|
|
|
class Theme:
|
|
_cycle_index = 0
|
|
_cycle = None
|
|
def __init__(self, config):
|
|
self._data = None
|
|
path = os.path.dirname(os.path.realpath(__file__))
|
|
with open("{}/{}.json".format(getpath(), config.theme())) as f:
|
|
self._data = json.load(f)
|
|
self._defaults = self._data.get("defaults", {})
|
|
self._cycle = self._defaults.get("cycle", [])
|
|
self._modules = {}
|
|
|
|
self.reset()
|
|
|
|
def _gettheme(self, obj, key):
|
|
module = obj.__module__.split(".")[-1]
|
|
module_theme = self._data.get(module, {})
|
|
|
|
value = getattr(obj, key)() if hasattr(obj, key) else None
|
|
value = self._defaults.get(key, value)
|
|
if len(self._cycle) > 0:
|
|
value = self._defaults["cycle"][self._cycle_index].get(key, value)
|
|
value = module_theme.get(key, value)
|
|
|
|
if hasattr(obj, "state"):
|
|
state = getattr(obj, "state")()
|
|
state_theme = module_theme.get("states", {}).get(state, {})
|
|
|
|
value = state_theme.get(key, value)
|
|
|
|
if type(value) is list:
|
|
# cycle through the values
|
|
if not obj in self._modules:
|
|
self._modules[obj] = { "idx": 0 }
|
|
else:
|
|
self._modules[obj]["idx"] += 1
|
|
if self._modules[obj]["idx"] >= len(value):
|
|
self._modules[obj]["idx"] = 0
|
|
value = value[self._modules[obj]["idx"]]
|
|
|
|
return value
|
|
|
|
def reset(self):
|
|
self._cycle_index = 0
|
|
self._previous_background = None
|
|
self._background = None
|
|
|
|
def urgent(self, obj):
|
|
self._gettheme(obj, "urgent")
|
|
|
|
def next(self):
|
|
self._cycle_index += 1
|
|
self._previous_background = self._background
|
|
if self._cycle_index >= len(self._cycle):
|
|
self._cycle_index = 0
|
|
|
|
def color(self, obj):
|
|
fg = None
|
|
if obj.warning():
|
|
fg = self._gettheme(obj, "fg-warning")
|
|
if obj.critical():
|
|
fg = self._gettheme(obj, "fg-critical")
|
|
if fg == None:
|
|
fg = self._gettheme(obj, "fg")
|
|
return fg
|
|
|
|
def background(self, obj):
|
|
self._background = None
|
|
if obj.warning():
|
|
self._background = self._gettheme(obj, "bg-warning")
|
|
if obj.critical():
|
|
self._background = self._gettheme(obj, "bg-critical")
|
|
|
|
if self._background == None:
|
|
self._background = self._gettheme(obj, "bg")
|
|
|
|
return self._background
|
|
|
|
def previous_background(self):
|
|
return self._previous_background
|
|
|
|
def separator(self, obj):
|
|
return self._gettheme(obj, "separator")
|
|
|
|
def default_separators(self, obj):
|
|
return self._gettheme(obj, "default_separators")
|
|
|
|
def prefix(self, obj):
|
|
return self._gettheme(obj, "prefix")
|
|
|
|
def suffix(self, obj):
|
|
return self._gettheme(obj, "suffix")
|
|
|
|
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
|