diff options
Diffstat (limited to 'python/quote_parsing/__main__.py')
| -rw-r--r-- | python/quote_parsing/__main__.py | 4 |
1 files changed, 2 insertions, 2 deletions
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() |