diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/process_queue.py | 8 |
1 files changed, 3 insertions, 5 deletions
diff --git a/python/process_queue.py b/python/process_queue.py index a1cc8ae1..bbdd779f 100644 --- a/python/process_queue.py +++ b/python/process_queue.py @@ -13,7 +13,7 @@ from ftplib import FTP import config import os from sqlalchemy import create_engine -from bbg_helpers import init_bbg_session, retreive_data, process_msgs +from bbg_helpers import init_bbg_session, retreive_data def decode_dict(d): return {k.decode() if isinstance(k, bytes) else k: \ @@ -62,13 +62,11 @@ def bbg_process(cursor, session, trade): bbg_id = (trade['cusip'] or trade['isin']) + ' Mtge' bbg_type = 'Mtge' - data = retreive_data(session, [bbg_id], fields, trade['settle_date']) - df = process_msgs(data) + data = retreive_data(session, [bbg_id], fields, overrides={'SETTLE_DT': trade['settle_date']}) if not df[bbg_id]: bbg_id = (trade['cusip'] or trade['isin']) + ' Corp' bbg_type = 'Corp' - data = retreive_data(session, [bbg_id], fields, trade['settle_date']) - df = process_msgs(data) + data = retreive_data(session, [bbg_id], fields, overrides={'SETTLE_DT': trade['settle_date']}) bbg_data = df[bbg_id] if bbg_data.get('MTG_FACTOR_SET_DT', 0) == 0: |