diff options
| -rw-r--r-- | python/process_queue.py | 29 |
1 files changed, 15 insertions, 14 deletions
diff --git a/python/process_queue.py b/python/process_queue.py index b446d3ab..5760883e 100644 --- a/python/process_queue.py +++ b/python/process_queue.py @@ -11,7 +11,7 @@ import sys from io import StringIO -from serenitas.analytics import CreditIndex +from serenitas.analytics.api import CreditIndex try: from serenitas.utils.env import DAILY_DIR @@ -22,11 +22,9 @@ from collections import defaultdict from pickle import dumps, loads from serenitas.analytics.bbg_helpers import init_bbg_session, retrieve_data from serenitas.analytics.dates import bus_day -from serenitas.utils.db import dbconn from serenitas.utils.exchange import ExchangeMessage, FileAttachment from serenitas.utils.remote import FtpClient, SftpClient from serenitas.utils import get_redis_queue -from functools import partial from pyisda.date import previous_twentieth from typing import Tuple, Union from quantlib.time.api import pydate_from_qldate, UnitedStates, Days, Date @@ -810,6 +808,11 @@ def build_termination( if __name__ == "__main__": + import os + + os.environ["SERENITAS_APP_NAME"] = "process_queue" + from functools import partial + from serenitas.utils.db2 import dawn_pool parser = argparse.ArgumentParser() parser.add_argument( @@ -817,8 +820,7 @@ if __name__ == "__main__": ) args = parser.parse_args() r = get_redis_queue() - dawndb = dbconn("dawndb") - with init_bbg_session() as session: + with dawn_pool.connection() as conn, init_bbg_session() as session: for trade_type in [ "cds", "swaption", @@ -833,7 +835,7 @@ if __name__ == "__main__": trade_type=trade_type, upload=not args.no_upload, session=session, - conn=dawndb, + conn=conn, ) r.transaction(p_list, trade_type) p_upload = partial( @@ -843,11 +845,10 @@ if __name__ == "__main__": ) r.transaction(p_upload, trade_type) - for trade_type in ("cds", "swaption", "capfloor"): - for fund in ("SERCGMAST", "BOWDST", "BRINKER"): - key = f"{trade_type}_{fund}_termination" - t_list = partial( - terminate_list, key=key, upload=not args.no_upload, conn=dawndb - ) - r.transaction(t_list, key) - dawndb.close() + for trade_type in ("cds", "swaption", "capfloor"): + for fund in ("SERCGMAST", "BOWDST", "BRINKER"): + key = f"{trade_type}_{fund}_termination" + t_list = partial( + terminate_list, key=key, upload=not args.no_upload, conn=conn + ) + r.transaction(t_list, key) |