Skip to content

Commit a43706f

Browse files
committed
Added exit prompt if no keys entered or exist
1 parent 5f76e39 commit a43706f

1 file changed

Lines changed: 9 additions & 1 deletion

File tree

remove-json-keys/src/remove_json_keys/__main__.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,13 +6,21 @@ def main():
66

77
if not cli.config.no_wizard:
88
while True: # prompt user for keys to remove
9+
910
if getattr(cli.config, 'keys', '') : print('\nCurrent keys to remove:', cli.config.keys)
1011
input_keys = input(
1112
f'\n{log.colors.bw}Enter key(s) to remove (comma-separated, or ENTER if done): {log.colors.nc}')
12-
if not input_keys : break
13+
14+
if not input_keys: # no keys entered
15+
if cli.config.keys : break # out of wizard
16+
user_resp = input('\nNo keys entered. Exit? (Y/n): ').lower()
17+
if user_resp == 'n' : continue # back to og prompt
18+
else : print(f'Exiting {cli.name}...') ; sys.exit(0)
19+
1320
new_keys = data.csv.parse(input_keys)
1421
existing_keys = set(cli.config.keys)
1522
truly_new_keys = []
23+
1624
for key in new_keys:
1725
if key not in existing_keys and key not in truly_new_keys:
1826
truly_new_keys.append(key)

0 commit comments

Comments
 (0)