mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-07-06 21:12:07 -07:00
54 lines
No EOL
1.7 KiB
Python
54 lines
No EOL
1.7 KiB
Python
|
|
|
|
def input_int(s, default_value, valid_list=None, help_message=None):
|
|
while True:
|
|
try:
|
|
inp = input(s)
|
|
if len(inp) == 0:
|
|
raise ValueError("")
|
|
|
|
if help_message is not None and inp == '?':
|
|
print (help_message)
|
|
continue
|
|
|
|
i = int(inp)
|
|
if (valid_list is not None) and (i not in valid_list):
|
|
return default_value
|
|
return i
|
|
except:
|
|
print (default_value)
|
|
return default_value
|
|
|
|
def input_bool(s, default_value, help_message=None):
|
|
while True:
|
|
try:
|
|
inp = input(s)
|
|
if len(inp) == 0:
|
|
raise ValueError("")
|
|
|
|
if help_message is not None and inp == '?':
|
|
print (help_message)
|
|
continue
|
|
|
|
return bool ( {"y":True,"n":False,"1":True,"0":False}.get(inp.lower(), default_value) )
|
|
except:
|
|
print ( "y" if default_value else "n" )
|
|
return default_value
|
|
|
|
def input_str(s, default_value, valid_list=None, help_message=None):
|
|
while True:
|
|
try:
|
|
inp = input(s)
|
|
if len(inp) == 0:
|
|
raise ValueError("")
|
|
|
|
if help_message is not None and inp == '?':
|
|
print (help_message)
|
|
continue
|
|
|
|
if (valid_list is not None) and (inp.lower() not in valid_list):
|
|
return default_value
|
|
return inp
|
|
except:
|
|
print (default_value)
|
|
return default_value |