extern/pywal
1
0
Fork 0
mirror of https://github.com/dylanaraps/pywal.git synced 2025-01-23 22:38:36 +01:00
Code Issues Packages Projects Releases Wiki Activity

General: Check if scheme exists

Browse Source
This commit is contained in:
Dylan Araps 2017-06-17 11:23:19 +10:00
parent 429e3c9f10
commit 23e01e5e8a
1 changed files with 28 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
wal.py
View File

@ -76,21 +76,10 @@ def get_image(img):
return rand_img
def get_colors(img):
"""Generate a colorscheme using imagemagick."""
def gen_colors(img):
"""Generate a color palette using imagemagick."""
colors = []
# Create colorscheme dir.
pathlib.Path(CACHE_DIR + "/schemes").mkdir(parents=True, exist_ok=True)
# Cache file.
cache_file = CACHE_DIR + "/schemes/" + img.replace('/', '_')
# Cache the wallpaper name.
wal = open(CACHE_DIR + "/wal", 'w')
wal.write(img + "\n")
wal.close()
# Long-ass imagemagick command.
magic = subprocess.Popen(["convert", img, "+dither", "-colors",
str(COLOR_COUNT), "-unique-colors", "txt:-"],
@ -107,17 +96,38 @@ def get_colors(img):
# Remove the first element, which isn't a color.
del colors[0]
# Cache the colorscheme.
scheme = open(cache_file, 'w')
for color in colors:
scheme.write(color + "\n")
scheme.close()
return colors
def get_colors(img):
"""Generate a colorscheme using imagemagick."""
# Cache file.
cache_file = Path(CACHE_DIR + "/schemes/" + img.replace('/', '_'))
if not cache_file.is_file():
# Cache the wallpaper name.
wal = open(CACHE_DIR + "/wal", 'w')
wal.write(img + "\n")
wal.close()
# Generate the colors.
colors = gen_colors(img)
# Cache the colorscheme.
scheme = open(cache_file, 'w')
for color in colors:
scheme.write(color + "\n")
scheme.close()
def main():
"""Main script function."""
args = get_args()
image = str(get_image(args.i))
# Create colorscheme dir.
pathlib.Path(CACHE_DIR + "/schemes").mkdir(parents=True, exist_ok=True)
get_colors(image)
return 0