def main(): ''' In this method we ask the user if they want to enter a show, and if they type 'N' the program defaults to Family Guy and American Dad. Due to an error in the IMDb PY data base, whenever Family Guy is searched the first output is "The Sorpranos". After testing we found that, this error does not happen on most tv shows. To fix this we coded in an index feature for each of the data sets. if the user wants to put in their own TV show the are prompted to selct theirapportare show from a list. From there we create the dataset and graph them 3 different ways. ''' reader = GetCSV() if input("Do you want to input TV shows? Y or N : ") == "Y": string1 = input("Enter 1st TV show: ") string2 = input("Enter 2nd TV show: ") print() list1 = reader.get_ttcode_list(string1) print("Top 5 Search Results for ", string1, " :") # print(list1) for i in range(5): print((i + 1), ". ", list1[i]) print() print('Select which show you prefer by their number.') # print('Index goes up by one after every line. Start from the top.') index1 = input("Select the index you prefer: ") index1 = str(int(index1) - 1) # index1 -= 1 print('\n') list2 = reader.get_ttcode_list(string2) # print(list2) print() print("Top 5 Search Results for ", string2, " :") # print(list2) for i in range(5): print((i + 1), ". ", list2[i]) print() print('Select the show that you prefer by their number.') # print('Index goes up by one after every line. Start from the top.') index2 = input("Select the index you prefer: ") index2 = str(int(index2) - 1) # index2 -= 1 else: string1 = "Family Guy" string2 = "American Dad" index1 = 1 index2 = 0 print("Please hold while we scrape your data") graphs = Graphs() tt_string1 = reader.get_ttcode(string1, index1) season_length1 = reader.get_season_length(tt_string1) # print(tt_string1, season_length1) data1 = reader.get_csv(tt_string1, season_length1, string1) tt_string2 = reader.get_ttcode(string2, index2) season_length2 = reader.get_season_length(tt_string2) data2 = reader.get_csv(tt_string2, season_length2, string2) graphs.testing(data1, data2, string1, string2) graphs.scatterplot(data1, data2, string1, string2) graphs.boxplot(data1, data2, string1, string2) graphs.multi(data1, data2, string1, string2)