85 lines
3.3 KiB
Python
85 lines
3.3 KiB
Python
import re
|
|
|
|
EMAIL_VALIDATOR = re.compile(r'([A-Za-z0-9]+[.-_])*[A-Za-z0-9]+@[A-Za-z0-9-]+(\.[A-Z|a-z]{2,})+')
|
|
|
|
class InputMenu:
|
|
|
|
def __init__(self, title):
|
|
self._title = title
|
|
self._fields = []
|
|
|
|
def add_option(self, key, title, type, value, min, max, validator):
|
|
self._fields.append({"key": key, "title": title, "type": type, "value": value, "min": min, "max": max, "validator": validator})
|
|
return self
|
|
|
|
def do_input(self):
|
|
print("")
|
|
print(f"/--[ {self._title} ]-------------")
|
|
|
|
for i in range(len(self._fields)):
|
|
data = None
|
|
|
|
while True:
|
|
data = ""
|
|
|
|
if self._fields[i]['value'] == None:
|
|
data = input(f"| ({self._fields[i]['title']}) => ")
|
|
else:
|
|
data = input(f"| ({self._fields[i]['title']}) [{self._fields[i]['value']}] => ")
|
|
|
|
if self._fields[i]['type'] == "STR":
|
|
|
|
# Check if value was preset and entered value was not changed
|
|
if self._fields[i]['value'] != None and data == "":
|
|
break
|
|
|
|
# Check length
|
|
if len(data) >= self._fields[i]['min'] and len(data) <= self._fields[i]['max']:
|
|
|
|
# Check regex validator id used
|
|
if self._fields[i]['validator'] != None:
|
|
if re.fullmatch(self._fields[i]['validator'], data):
|
|
# Set value of current field
|
|
self._fields[i]['value'] = data
|
|
break
|
|
else:
|
|
print("| Invalid input! Try again. \n|")
|
|
continue
|
|
|
|
# Set value of current field
|
|
self._fields[i]['value'] = data
|
|
break
|
|
|
|
|
|
print("| Invalid input! Try again. \n|")
|
|
continue
|
|
|
|
elif self._fields[i]['type'] == "INT":
|
|
|
|
# Check if value was preset and entered value was not changed
|
|
if self._fields[i]['value'] != None and data == "":
|
|
break
|
|
|
|
# Try and parse string into int, set value if no errors occur
|
|
try:
|
|
num = int(data)
|
|
if num >= self._fields[i]['min'] and num <= self._fields[i]['max']:
|
|
self._fields[i]['value'] = data
|
|
break
|
|
else:
|
|
print("| Invalid input! Try again. \n|")
|
|
except:
|
|
print("| Invalid input! Try again. \n|")
|
|
else:
|
|
exit("Invalid input type!")
|
|
|
|
print(f"\-------------------------------")
|
|
print("")
|
|
return self
|
|
|
|
def get_value(self, key):
|
|
for i in range(len(self._fields)):
|
|
if self._fields[i]["key"] == key:
|
|
return self._fields[i]["value"]
|
|
return None
|