From 1ad1ec51ad6c1b70ef38a37a5b72b62290f3de37 Mon Sep 17 00:00:00 2001 From: elvis kahoro Date: Thu, 23 Dec 2021 11:57:22 -0500 Subject: [PATCH] scripts --- scripts/gen_theme_previews.py | 37 +++++++++++++++++++---------------- scripts/preview.svg | 3 +-- 2 files changed, 21 insertions(+), 19 deletions(-) diff --git a/scripts/gen_theme_previews.py b/scripts/gen_theme_previews.py index 11cb759..3b3847d 100644 --- a/scripts/gen_theme_previews.py +++ b/scripts/gen_theme_previews.py @@ -7,8 +7,7 @@ from pathlib import Path def get_all_input_files(input_dir: str) -> List[str]: filenames = next(os.walk(input_dir), (None, None, []))[2] - files = filter(lambda f: (f.endswith("yaml") - or f.endswith("yml")), filenames) + files = filter(lambda f: (f.endswith("yaml") or f.endswith("yml")), filenames) return list(files) @@ -17,14 +16,16 @@ def ensure_output_dir(output_dir: str): os.makedirs(output_dir) -def add_color_to_dict(output: Dict[str, str], obj: Dict[str, str], key: str, prefix: Optional[str] = None): +def add_color_to_dict( + output: Dict[str, str], obj: Dict[str, str], key: str, prefix: Optional[str] = None +): if not prefix: prefix = "" output[f"{prefix}{key}"] = obj[key] def get_color_dict(input_dir: str, file_name: str) -> Dict[str, str]: - file = open(os.path.join(input_dir, file_name), 'r') + file = open(os.path.join(input_dir, file_name), "r") loaded_theme = yaml.safe_load(file) output = {} add_color_to_dict(output, loaded_theme, "accent") @@ -43,7 +44,7 @@ def get_color_dict(input_dir: str, file_name: str) -> Dict[str, str]: def file_name_to_display(file_name: str) -> str: - file_name = Path(file_name).with_suffix('').name + file_name = Path(file_name).with_suffix("").name split = file_name.split("_") output = [] @@ -62,14 +63,16 @@ def gen_svg_for_theme(color_dict: Dict[str, str], svg_template: str) -> str: def main(): parser = argparse.ArgumentParser( - description='Generate README.md with embedded SVG previews.') - parser.add_argument('input_dir', type=str, - help='Directory from which to read in all Warp themes.') - parser.add_argument('output_dir', type=str, help='Where to save README.md') - parser.add_argument("svg_path", type=str, - help="Path to svg template file.") - parser.add_argument("intro_file", type=str, - help="What should go on top of README.md.") + description="Generate README.md with embedded SVG previews." + ) + parser.add_argument( + "input_dir", type=str, help="Directory from which to read in all Warp themes." + ) + parser.add_argument("output_dir", type=str, help="Where to save README.md") + parser.add_argument("svg_path", type=str, help="Path to svg template file.") + parser.add_argument( + "intro_file", type=str, help="What should go on top of README.md." + ) args = parser.parse_args() ensure_output_dir(args.output_dir) @@ -78,11 +81,11 @@ def main(): markdown = [] markdown.append("|Theme name | Preview|") markdown.append("| --- | --- |") - svg = open(args.svg_path, 'r').read() + svg = open(args.svg_path, "r").read() svg_dir = os.path.join(args.output_dir, "previews") os.makedirs(svg_dir, exist_ok=True) - intro = open(args.intro_file, 'r').read() + intro = open(args.intro_file, "r").read() for input_file in sorted(filenames): print(f"Generating for {input_file}") @@ -90,13 +93,13 @@ def main(): color_dict = get_color_dict(args.input_dir, input_file) theme_svg = gen_svg_for_theme(color_dict, svg) theme_svg_path = os.path.join(svg_dir, f"{input_file}.svg") - with open(theme_svg_path, 'w') as svg_out: + with open(theme_svg_path, "w") as svg_out: svg_out.write(theme_svg) cell += f"|" markdown.append(cell) output_str = intro + "\n".join(markdown) - with open(os.path.join(args.output_dir, "README.md"), 'w') as output: + with open(os.path.join(args.output_dir, "README.md"), "w") as output: output.write(output_str) diff --git a/scripts/preview.svg b/scripts/preview.svg index 213ef1d..4f30d3b 100644 --- a/scripts/preview.svg +++ b/scripts/preview.svg @@ -1,5 +1,5 @@ - + ls @@ -88,4 +88,3 @@ git( -