Minor usability improvements

This commit is contained in:
2022-09-15 10:47:12 +02:00
parent 5ab0d996d2
commit 3e478b8b52
3 changed files with 30 additions and 12 deletions

View File

@@ -1,14 +1,20 @@
from typing import Optional
def query(text):
return input(f"{text} \n> ")
def query_int(text):
def query_int(text: str, default: Optional[int] = None):
try:
ans = query(text)
return int(ans)
if ans or default is None:
return int(ans)
else:
return default
except ValueError:
print("Please enter a valid number!")
return query_int(text)
return query_int(text, default)
def query_yes_no(question, default="yes"):