blob: 9a3a7a1765968cc98d8fe255377048ac77522d9e (
plain)
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
|
#!/usr/bin/python3
import sys
import sqlite3
usageInfo = f"usage: {sys.argv[0]}\n"
usageInfo += "Reads alt-names from a 'names' table in a database, and adds a spellfix \n"
usageInfo += "table 'spellfix_alt_names' usable for fuzzy-searching those names.\n"
if len(sys.argv) > 1:
print(usageInfo, file=sys.stderr)
sys.exit(1)
dbFile = "data.db"
# Connect to db, and load spellfix extension
dbCon = sqlite3.connect(dbFile)
dbCon.enable_load_extension(True)
dbCon.load_extension('./spellfix')
# Create spellfix table, and insert alt-names
spellfixCur = dbCon.cursor()
spellfixCur.execute("CREATE VIRTUAL TABLE spellfix_alt_names USING spellfix1")
namesCur = dbCon.cursor()
iterationNum = 0
for row in namesCur.execute("SELECT DISTINCT alt_name FROM names"):
iterationNum += 1
if iterationNum % 10000 == 0:
print("Loop {}: {}".format(iterationNum, row[0]))
# Insert alt-name
spellfixCur.execute("INSERT INTO spellfix_alt_names(word) VALUES (?)", (row[0],))
# Close db
dbCon.commit()
dbCon.close()
|
