begin bridge

1 files changed, 30 insertions, 0 deletions

diff --git a/cli/commands/bridge/words.py b/cli/commands/bridge/words.py
new file mode 100644
index 0000000..277b069
--- /dev/null
+++ b/cli/commands/bridge/words.py
@@ -0,0 +1,30 @@
+"""
+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}")
+  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))