diff --git a/leggen/commands/bank/add.py b/leggen/commands/bank/add.py index 0cf3c38..089dc91 100644 --- a/leggen/commands/bank/add.py +++ b/leggen/commands/bank/add.py @@ -14,7 +14,42 @@ def add(ctx): """ country = click.prompt( "Bank Country", - type=click.Choice(["PT", "GB"], case_sensitive=True), + type=click.Choice( + [ + "AT", + "BE", + "BG", + "HR", + "CY", + "CZ", + "DK", + "EE", + "FI", + "FR", + "DE", + "GR", + "HU", + "IS", + "IE", + "IT", + "LV", + "LI", + "LT", + "LU", + "MT", + "NL", + "NO", + "PL", + "PT", + "RO", + "SK", + "SI", + "ES", + "SE", + "GB", + ], + case_sensitive=True, + ), default="PT", ) info(f"Getting bank list for country: {country}")