diff options
Diffstat (limited to 'python/markit/__main__.py')
| -rw-r--r-- | python/markit/__main__.py | 21 |
1 files changed, 15 insertions, 6 deletions
diff --git a/python/markit/__main__.py b/python/markit/__main__.py index fb309ad7..7041e160 100644 --- a/python/markit/__main__.py +++ b/python/markit/__main__.py @@ -23,7 +23,7 @@ from serenitas.utils import SerenitasFileHandler from serenitas.utils.db import serenitas_pool from serenitas.analytics.yieldcurve import CurveType from serenitas.analytics.utils import prev_business_day -from serenitas.analytics.bbg_helpers import init_bbg_session, BBG_IP +from serenitas.analytics.bbg_helpers import init_bbg_session # parse arguments parser = argparse.ArgumentParser() @@ -142,9 +142,18 @@ elif args.rates: else: logger.info(f"Downloaded {curr} rates") break - with init_bbg_session(BBG_IP) as session: - for c in CurveType: - update_bbg_members(conn, session, c, workdate) - update_bbg_desc(conn, session, workdate) - get_bbg_quotes(conn, session, workdate) + + retry = 0 + max_tries = 2 + while retry < max_tries: + try: + with init_bbg_session() as session: + for c in CurveType: + update_bbg_members(conn, session, c, workdate) + update_bbg_desc(conn, session, workdate) + get_bbg_quotes(conn, session, workdate) + except RuntimeError: + retry += 1 + else: + break serenitas_pool.putconn(conn) |