diff --git a/src/Python/create_template.py b/src/Python/create_template.py index a9e8753..95e9a1f 100755 --- a/src/Python/create_template.py +++ b/src/Python/create_template.py @@ -10,7 +10,7 @@ import inspect def create_problem(): with open(Problem, "w+") as f: - template = inspect.cleandoc(f'''#!/usr/bin/python3 + template = inspect.cleandoc(f'''#!/usr/bin/env python3 """ Created on {today} @@ -21,7 +21,10 @@ def create_problem(): https://projecteuler.net/problem={args['problem']} """ + from utils import timeit + + @timeit("Problem {args['problem']}") def compute(): """ # Statement @@ -32,7 +35,7 @@ def create_problem(): if __name__ == "__main__": - print("Result for problem {args['problem']}: ", compute()) + print(f"Result for problem {args['problem']}: {{compute()}}") ''') f.write(template) @@ -40,7 +43,7 @@ def create_problem(): if __name__ == '__main__': - today = datetime.datetime.now().strftime("%a %b %d %H:%M:%S %Y") + today = datetime.datetime.now().strftime("%d %b %Y") parser = ArgumentParser(description=__doc__) # Add your arguments here parser.add_argument("-p", "--problem",