From 9ece4f582e260e621dcaa3e8cb976f1978b404ba Mon Sep 17 00:00:00 2001 From: InsanePrawn Date: Mon, 25 Oct 2021 18:19:14 +0200 Subject: [PATCH] cmd_boot(): add choices list to `type` arg --- boot.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/boot.py b/boot.py index 4e1b858..416717b 100644 --- a/boot.py +++ b/boot.py @@ -15,7 +15,7 @@ TYPES = [LK2ND, JUMPDRIVE, BOOTIMG] @click.command(name='boot') -@click.argument('type', required=False, default=BOOTIMG) +@click.argument('type', required=False, default=BOOTIMG, type=click.Choice(TYPES)) def cmd_boot(type): f"""Flash one of {', '.join(TYPES)}""" enforce_wrap()