refactered select_choice function for early return

This commit is contained in:
Rip&Tear
2024-10-13 18:09:33 +08:00
parent 1f9baf9b2c
commit 5d18c6312d

View File

@@ -69,12 +69,13 @@ def select_choice(prompt_message, choices):
except click.exceptions.Abort:
click.secho("Operation aborted by the user.", fg="red")
return None
if 0 <= selected_index < len(choices):
return choices[selected_index]
else:
if not (0 <= selected_index < len(choices)):
click.secho("Invalid selection.", fg="red")
return None
return choices[selected_index]
def select_provider(provider, all_providers, PROVIDERS):
if provider and provider.lower() not in all_providers and provider.lower() != 'other':
click.secho(f"Invalid provider: {provider}", fg="red")