cli_main.py

from src.rdf import Manager
import os

#Variables
filename = os.path.abspath("./output.rdf")
arrest_reports_url = "https://data.lacity.org/resource/amvf-fr72"
crime_reports_url = "https://data.lacity.org/resource/2nrs-mtv8"
max_data_count = 999999999
done = False
sucess_option_1 = False
sucess_option_2 = False
sucess_option_3 = False
sucess_option_4 = False

manager = None
query_result = None

#CLI logic
while (not done):

    #Print instructions
    print("Modeling RDF Data with LA Public Safety Data")

    print(" This program will: \n     \u2022 Download datasets for Arrest Reports and Crime Reports \n     \u2022 Generate RDF graph from both reports \n     \u2022 Save RDF graph to a file")

    print(" Parameters: \n     \u2022 Arrest Reports URL: %s \n     \u2022 Crime Reports URL: %s \n     \u2022 RDF Filename: %s \n     \u2022 Max Data Count to Download: %s" % (arrest_reports_url, crime_reports_url, filename, max_data_count))

    print(" Options: \n     1. Generate RDF file \n     2. Query \n     3. Modify filename \n     4. Modify max max data count \n     5. Exit")

    #User's input feedback
    if sucess_option_1:
        print("INFO: Successfully generate RDF file...")
        sucess_option_1=False
    elif sucess_option_2:
        print("INFO: Successfully query RDF graph")
        print(query_result)
        sucess_option_2=False
    elif sucess_option_3:
        print("INFO: Successfully modify filename")
        sucess_option_3=False
    elif sucess_option_4:
        print("INFO: Successfully modify max data count")
        sucess_option_4=False

    #Obtain user's input
    user_input = input("Enter an option: ")

    #Option 1: Generate RDF and save to file
    if (user_input=="1"):
        if not manager:
            manager = Manager()
            manager.import_reports(max_data_count)

        manager.export_file(filename)
        sucess_option_1=True
        pass

    #Option 2: Query graph
    elif (user_input=="2"):
        if not manager:
            manager = Manager()
            manager.import_reports(max_data_count)

        q = input("Enter query: ")
        query_result = manager.query(q)
        sucess_option_2=True
        pass

    #Option 3: Modify filename
    elif (user_input=="3"):
        filename = input("Enter filename: ")
        filename = os.path.abspath(filename)
        sucess_option_3=True
        pass

    #Option 4: Modify max dataset size
    elif (user_input=="4"):
        max_data_count = int(input("Enter size: "))
        sucess_option_4=True
        pass

    #Option 5: Exit
    elif (user_input=="5"):
        done = True
        pass

    #Clear terminal to prevent overcrowded 
    os.system('cls' if os.name == 'nt' else 'clear')