extern/pywal
1
0
Fork 0
mirror of https://github.com/dylanaraps/pywal.git synced 2025-08-16 01:07:49 +02:00
Code Issues Packages Projects Releases Wiki Activity

backend: dynamic fo shizle

Browse Source
This commit is contained in:
Dylan Araps
2018-03-31 13:15:08 +11:00
parent 26677bb64b
commit 277edbc5e0
1 changed files with 6 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
pywal/colors.py
View File

@ -1,27 +1,18 @@
""" """
Generate a palette using various backends. Generate a palette using various backends.
""" """
import re
import os import os
import re
import sys
from . import backends from . import backends
from . import util from . import util
from .settings import CACHE_DIR, __cache_version__ from .settings import CACHE_DIR, __cache_version__
def get(backend_type="wal"):
"""Get backend function name from name."""
return {
"colorthief": backends.colorthief.get,
"colorz": backends.colorz.get,
"wal": backends.wal.get,
}.get(backend_type, backends.wal.get)
def list_backends(): def list_backends():
"""List color backends.""" """List color backends."""
# TODO: Dynamically generate this. return [backend for backend in dir(backends) if "__" not in backend]
return "colorthief, colorz, wal"
def colors_to_dict(colors, img): def colors_to_dict(colors, img):
@ -58,8 +49,9 @@ def gen(img, light=False, backend="wal", cache_dir=CACHE_DIR):
else: else:
print("wal: Generating a colorscheme...") print("wal: Generating a colorscheme...")
colors = get(backend)(img, light) # Dynamic shiz.
colors = colors_to_dict(colors, img) backend = sys.modules["pywal.backends.%s" % backend]
colors = colors_to_dict(getattr(backend, "get")(img, light), img)
util.save_file_json(colors, cache_file) util.save_file_json(colors, cache_file)
print("wal: Generation complete.") print("wal: Generation complete.")