bumblebee-status/bumblebee/theme.py

import os
import json

def getpath():
    return os.path.dirname("{}/themes/".format(os.path.dirname(os.path.realpath(__file__))))

class Theme:
    _cycle_index = 0
    _cycle = None
    def __init__(self, name="default"):
        self._data = None
        path = os.path.dirname(os.path.realpath(__file__))
        with open("{}/{}.json".format(getpath(), name)) 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