1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
"""
Find connections between two words
"""
import click
import simplejson as json
from app.thesaurus.api import Thesaurus
@click.command()
@click.option('-a', '--a', 'opt_word_a', required=True,
help='Starting word')
@click.option('-b', '--b', 'opt_word_b', required=True,
help='Ending word')
@click.pass_context
def cli(ctx, opt_word_a, opt_word_b):
"""Find connections between two words
"""
thesaurus = Thesaurus()
print(f"Starting word: {opt_word_a}")
print(f"Ending word: {opt_word_b}")
visited = set()
results_a = thesaurus.search(opt_word_a)
results_b = thesaurus.search(opt_word_b)
# use sets
# make set of results_a
# find overlap with results_b
# if there's no match...
# search for first word in results_a
# loop over results...
# print(json.dumps(results, indent=2))
dist = 0
marked = {}
queue = [opt_word_a]
found = False
# First compute distance to each node to find a path
while len(queue):
dist = dist + 1
print(f"Iteration: distance {dist}, {len(queue)} items in queue")
newqueue = []
for word_q in queue:
word_result = thesaurus.search(word_q)
for cat in word_result['categories']:
catid = cat['catid']
if catid in marked:
continue
marked[catid] = dist
category_result = thesaurus.category(catid)
for word_c in category_result['words']:
word_n = word_c['word']
if word_n in marked:
continue
marked[word_n] = dist
if word_n == opt_word_b:
thesaurus.search(word_n)
found = True
break
newqueue.append(word_n)
queue = newqueue
if not Found:
print(f"No path found, distance of {dist} reached, {len(marked)} nodes checked")
return
# Then follow the chain of shortest distance to follow the path
word_n = opt_word_b
while word_n != opt_word_a:
dist = 999999
next_catid = ""
next_word = ""
word_result = thesaurus.search(word_n)
print(f"-> {word_result['word']}")
for cat in word_result['categories']
catid = cat['catid']
if catid in marked and marked[catid] < dist:
dist = marked[catid]
next_catid = catid
cat_result = thesaurus.category(catid)
print(f"-> {cat_result['category']}")
for word_c in category_result:
word_n = word_c['word']
if word_n in marked and marked[word_n] < dist:
next_word = word_n
word_n = next_word
print(f"-> {word_n}")
|
