2Shirt/WizardKit
2Shirt/WizardKit
1
0
Fork 0
Code Issues 2 Pull requests Projects Releases 26 Packages Wiki Activity Actions

Optionally disallow empty responses to input_text

Browse source
This commit is contained in:
2Shirt 2019-12-26 17:19:26 -07:00
parent c083a124ad
commit 20787da275
Signed by: 2Shirt
GPG key ID: 152FAC923B0E132C
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
scripts/wk/std.py
View file

@ -807,7 +807,7 @@ def get_log_filepath():
return log_filepath return log_filepath
def input_text(prompt='Enter text'): def input_text(prompt='Enter text', allow_empty_response=True):
"""Get text from user, returns string.""" """Get text from user, returns string."""
prompt = str(prompt) prompt = str(prompt)
response = None response = None
@ -823,6 +823,11 @@ def input_text(prompt='Enter text'):
# Ignore and try again # Ignore and try again
LOG.warning('Exception occured', exc_info=True) LOG.warning('Exception occured', exc_info=True)
print('', flush=True) print('', flush=True)
if not allow_empty_response:
if response is None or not response.strip():
# The None check here is used to avoid a TypeError if response is None
print(f'\r{prompt}', end='', flush=True)
response = None
return response return response