accelegator.py

"""Accelegator: A software tool for accelerated and adaptive advising."""
# python libraries
import sys
# local dependencies
from queries import parse_csv_into_dataframe
from parse_arguments import parse_arguments
import display_strings
from defaults import DEFAULT_CSVFILE
from spreadsheet import create_csv
import repl
import route_commands
import write_to_file
import logging
import accelegator_NLP


if __name__ == '__main__':
    ARGUMENTS = parse_arguments(sys.argv[1:])
    print(display_strings.WELCOME)
    print(display_strings.URL)
    create_csv()
    FIRST_QUESTION = int(
        input("Enter number of first textual question to be analyzed: "))
    DATAFRAME = parse_csv_into_dataframe(DEFAULT_CSVFILE)
    arg1 = ""
    arg2 = ""
    arg3 = ""
    call = 0

    defined_commands = {
        "help",
        "gensim",
        "list",
        "show",
        "search",
        "write",
        "quit"}
    fSet = frozenset(defined_commands)
    args = []
    command = str(input('>>> (Type "help" for list of commands) '))  # will display only for first command
    while command != "quit":
        args = command.rsplit()
        command = args[0]
        while args[0] not in defined_commands:
            print("Invalid command. Type \"help\" to see list of valid commands.")
            command = str(input('>>> '))
            args = command.rsplit()
            command = args[0]
        if len(args) == 2:
            arg1 = args[1]
        elif len(args) == 3:
            arg1 = args[1]
            arg2 = args[2]
        elif len(args) == 4:
            arg1 = args[1]
            arg2 = args[2]
            arg3 = args[3]
        call = route_commands.route_commands(command, arg1, arg2)
        output = repl.repl(DATAFRAME, call, command, arg1, arg2)
        if (call == 6):
            inner_call = route_commands.route_commands(arg1, arg2, arg3)
            output = repl.repl(DATAFRAME, inner_call, arg1, arg2, arg3)
            file_name = str(input('File to write to: '))
            logging.info("Writing to file: " + file_name)
            write_to_file.write(output, file_name)
        elif(call == 9):
            accelegator_NLP.read_responses_person(
                DATAFRAME, FIRST_QUESTION, "all")
        elif(call == 10):
            accelegator_NLP.read_responses_person(
                DATAFRAME, FIRST_QUESTION, arg2)
        elif(call == 11):
            accelegator_NLP.read_responses_question(
                DATAFRAME, FIRST_QUESTION, "all")
        elif(call == 12):
            accelegator_NLP.read_responses_question(
                DATAFRAME, FIRST_QUESTION, int(arg2))
        elif(call == 13):
            accelegator_NLP.read_responses_all(DATAFRAME, FIRST_QUESTION)
        else:
            print(output)
        arg1 = ""
        arg2 = ""
        arg3 = ""
        command = str(input('>>> '))
    print()