refactor: clean up range_type
This commit is contained in:
parent
1a71d83ebd
commit
e032d04157
@ -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}")
|
||||
|
@ -1,7 +1,7 @@
|
||||
import functools
|
||||
import re
|
||||
|
||||
import arguments
|
||||
|
||||
import commands
|
||||
import utils
|
||||
|
||||
@ -14,9 +14,7 @@ async def clear(message):
|
||||
)
|
||||
parser.add_argument(
|
||||
"count",
|
||||
type=int,
|
||||
choices=range(1, 1001),
|
||||
metavar="[1-1000]",
|
||||
type=functools.partial(arguments.range_type, min=1, max=1000),
|
||||
help="amount of messages to delete",
|
||||
)
|
||||
parser.add_argument(
|
||||
|
@ -26,7 +26,6 @@ async def queue_or_play(message):
|
||||
"--volume",
|
||||
default=50,
|
||||
type=functools.partial(arguments.range_type, min=0, max=150),
|
||||
metavar="[0-150]",
|
||||
help="the volume level (0 - 150)",
|
||||
)
|
||||
group.add_argument(
|
||||
@ -228,7 +227,6 @@ async def volume(message):
|
||||
"volume",
|
||||
nargs="?",
|
||||
type=functools.partial(arguments.range_type, min=0, max=150),
|
||||
metavar="[0-150]",
|
||||
help="the volume level (0 - 150)",
|
||||
)
|
||||
if not (args := await parser.parse_args(message, tokens)):
|
||||
|
Loading…
x
Reference in New Issue
Block a user