summaryrefslogtreecommitdiffstats
path: root/partialword.py
diff options
context:
space:
mode:
Diffstat (limited to 'partialword.py')
-rw-r--r--partialword.py20
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