def process_inputs(inputs): ''' Check the inputs are valid call load the appropriate functions. ''' db_location = 'database_steam_reviews.db' input_length = len(inputs) if inputs[1] == 'scrape_reviews': if input_length == 2: scraper.get_reviews(db_location) elif inputs[2] == 'continue': scraper.get_reviews(db_location) elif inputs[2] == 'new': database_manager.drop_steam_reviews() scraper.get_reviews(db_location) else: return inputs_feedback() elif inputs[1] == 'classify_data': train_classify_data.classify_reviews(db_location) else: return inputs_feedback()
def tearDown(self): db_location = 'database_test.db' database_manager.drop_steam_reviews(db_location)