diff options
Diffstat (limited to 'python/quote_parsing/__main__.py')
| -rw-r--r-- | python/quote_parsing/__main__.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/python/quote_parsing/__main__.py b/python/quote_parsing/__main__.py index 1b59bd20..4ac58169 100644 --- a/python/quote_parsing/__main__.py +++ b/python/quote_parsing/__main__.py @@ -10,6 +10,7 @@ from . import SerenitasRotatingFileHandler from . import logger from .download_emails import save_emails from .parse_emails import parse_email, write_todb +from utils.db import serenitas_pool fh = SerenitasRotatingFileHandler("emails_parsing.log", 1_000_000, 5) logger.addHandler(fh) @@ -31,6 +32,7 @@ try: except FileNotFoundError: already_uploaded = {} +conn = serenitas_pool.getconn() for f in emails: date_composed, msg_id = f.name.split("_") date_composed = datetime.datetime.strptime(date_composed, "%Y-%m-%d %H-%M-%S") @@ -70,9 +72,8 @@ index_data = index_data.reset_index() index_data = index_data.drop_duplicates( ["quotedate", "index", "series", "expiry", "quote_source"] ) -from utils.db import serenitas_pool -conn = serenitas_pool.getconn() + write_todb(swaption_stack, index_data, conn) serenitas_pool.putconn(conn) |