mirror of
https://github.com/dylanaraps/pywal.git
synced 2025-05-01 14:44:26 +02:00
121 lines
3.1 KiB
Python
121 lines
3.1 KiB
Python
"""
|
|
'||
|
|
... ... .... ... ... ... ... .... ||
|
|
||' || '|. | || || | '' .|| ||
|
|
|| | '|.| ||| ||| .|' || ||
|
|
||...' '| | | '|..'|' .||.
|
|
|| .. |
|
|
'''' ''
|
|
Created by Dylan Araps.
|
|
"""
|
|
|
|
import argparse
|
|
import os
|
|
import shutil
|
|
import sys
|
|
|
|
from .settings import __version__, __cache_dir__
|
|
from . import colors
|
|
from . import image
|
|
from . import reload
|
|
from . import sequences
|
|
from . import template
|
|
from . import util
|
|
from . import wallpaper
|
|
|
|
|
|
def get_args():
|
|
"""Get the script arguments."""
|
|
description = "wal - Generate colorschemes on the fly"
|
|
arg = argparse.ArgumentParser(description=description)
|
|
|
|
arg.add_argument("-c", action="store_true",
|
|
help="Delete all cached colorschemes.")
|
|
|
|
arg.add_argument("-i", metavar="\"/path/to/img.jpg\"",
|
|
help="Which image or directory to use.")
|
|
|
|
arg.add_argument("-f", metavar="\"/path/to/colorscheme/file\"",
|
|
help="Which colorscheme file to use.")
|
|
|
|
arg.add_argument("-n", action="store_true",
|
|
help="Skip setting the wallpaper.")
|
|
|
|
arg.add_argument("-o", metavar="\"script_name\"",
|
|
help="External script to run after \"wal\".")
|
|
|
|
arg.add_argument("-q", action="store_true",
|
|
help="Quiet mode, don\"t print anything and \
|
|
don't display notifications.")
|
|
|
|
arg.add_argument("-r", action="store_true",
|
|
help="Reload current colorscheme.")
|
|
|
|
arg.add_argument("-t", action="store_true",
|
|
help="Fix artifacts in VTE Terminals. \
|
|
(Termite, xfce4-terminal)")
|
|
|
|
arg.add_argument("-v", action="store_true",
|
|
help="Print \"wal\" version.")
|
|
|
|
return arg.parse_args()
|
|
|
|
|
|
def process_args(args):
|
|
"""Process args."""
|
|
if not len(sys.argv) > 1:
|
|
print("error: wal needs to be given arguments to run.\n"
|
|
" Refer to \"wal -h\" for more info.")
|
|
exit(1)
|
|
|
|
if args.i and args.f:
|
|
print("error: Conflicting arguments -i and -f.\n"
|
|
" Refer to \"wal -h\" for more info.")
|
|
exit(1)
|
|
|
|
if args.v:
|
|
print(f"wal {__version__}")
|
|
exit(0)
|
|
|
|
if args.q:
|
|
sys.stdout = sys.stderr = open(os.devnull, "w")
|
|
|
|
if args.c:
|
|
shutil.rmtree(__cache_dir__ / "schemes", ignore_errors=True)
|
|
|
|
if args.r:
|
|
reload.colors(args.t)
|
|
|
|
if args.i:
|
|
image_file = image.get(args.i)
|
|
colors_plain = colors.get(image_file, notify=not args.q)
|
|
|
|
if args.f:
|
|
colors_plain = util.read_file_json(args.f)
|
|
|
|
if args.i or args.f:
|
|
sequences.send(colors_plain, args.t)
|
|
|
|
if not args.n:
|
|
wallpaper.change(colors_plain["wallpaper"])
|
|
|
|
template.export_all(colors_plain)
|
|
reload.env()
|
|
|
|
if args.o:
|
|
util.disown(args.o)
|
|
|
|
|
|
def main():
|
|
"""Main script function."""
|
|
args = get_args()
|
|
process_args(args)
|
|
|
|
# This saves 10ms.
|
|
# pylint: disable=W0212
|
|
# os._exit(0)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|