Fix problem number

This commit is contained in:
David Doblas Jiménez 2021-10-30 18:57:07 +02:00
parent 7190721c5f
commit 022ae034d6

View File

@ -10,7 +10,8 @@ import inspect
def create_problem():
with open(Problem, "w+") as f:
template = inspect.cleandoc(f'''#!/usr/bin/env python3
template = inspect.cleandoc(
f'''#!/usr/bin/env python3
"""
Created on {today}
@ -35,20 +36,19 @@ def create_problem():
if __name__ == "__main__":
print(f"Result for Problem {{int({args['problem']}):003d}}: {{compute()}}")
''')
print(f"Result for Problem {args['problem']}: {{compute()}}")
'''
)
f.write(template)
if __name__ == '__main__':
if __name__ == "__main__":
today = datetime.datetime.now().strftime("%d %b %Y")
parser = ArgumentParser(description=__doc__)
# Add your arguments here
parser.add_argument("-p", "--problem",
help="number of the problem to solve")
parser.add_argument("-p", "--problem", help="number of the problem to solve")
args = vars(parser.parse_args())
Problem = f"Problem{(args['problem']):0>3}.py"
create_problem()