diff options
Diffstat (limited to 'backend/hist_data/enwiki/gen_pageview_data.py')
| -rwxr-xr-x | backend/hist_data/enwiki/gen_pageview_data.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/backend/hist_data/enwiki/gen_pageview_data.py b/backend/hist_data/enwiki/gen_pageview_data.py index 57d6c7b..2dc6eb7 100755 --- a/backend/hist_data/enwiki/gen_pageview_data.py +++ b/backend/hist_data/enwiki/gen_pageview_data.py @@ -22,9 +22,10 @@ from collections import defaultdict import bz2 import sqlite3 -PAGEVIEW_FILES = glob.glob('./pageviews/pageviews-*-user.bz2') -DUMP_INDEX_DB = 'dump_index.db' -DB_FILE = 'pageview_data.db' +ENWIKI_DIR = os.path.dirname(os.path.realpath(__file__)) +PAGEVIEW_FILES = glob.glob(os.path.join(ENWIKI_DIR, './pageviews/pageviews-*-user.bz2')) +DUMP_INDEX_DB = os.path.join(ENWIKI_DIR, 'dump_index.db') +DB_FILE = os.path.join(ENWIKI_DIR, 'pageview_data.db') def genData(pageviewFiles: list[str], dumpIndexDb: str, dbFile: str) -> None: if os.path.exists(dbFile): |