diff options
Diffstat (limited to 'backend/hist_data/gen_pop_data.py')
| -rwxr-xr-x | backend/hist_data/gen_pop_data.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/backend/hist_data/gen_pop_data.py b/backend/hist_data/gen_pop_data.py index 8d50b6b..4a4d972 100755 --- a/backend/hist_data/gen_pop_data.py +++ b/backend/hist_data/gen_pop_data.py @@ -8,8 +8,9 @@ import argparse import os import sqlite3 -PAGEVIEWS_DB = os.path.join('enwiki', 'pageview_data.db') -DB_FILE = 'data.db' +DATA_DIR = os.path.dirname(os.path.realpath(__file__)) +PAGEVIEWS_DB = os.path.join(DATA_DIR, 'enwiki', 'pageview_data.db') +DB_FILE = os.path.join(DATA_DIR, 'data.db') def genData(pageviewsDb: str, dbFile: str) -> None: dbCon = sqlite3.connect(dbFile) |