custominput.py

from .responseoption import ResponseOption, option, optionList
import sys, re
sys.path.append("..")

def int_input(output, default=None):
    message = "{} ".format(output)
    if default:
        message += "[{}] ".format(default)

    while (True):
        user_input = input(message)
        if default and user_input == "":
            return default
        else:
            try:
                valid_input = int(user_input)
                return valid_input
            except:
                print("I'm sorry, I didn't understand that input.")

def confirm_input(output):
    valid_password = False

    while (not valid_password):
        first_password = input(output)
        second_password = input("Confirm your previous input by typing it " \
            "again please: ")

        while (second_password != first_password):
            second_password = input("Inputs don't match. Please re-enter " \
                "the original input: (Enter '!' to restart) ")
            if second_password == "!":
                break
        else:
            return first_password

def email_input(output, provider=None, multiples=False):
    message = "{} ".format(output)
    if multiples:
        message += "(To enter multiple addresses, seperate each with ', ') "
    
    valid_inputs = False
    valid_emails = []
    while (not valid_inputs):
        user_response = input(message)
        if multiples:
            emails_split = user_response.split(", ")
            for email in emails_split:
                if __validate_email(email):
                    valid_emails.append(email)
                else:
                    new_email = __email_error(email)
                    valid_emails.append(new_email)
            else:
                return valid_emails
        else:
            if __validate_email(user_response, provider):
                return user_response
            else:
                print("I'm sorry, {} isn't a valid email.".format(user_response))

def __validate_email(email, provider=None):
    regex = r'\b[A-Za-z0-9._%+-]+@[A-Za-z0-9.-]+\.[A-Z|a-z]{2,}\b'

    if (re.fullmatch(regex, email)):
        if provider:
            provider = provider + ".com"
            if provider in email:
                return True
        else:
            return True

    return False

def __email_error(email):
    email = email
    while (True):
        message = "I'm sorry, {} is not a valid email. Please try again: ".format(email)
        email = input(message)
        if __validate_email(email):
            return email

def bool_input(output, default=None, abrv=True):
    '''A function to get a user's response to a question that has a yes or no
    (True or False) answer.
    
    Keyword arguments:
        default -- This is the default response if the user does not input
    anything. This parameter can be either True, False, or None. If set to None,
    the function will not accept an empty answer. It will require one of the
    given responses. (Defaults to None)
        abrv -- Uses the abbreviated response. (Defaults to True)'''
    true_answers = ["y", "yes"]
    false_answers = ["n", "no"]
    used_index = 0 if abrv else 1
    ammendment = ""
    valid_answers = []
    if default is None:
        ammendment = "[{}/{}]".format(true_answers[used_index],
                                      false_answers[used_index])
        valid_answers = true_answers + false_answers
    elif default:
        ammendment = "[{}/{}]".format(true_answers[used_index].upper(),
                                      false_answers[used_index])
        valid_answers = true_answers + false_answers + [""]
    else:
        ammendment = "[{}/{}]".format(true_answers[used_index],
                                      false_answers[used_index].upper())
        valid_answers = true_answers + false_answers + [""]

    message = "{} {} ".format(output, ammendment)
    valid_input = False

    while (not valid_input):
        user_response = input(message)

        if user_response.lower() in true_answers:
            return True
        elif user_response.lower() in false_answers:
            return False
        elif user_response.lower() in valid_answers:
            return default

        print("I'm sorry, I didn't understand that input.")

def choice_input(output, options, default=None, abrv=True):
    ''''''
    option_list = optionList(options, default, abrv)
    message = "{} {} ".format(output, option_list)
    valid_input = False

    while (not valid_input):
        user_input = input(message)
        possible_option = option(options, user_input)

        if user_input == "" and default is not None:
            return default.response
        elif possible_option is None:
            print("I'm sorry, I didn't understand that input.")
        else:
            return possible_option.response

def range_input(output, lower, upper, default=None):
    ''''''
    message = "{} [{}-{}] ".format(output, lower, upper)
    user_input = None
    valid_input = False

    while (not valid_input):
        use_default = False
        try:
            user_input = input(message)

            if user_input == "":
                use_default = True

            user_input = int(user_input)
        except:
            pass

        if user_input in range(lower, upper + 1):
            return user_input
        elif use_default and default is not None:
            return default
        else:
            print("I'm sorry, I didn't understand that input.")

def time_input(output, default=None):
    regex = r'^([012])?\d:[0-5][0-9] (p|a|24)'
    ammendment = "[##:## a/p/24]"
    message = "{} {} ".format(output, ammendment)
    user_response = ""
    valid_input = False

    while (not valid_input):
        user_response = input(message)
        valid_input = re.fullmatch(regex, user_response)
    else:
        return user_response

def version_input(output):
    regex = r'\d+\.\d+(.\d+)?'
    ammendment = "[#.##.#/#.##]"
    message = "{} {} ".format(output, ammendment)
    user_response = None
    valid_input = False

    while (not valid_input):
        user_response = input(message)
        if user_response == "" or re.fullmatch(regex, user_response):
            valid_input = True
    else:
        return user_response

# def cron_date_input(output):
#     freq_output = "How frequently would you like your server to " \
#         "{}?".format(output)
#     week_day_output = "What day of the week would you like your server to " \
#         "{}?".format(output)
#     month_output = "What day of the month would ou like your server to " \
#         "{}?".format(output)
#     time_output = "What time of the day would you like your server to " \
#         "{}?".format(output)
#     freq = choice_input(freq_output, options=CronDate.FREQUENCY_OPTIONS,
#                         abrv=False)
#     week_day = None
#     month_day = None
#     if freq == CronFrequency.WEEKLY:
#         week_day = choice_input(week_day_output,
#                                 options=CronDate.WEEK_DAY_OPTIONS,
#                                 abrv=False)
#     elif freq == CronFrequency.MONTHLY:
#         month_day = range_input(month_output, lower=1, upper=28)

#     time = time_input(time_output)

#     cron_date = CronDate(freq, week_day, month_day, time)
#     return cron_date.convertToCronTime()