extern/pywal
1
0
Fork 0
mirror of https://github.com/dylanaraps/pywal.git synced 2025-01-24 14:58:44 +01:00
Code Issues Packages Projects Releases Wiki Activity

backend: Import colorz instead of running it as a subprocess.

Browse Source
This commit is contained in:
Dylan Araps 2018-05-31 07:15:58 +10:00
parent 1ef4c742a2
commit 22a2618c21
1 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
pywal/backends/colorz.py
View File

@ -6,14 +6,22 @@ import shutil
import subprocess import subprocess
import sys import sys
try:
import colorz
except ImportError:
logging.error("colorz wasn't found on your system.")
logging.error("Try another backend. (wal --backend)")
sys.exit(1)
from .. import colors from .. import colors
from .. import util from .. import util
def gen_colors(img): def gen_colors(img):
"""Generate a colorscheme using Colorz.""" """Generate a colorscheme using Colorz."""
cmd = ["colorz", "-n", "6", "--bold", "0", "--no-preview", "--no-bg-img"] raw_colors = colorz.colorz(img, n=6, bold_add=0)
return subprocess.check_output([*cmd, img]).splitlines() return [util.rgb_to_hex([*color[0]]) for color in raw_colors]
def adjust(cols, light): def adjust(cols, light):
@ -28,10 +36,11 @@ def adjust(cols, light):
def get(img, light=False): def get(img, light=False):
"""Get colorscheme.""" """Get colorscheme."""
if not shutil.which("colorz"): cols = gen_colors(img)
logging.error("Colorz wasn't found on your system.")
logging.error("Try another backend. (wal --backend)") if len(cols) < 6:
logging.error("colorz failed to generate enough colors.")
logging.error("Try another backend or another image. (wal --backend)")
sys.exit(1) sys.exit(1)
cols = [col.decode('UTF-8').split()[0] for col in gen_colors(img)]
return adjust(cols, light) return adjust(cols, light)