Add decorator for timing

This commit is contained in:
David Doblas Jiménez 2021-06-03 19:37:19 +02:00
parent b7be53d5c5
commit 5ecda2c66c

View File

@ -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",