diff options
Diffstat (limited to 'python/markit')
| -rw-r--r-- | python/markit/__main__.py | 23 |
1 files changed, 9 insertions, 14 deletions
diff --git a/python/markit/__main__.py b/python/markit/__main__.py index 7041e160..44995bee 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 +from serenitas.analytics.bbg_helpers import bbg_retry # parse arguments parser = argparse.ArgumentParser() @@ -143,17 +143,12 @@ elif args.rates: logger.info(f"Downloaded {curr} rates") break - 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 + @bbg_retry(2) + def bbg_call(session, conn, workdate): + for c in CurveType: + update_bbg_members(conn, session, c, workdate) + update_bbg_desc(conn, session, workdate) + get_bbg_quotes(conn, session, workdate) + + bbg_call(conn, workdate) serenitas_pool.putconn(conn) |