diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/env.py | 1 | ||||
| -rw-r--r-- | python/quote_parsing/__main__.py | 4 |
2 files changed, 3 insertions, 2 deletions
diff --git a/python/env.py b/python/env.py index 8c952a11..3df56531 100644 --- a/python/env.py +++ b/python/env.py @@ -3,3 +3,4 @@ from pathlib import Path DAILY_DIR = Path(os.environ["DAILY_DIR"]) LOG_DIR = Path(os.environ["LOG_DIR"]) +DATA_DIR = Path(os.environ["DATA_DIR"]) diff --git a/python/quote_parsing/__main__.py b/python/quote_parsing/__main__.py index 95657dea..5fef3463 100644 --- a/python/quote_parsing/__main__.py +++ b/python/quote_parsing/__main__.py @@ -5,6 +5,7 @@ import pandas as pd import pickle import sys +from env import DATA_DIR from googleapiclient import errors from . import SerenitasFileHandler from . import logger @@ -21,8 +22,7 @@ except (errors.HttpError, FileNotFoundError) as e: logger.error(e) save_emails(update=False) -data_dir = os.path.join(os.getenv("DATA_DIR"), "swaptions") -emails = [f for f in os.scandir(data_dir) if f.is_file()] +emails = [f for f in (DATA_DIR / "swaptions").iterdir() if f.is_file()] swaption_stack = {} index_data = pd.DataFrame() |