From ff638e269d439c853b5182b68ff10777f12597f1 Mon Sep 17 00:00:00 2001 From: Terry Truong Date: Sun, 5 Feb 2023 13:55:00 +1100 Subject: Use relative imports between data generation scripts Avoids the need for code that modifies sys.path, but requires running of the scripts using 'python -m' with backend/ as cwd. Also expects constants like DB_FILE to be non-relative, due to running from backend/. --- backend/hist_data/enwiki/download_imgs.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'backend/hist_data/enwiki/download_imgs.py') diff --git a/backend/hist_data/enwiki/download_imgs.py b/backend/hist_data/enwiki/download_imgs.py index e484b33..686f912 100755 --- a/backend/hist_data/enwiki/download_imgs.py +++ b/backend/hist_data/enwiki/download_imgs.py @@ -20,8 +20,9 @@ import sqlite3 import urllib.parse import requests -IMG_DB = 'img_data.db' # About 130k image names -OUT_DIR = 'imgs' +ENWIKI_DIR = os.path.dirname(os.path.realpath(__file__)) +IMG_DB = os.path.join(ENWIKI_DIR, 'img_data.db') # About 130k image names +OUT_DIR = os.path.join(ENWIKI_DIR, 'imgs') LICENSE_REGEX = re.compile(r'cc0|cc([ -]by)?([ -]sa)?([ -][1234]\.[05])?( \w\w\w?)?', flags=re.IGNORECASE) USER_AGENT = 'terryt.dev (terry06890@gmail.com)' -- cgit v1.2.3