diff options
Diffstat (limited to 'backend/hist_data/gen_desc_data.py')
| -rwxr-xr-x | backend/hist_data/gen_desc_data.py | 10 |
1 files changed, 4 insertions, 6 deletions
diff --git a/backend/hist_data/gen_desc_data.py b/backend/hist_data/gen_desc_data.py index 68f9e56..6c9fee2 100755 --- a/backend/hist_data/gen_desc_data.py +++ b/backend/hist_data/gen_desc_data.py @@ -1,10 +1,10 @@ #!/usr/bin/python3 """ -Maps events to short descriptions from Wikipedia, -and stores them in the database. +Maps events to short descriptions from Wikipedia, and stores them in the database. """ +import argparse import os, sqlite3 ENWIKI_DB = os.path.join('enwiki', 'desc_data.db') @@ -16,10 +16,9 @@ def genData(enwikiDb: str, dbFile: str) -> None: dbCur = dbCon.cursor() dbCur.execute('CREATE TABLE descs (id INT PRIMARY KEY, wiki_id INT, desc TEXT)') # - print('Getting events with images') + print('Getting events') titleToId: dict[str, int] = {} - query = 'SELECT events.id, events.title FROM events INNER JOIN event_imgs ON events.id = event_imgs.id' - for eventId, title in dbCur.execute(query): + for eventId, title in dbCur.execute('SELECT id, title FROM events'): titleToId[title] = eventId # print('Getting Wikipedia descriptions') @@ -53,7 +52,6 @@ def genData(enwikiDb: str, dbFile: str) -> None: dbCon.close() if __name__ == '__main__': - import argparse parser = argparse.ArgumentParser(description=__doc__, formatter_class=argparse.RawDescriptionHelpFormatter) args = parser.parse_args() # |