diff options
Diffstat (limited to 'python/task_server/insert_tranche_quotes.py')
| -rw-r--r-- | python/task_server/insert_tranche_quotes.py | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/python/task_server/insert_tranche_quotes.py b/python/task_server/insert_tranche_quotes.py index 8421e5e5..7c3a257b 100644 --- a/python/task_server/insert_tranche_quotes.py +++ b/python/task_server/insert_tranche_quotes.py @@ -1,6 +1,6 @@ from sqlalchemy import Table, create_engine, MetaData from sqlalchemy.exc import IntegrityError, DataError, SAWarning -import os +from serenitas.utils.env import BASE_DIR import csv import datetime import logging @@ -39,10 +39,10 @@ runningdict2 = {0: 500, 3:500, 7:500, 10:100, 15:100, 30:100} def insert_quotes(year=2016, quote_dir=None): if not quote_dir: - quote_dir = os.path.join(os.environ['BASE_DIR'], 'Tranche_data', 'Quotes') - quotefiles = [f for f in os.listdir(quote_dir) if f.endswith('csv')] + quote_dir = BASE_DIR / 'Tranche_data' / 'Quotes' + quotefiles = [f for f in quote_dir.iterdir() if f.stem == ".csv"] for quotefile in quotefiles: - with open(os.path.join(quote_dir, quotefile)) as fh: + with quotefile.open() as fh: reader = csv.DictReader(fh) data = [] for csvdict in reader: @@ -204,7 +204,8 @@ def insert_quotes(year=2016, quote_dir=None): logging.debug(e.orig) t.rollback() - os.unlink(os.path.join(quote_dir, quotefile)) + quotefile.unlink() + if __name__ == "__main__": insert_quotes() |