From 9b6be86ccd2fdf28a013808253b1344d08c4ef77 Mon Sep 17 00:00:00 2001
From: cieplik <cieplik@warp.dev>
Date: Tue, 12 Oct 2021 15:27:58 -0700
Subject: [PATCH] Lint + reqs (#7)

---
 scripts/gen_theme_previews.py | 29 +++++++++++++++++++++--------
 scripts/requirements.txt      |  1 +
 2 files changed, 22 insertions(+), 8 deletions(-)
 create mode 100644 scripts/requirements.txt

diff --git a/scripts/gen_theme_previews.py b/scripts/gen_theme_previews.py
index f296497..11cb759 100644
--- a/scripts/gen_theme_previews.py
+++ b/scripts/gen_theme_previews.py
@@ -4,20 +4,25 @@ import argparse
 import os
 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)
 
+
 def ensure_output_dir(output_dir: str):
     if not os.path.exists(output_dir):
         os.makedirs(output_dir)
 
+
 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')
     loaded_theme = yaml.safe_load(file)
@@ -36,6 +41,7 @@ def get_color_dict(input_dir: str, file_name: str) -> Dict[str, str]:
 
     return output
 
+
 def file_name_to_display(file_name: str) -> str:
     file_name = Path(file_name).with_suffix('').name
 
@@ -45,6 +51,7 @@ def file_name_to_display(file_name: str) -> str:
         output.append(s.capitalize())
     return " ".join(output)
 
+
 def gen_svg_for_theme(color_dict: Dict[str, str], svg_template: str) -> str:
     output = svg_template
 
@@ -52,12 +59,17 @@ def gen_svg_for_theme(color_dict: Dict[str, str], svg_template: str) -> str:
         output = output.replace(f"{{{key}}}", value)
     return output
 
+
 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 = 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.")
+    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)
@@ -68,7 +80,7 @@ def main():
     markdown.append("| --- | --- |")
     svg = open(args.svg_path, 'r').read()
     svg_dir = os.path.join(args.output_dir, "previews")
-    os.makedirs(svg_dir, exist_ok = True)
+    os.makedirs(svg_dir, exist_ok=True)
 
     intro = open(args.intro_file, 'r').read()
 
@@ -83,9 +95,10 @@ def main():
         cell += f"<img src='previews/{input_file}.svg' width='300'>|"
         markdown.append(cell)
 
-    output_str = intro +  "\n".join(markdown)
-    with open(os.path.join(args.output_dir,"README.md"), 'w') as output:
+    output_str = intro + "\n".join(markdown)
+    with open(os.path.join(args.output_dir, "README.md"), 'w') as output:
         output.write(output_str)
 
+
 if __name__ == "__main__":
     main()
diff --git a/scripts/requirements.txt b/scripts/requirements.txt
new file mode 100644
index 0000000..932bd69
--- /dev/null
+++ b/scripts/requirements.txt
@@ -0,0 +1 @@
+PyYAML==5.4.1