aboutsummaryrefslogtreecommitdiff
path: root/generate
diff options
context:
space:
mode:
Diffstat (limited to 'generate')
-rwxr-xr-xgenerate10
1 files changed, 9 insertions, 1 deletions
diff --git a/generate b/generate
index 5773fd8..d4a40a2 100755
--- a/generate
+++ b/generate
@@ -71,6 +71,11 @@ def main():
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',
choices=languages,
@@ -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__':