checked_input.py

from enum import Enum,auto

class Input:
 def __init__(self,prompt):
 self.prompt = prompt + " "
 self.ready = False
 self.value = None

 def run(self):
 self.value = input(self.prompt)
 self.ready = True
 
class NumberType(Enum):
 FLOAT = auto()
 INTEGER = auto()

class NumberInput(Input):
 def __init__(self,prompt,num_type=NumberType.FLOAT):
 self.type = num_type
 Input.__init__(self,prompt + " ")

 def run(self):
 self.value = input(self.prompt)
 input_is_num = False
 while(not input_is_num):
 try:
 if self.type == NumberType.FLOAT:
 self.value = float(self.value)
 else:
 self.value = int(self.value)
 input_is_num = True
 except:
 if self.type == NumberType.FLOAT:
 print("Your response could not be converted into a floating point number. Please try again!")
 else:
 print("Your response could not be converted into a integer number. Please try again!") 
 self.ready = True

class FlagInput(Input):
 def __init__(self,prompt,flags):
 self.flags = []
 for flag in flags:
 self.flags.append(flag)
 Input.__init__(self,prompt + " ")

 def run(self):
 is_flag = False
 while not is_flag:
 response = input(self.prompt)
 if response in self.flags:
 is_flag = True
 self.value = response
 self.ready = True
 else:
 print("That is not a valid response--please try one of the following flags: ",end = "")
 for num in range(len(self.flags)-1):
 print(self.flags[num],end = " ")
 print(self.flags[len(self.flags)-1])