91 lines
2.8 KiB
Python
91 lines
2.8 KiB
Python
|
import csv
|
||
|
|
import tkinter as tk
|
||
|
|
import os.path
|
||
|
|
import pickle
|
||
|
|
|
||
|
|
|
||
|
|
def word_from_list(event):
|
||
|
|
if synonym_list_box.curselection() == ():
|
||
|
|
return
|
||
|
|
word = synonym_list_box.get(synonym_list_box.curselection()).lower()
|
||
|
|
search_word.delete(0, tk.END)
|
||
|
|
search_word.insert(tk.END, string=word)
|
||
|
|
search_for_word()
|
||
|
|
|
||
|
|
|
||
|
|
def search_for_word():
|
||
|
|
word = search_word.get().lower()
|
||
|
|
word_list = thesaurus.get(word)
|
||
|
|
synonym_list_box.delete(0, tk.END)
|
||
|
|
if word_list is not None:
|
||
|
|
word_list = sorted(list(word_list))
|
||
|
|
for i, word in enumerate(word_list):
|
||
|
|
synonym_list_box.insert(i, word)
|
||
|
|
else:
|
||
|
|
synonym_list_box.insert(0, 'Not found')
|
||
|
|
|
||
|
|
|
||
|
|
def build_thesaurus():
|
||
|
|
d = {}
|
||
|
|
if os.path.exists('synonyms.pickle'):
|
||
|
|
# Load pickle file
|
||
|
|
with open('synonyms.pickle', 'rb') as pck:
|
||
|
|
data = pck.read()
|
||
|
|
d = pickle.loads(data)
|
||
|
|
else:
|
||
|
|
# Build dictionary d from synonyms.txt file,
|
||
|
|
# then save it to a pickle file
|
||
|
|
with open('synonyms.txt', 'rt') as file:
|
||
|
|
# First line is copyright info, so burn it
|
||
|
|
# Commented out code is for making list of lists without csv
|
||
|
|
#file.readline()
|
||
|
|
# read remainder of lines into a list
|
||
|
|
#all_words_list = file.readlines()
|
||
|
|
all_words_list = list(csv.reader(file))[:-1]
|
||
|
|
|
||
|
|
for words in all_words_list:
|
||
|
|
# Split words into list of words, remove newline from last word
|
||
|
|
#words_list = words.strip().split(',')
|
||
|
|
#print(words_list)
|
||
|
|
for word_key in words:
|
||
|
|
word_key = word_key.lower()
|
||
|
|
# If word not a key in dict, add to dict with value as empty set
|
||
|
|
if word_key.lower() not in d:
|
||
|
|
d[word_key] = set()
|
||
|
|
|
||
|
|
# Go through all words in words_list to add to value set as synonym
|
||
|
|
for synonym in words:
|
||
|
|
if word_key != synonym:
|
||
|
|
d[word_key].add(synonym)
|
||
|
|
# Pickle the dict
|
||
|
|
with open('synonyms.pickle', 'wb') as pck: # And the Warner sister, Dot!
|
||
|
|
pickle.dump(d, pck, protocol=pickle.HIGHEST_PROTOCOL)
|
||
|
|
|
||
|
|
# Return the thesaurus/dict (d)
|
||
|
|
return d
|
||
|
|
|
||
|
|
|
||
|
|
window = tk.Tk()
|
||
|
|
window.title('Thesaurus')
|
||
|
|
window.minsize(width=300, height=600)
|
||
|
|
|
||
|
|
thesaurus = build_thesaurus()
|
||
|
|
|
||
|
|
title_label = tk.Label(text='Thesaurus', font='Arial 24 bold italic')
|
||
|
|
title_label.pack(pady=(5, 15))
|
||
|
|
|
||
|
|
search_word = tk.Entry(width=30)
|
||
|
|
search_word.focus()
|
||
|
|
search_word.pack(pady=5)
|
||
|
|
|
||
|
|
search_button = tk.Button(text='Find Synonyms', command=search_for_word)
|
||
|
|
search_button.pack(pady=(5, 25))
|
||
|
|
|
||
|
|
synonym_list_box = tk.Listbox(height=25, width=40)
|
||
|
|
synonym_list_box.bind(sequence='<<ListboxSelect>>', func=word_from_list)
|
||
|
|
synonym_list_box.pack(pady=5)
|
||
|
|
|
||
|
|
|
||
|
|
# Always at the end
|
||
|
|
window.mainloop()
|