refactor: clean up range_type

This commit is contained in:
2024-12-30 20:33:44 -05:00
parent 1a71d83ebd
commit e032d04157
3 changed files with 8 additions and 8 deletions

View File

@@ -35,8 +35,12 @@ class ArgumentParser:
def range_type(string, min=0, max=100):
value = int(string)
try:
value = int(string)
except ValueError:
raise argparse.ArgumentTypeError(f"value not a valid integer")
if min <= value <= max:
return value
else:
raise argparse.ArgumentTypeError("value not in range %s-%s" % (min, max))
raise argparse.ArgumentTypeError(f"value not in range {min}-{max}")