DeepFaceLab/utils/console_utils.py
2019-01-02 18:38:11 +04:00

26 lines
No EOL
717 B
Python

def input_int(s, default_value, valid_list=None):
try:
inp = input(s)
i = int(inp)
if (valid_list is not None) and (i not in valid_list):
return default_value
return i
except:
return default_value
def input_bool(s, default_value):
try:
return bool ( {"y":True,"n":False,"1":True,"0":False}.get(input(s).lower(), default_value) )
except:
return default_value
def input_str(s, default_value, valid_list=None):
try:
inp = input(s)
if (valid_list is not None) and (inp.lower() not in valid_list):
return default_value
return inp
except:
return default_value