From 6a0cc649f0b970312d8b8b99071767846800c510 Mon Sep 17 00:00:00 2001
From: Charles Cabergs <me@cacharle.xyz>
Date: Mon, 21 Jun 2021 19:42:22 +0200
Subject: Added generate script --preview option

---
 generate | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

(limited to 'generate')

diff --git a/generate b/generate
index 5773fd8..d4a40a2 100755
--- a/generate
+++ b/generate
@@ -70,6 +70,11 @@ def main():
     with open(LANGUAGES_FILENAME, 'r') as file:
         languages = json.load(file)
     parser = ArgumentParser(description='Project Euler problem file generator')
+    parser.add_argument(
+        '-p', '--preview',
+        action='store_true',
+        help='see a preview of the file without creating it',
+    )
     parser.add_argument(
         'language',
         metavar='LANGUAGE',
@@ -87,7 +92,10 @@ def main():
     for index in args.indices:
         language = languages[args.language]
         language['name'] = args.language
-        Problem(index, language).fetch().write()
+        problem = Problem(index, language)
+        problem.fetch()
+        if not args.preview:
+            problem.write()
 
 
 if __name__ == '__main__':
-- 
cgit