print()
if len(sys.argv) <= 1:
print("Please Enter the Problem Number: ", end = "")
t = input()
else:
t = sys.argv[1]
if re.match(r'^\s*\d+\s*$', t):
pn = int(t)
if pn == 0:
db = PEDB()
solved = sorted([int(x) for x in db.db.keys() if db.getProblem(x)['solved']])
print("There are " + str(len(solved)) + " problems solved")
t1 = time()
for m in solved:
ans, t = run(m)
db.record(m, solved=True, time=t)
t2 = time()
db.writeFile()
print("Finished all in " + str(int(t2-t1)) + 's')
else:
ans, t = run(pn)
print("Is it correct?(y/[n])", end = " ")
if (input()+' ')[0] in 'yY':
db = PEDB()
db.record(pn, solved=True, time=t, answer=ans)
db.writeFile()
else:
pbs = [f for f in os.listdir('.') if re.match(r'^pr\d{3}\.py$', f)]
pbs += [f for f in os.listdir('.') if re.match(r'^pr\d{3}$', f)]
pbs.sort(key=lambda x: os.stat(x).st_mtime, reverse=True)
problem = int(re.findall(r'[1-9]\d*', pbs[0])[0])
