diff options
Diffstat (limited to 'python/bespoke_utils.py')
| -rw-r--r-- | python/bespoke_utils.py | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/python/bespoke_utils.py b/python/bespoke_utils.py index a7aa60e8..19ac4919 100644 --- a/python/bespoke_utils.py +++ b/python/bespoke_utils.py @@ -33,7 +33,7 @@ def insert_bbg_tickers(conn: connection, tickers: List[str]) -> Dict[str, Tuple[ for k, v in ticker_data.items(): c.execute(sql_str, (v["cds_company_name"], v["cds_company_id"], - v["cds_corp_tkr"], + v.get("cds_corp_tkr"), v["crncy"], v["sw_seniority"], v["cds_restructuring_type"].replace("-", " "), @@ -73,7 +73,7 @@ def backpopulate_short_codes(conn): for v in short_code.values() if 'restructuring_type_short_code' in v]) conn.commit() -def get_bbg_ids(conn: connection, df: pd.DataFrame = pd.read_clipboard(header=None), +def get_bbg_ids(conn: connection, df: pd.DataFrame, date: datetime.date = datetime.date.today()): df_mapping = pd.read_sql_query("SELECT company_id, markit_ticker, markit_tier, " "seniority FROM historical_cds_issuers(%s) " @@ -101,9 +101,12 @@ def add_basketid(conn: connection, basketid: int, if __name__ == "__main__": df = pd.read_clipboard(header=None) - bbg_tickers = df[1].to_list() from db import serenitas_pool conn = serenitas_pool.getconn() + df = get_bbg_ids(conn, pd.read_clipboard(header=None)) + #missing tickers + missing_tickers = df[df.company_id.isnull()] + bbg_tickers = df[1].to_list() d = insert_bbg_tickers(conn, bbg_tickers) insert_bbg_markit_mapping(conn, d) serenitas_pool.putconn(conn) |