using pkg_resources

This commit is contained in:
nginsburg 2020-02-04 19:06:26 -05:00
parent 3431129536
commit dcc42036e8
30 changed files with 24 additions and 22 deletions

1
.gitignore vendored
View file

@ -26,6 +26,7 @@ var/
*.egg-info/ *.egg-info/
.installed.cfg .installed.cfg
*.egg *.egg
build/
# PyInstaller # PyInstaller
# Usually these files are written by a python script from a template # Usually these files are written by a python script from a template

View file

@ -9,31 +9,30 @@ import json
import io import io
import re import re
import logging import logging
import pkg_resources
try:
import requests
from requests.exceptions import RequestException
except ImportError:
pass
import bumblebee.error import bumblebee.error
def theme_path(): def theme_path():
"""Return the path of the theme directory""" """Return the path of the theme directory"""
return [ return [
os.path.realpath(x) for x in [
os.path.dirname("{}/../themes/".format(os.path.dirname(os.path.realpath(__file__)))), os.path.dirname("{}/../themes/".format(os.path.dirname(os.path.realpath(__file__)))),
os.path.dirname("{}/../../../../share/bumblebee-status/themes/".format(os.path.dirname(os.path.realpath(__file__)))), os.path.dirname(
"{}/../../../../share/bumblebee-status/themes/".format(
os.path.dirname(os.path.realpath(__file__)))),
os.path.dirname(os.path.expanduser("~/.config/bumblebee-status/themes/")), os.path.dirname(os.path.expanduser("~/.config/bumblebee-status/themes/")),
pkg_resources.resource_filename('bumblebee', 'themes')] if os.path.exists(x)
] ]
def themes(): def themes():
themes = {} themes_dict = {}
for path in theme_path(): for path in theme_path():
for filename in glob.iglob("{}/*.json".format(path)): for filename in glob.iglob("{}/*.json".format(path)):
if "test" not in filename: if "test" not in filename:
themes[os.path.basename(filename).replace(".json", "")] = 1 themes_dict[os.path.basename(filename).replace(".json", "")] = 1
result = list(themes.keys()) result = list(themes_dict.keys())
result.sort() result.sort()
return result return result
@ -59,7 +58,7 @@ class Theme(object):
path = os.path.expanduser("~/.config/bumblebee-status/") path = os.path.expanduser("~/.config/bumblebee-status/")
try: try:
os.makedirs(path) os.makedirs(path)
except Exception: except OSError:
pass pass
try: try:
if os.path.exists("{}/symbols.json".format(path)): if os.path.exists("{}/symbols.json".format(path)):
@ -73,8 +72,8 @@ class Theme(object):
code = chr(code) code = chr(code)
self._symbols["${{{}}}".format(icon["id"])] = code self._symbols["${{{}}}".format(icon["id"])] = code
self._symbols["${{{}}}".format(icon["name"])] = code self._symbols["${{{}}}".format(icon["name"])] = code
except Exception as e: except Exception as err:
logging.error("failed to load symbols: {}".format(str(e))) logging.error("failed to load symbols: %s", err)
def _init(self, data): def _init(self, data):
"""Initialize theme from data structure""" """Initialize theme from data structure"""
@ -96,7 +95,7 @@ class Theme(object):
def reset(self): def reset(self):
"""Reset theme to initial state""" """Reset theme to initial state"""
self._cycle = self._cycles[0] if len(self._cycles) > 0 else {} self._cycle = self._cycles[0] if self._cycles else {}
self._cycle_idx = 0 self._cycle_idx = 0
self._widget = None self._widget = None
self._prevbg = None self._prevbg = None
@ -105,6 +104,7 @@ class Theme(object):
icon = self._get(widget, "icon", None) icon = self._get(widget, "icon", None)
if icon is None: if icon is None:
return self._get(widget, "prefix", None) return self._get(widget, "prefix", None)
return None
def get(self, widget, attribute, default_value=""): def get(self, widget, attribute, default_value=""):
return self._get(widget, attribute, default_value) return self._get(widget, attribute, default_value)
@ -189,10 +189,10 @@ class Theme(object):
def _load_colors(self, name): def _load_colors(self, name):
"""Load colors for a theme""" """Load colors for a theme"""
try: try:
if name == "wal": if name.lower() == "wal":
return self._load_wal_colors() return self._load_wal_colors()
except Exception as e: except Exception as err:
logging.error("failed to load colors: {}".format(str(e))) logging.error("failed to load colors: %s", err)
def _load_icons(self, name): def _load_icons(self, name):
"""Load icons for a theme""" """Load icons for a theme"""
@ -217,7 +217,7 @@ class Theme(object):
if os.path.isfile(full_name): if os.path.isfile(full_name):
path = os.path.dirname(full_name) path = os.path.dirname(full_name)
name = os.path.basename(full_name) name = os.path.basename(full_name)
name,_,_ = name.rpartition(".json") name, _, _ = name.rpartition(".json")
return self.load(name, path) return self.load(name, path)
if not isinstance(path, list): if not isinstance(path, list):
@ -249,7 +249,7 @@ class Theme(object):
if self._widget != widget: if self._widget != widget:
self._prevbg = self.bg(self._widget) self._prevbg = self.bg(self._widget)
self._widget = widget self._widget = widget
if len(self._cycles) > 0: if self._cycles:
self._cycle_idx = (self._cycle_idx + 1) % len(self._cycles) self._cycle_idx = (self._cycle_idx + 1) % len(self._cycles)
self._cycle = self._cycles[self._cycle_idx] self._cycle = self._cycles[self._cycle_idx]
@ -283,7 +283,7 @@ class Theme(object):
if mod and not mod.parameter("is-unittest"): if mod and not mod.parameter("is-unittest"):
value = widget.get_module().parameter("theme.{}".format(name), value) value = widget.get_module().parameter("theme.{}".format(name), value)
if isinstance(value, list) or isinstance(value, dict): if isinstance(value, (dict, list)):
return value return value
return self._colorset.get(value, value) return self._colorset.get(value, value)

1
themes Symbolic link
View file

@ -0,0 +1 @@
bumblebee/themes/