write getPartialWordList func

1 files changed, 20 insertions, 0 deletions

diff --git a/partialword.py b/partialword.py
index 30e1ba0..d82b7e4 100644
--- a/partialword.py
+++ b/partialword.py
@@ -92,3 +92,23 @@ def dropNgramTableClone(conn):
     cur.execute(DROP_NGRAM_FTS_DML)
 
     conn.commit()
+
+
+#from 2-gram.db
+def getPartialWordList(conn, threshold):
+    print(threshold)
+
+    words_list = []
+    sep = config.getWordSep()
+
+    cur = conn.cursor()
+    rows = cur.execute(SELECT_PARTIAL_WORD_DML, (threshold, )).fetchall()
+
+    for row in rows:
+        (words_str, freq) = row
+        (prefix, postfix) = words_str.strip(sep).split(sep, 1)
+        merged_word = prefix + postfix
+        words_list.append((merged_word, prefix, postfix, freq))
+
+    conn.commit()
+    return words_list