from word_dictionary import * def main(input_1, input_2): result = [] word_dict = WordDictionary() result.append(None) for i in range(1, len(input_1)): if input_1[i] == "add_word": res = word_dict.add_word(input_2[i][0]) result.append(None) elif input_1[i] == "search_word": res = word_dict.search_word(input_2[i][0]) result.append(res) elif input_1[i] == "get_words": res = word_dict.get_words() if res: res.sort() result.append(res) return result