diff options
| -rwxr-xr-x | backend/data/enwiki/lookupPage.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/backend/data/enwiki/lookupPage.py b/backend/data/enwiki/lookupPage.py index 5d6afe9..c795c35 100755 --- a/backend/data/enwiki/lookupPage.py +++ b/backend/data/enwiki/lookupPage.py @@ -13,14 +13,14 @@ if len(sys.argv) != 2: dumpFile = "enwiki-20220501-pages-articles-multistream.xml.bz2" indexDb = "dumpIndex.db" -pageTitle = sys.argv[1] +pageTitle = sys.argv[1].replace("_", " ") # Searching index file print("Lookup offset in index db") dbCon = sqlite3.connect(indexDb) dbCur = dbCon.cursor() -row = dbCur.execute("SELECT title, offset, next_offset FROM offsets WHERE title = ?", - (pageTitle.replace("_", " "),)).fetchone() +query = "SELECT title, offset, next_offset FROM offsets WHERE title = ?" +row = dbCur.execute(query, (pageTitle,)).fetchone() if row == None: print("Title not found") sys.exit(0) |