input.py

"""Helper methods for getting user input"""


def get_bool_input(message):
    """Gets boolean input from the terminal"""
    values = {'y': True, 'yes': True, 'n': False, 'no': False, '': False}
    return get_dict_input(message, values)


def get_dict_input(message, values):
    """Selects value from a dictionary by user input"""
    input_value = validate_input_of_values(message=message, valid_values=set(values.keys()))
    return values[input_value]


def get_natural_input(message):
    """Gets an integer input from the terminal"""
    return int(validate_input_of_predicate(
        message=message,
        condition=lambda x: x.isdigit() and int(x) > 0,
        failure_message='Please enter a positive integer.'
    ))


def validate_input_of_predicate(message, condition, failure_message):
    """Applies a condition to input to check it"""
    text = input(message)
    while not condition(text):
        text = input(failure_message)
    return text


def validate_input_of_values(message, valid_values):
    """Checks whether an input is in a set of valid inputs"""
    text = input(message).lower()
    while text not in valid_values:
        text = input(f'Valid inputs: {valid_values}')
    return text