util.log: add yes_or_no()

This commit is contained in:
Mort Yao 2018-10-17 22:26:55 +02:00
parent 251b80962a
commit f94c8d530d
No known key found for this signature in database
GPG Key ID: 07DA00CB78203251

View File

@ -96,3 +96,9 @@ def wtf(message, exit_code=1):
print_log(message, RED, BOLD) print_log(message, RED, BOLD)
if exit_code is not None: if exit_code is not None:
sys.exit(exit_code) sys.exit(exit_code)
def yes_or_no(message):
ans = str(input('%s (y/N) ' % message)).lower().strip()
if ans == 'y':
return True
return False