1 files changed, 45 insertions, 9 deletions
diff --git a/python/risk/ir.py b/python/risk/ir.py
index 3864569e..0cc68c89 100644
--- a/python/risk/ir.py
+++ b/python/risk/ir.py
@@ -1,15 +1,51 @@
-from serenitas.analytics import Portfolio, IRSwaption
 import logging
+from psycopg import sql
 
 logger = logging.getLogger(__name__)
 
 
-def get_ir_portfolio(date, conn, fund="SERCGMAST", **kwargs):
-    sql_str = "SELECT deal_id FROM list_ir_swaption_positions(%s, %s)"
+def insert_ir_portfolio(portf, conn):
+    cols = [
+        "date",
+        "swpt_id",
+        "notional",
+        "pv",
+        "vol",
+        "vol_type",
+        "DV01",
+        "IRGamma1bp",
+        "vega",
+    ]
+    sql_str = sql.SQL(
+        "INSERT INTO ir_swaption_risk({columns}) "
+        "VALUES({placeholders}) "
+        " ON CONFLICT (date, swpt_id) DO UPDATE "
+        "SET {update_str}"
+    ).format(
+        columns=sql.SQL(",").join([sql.Identifier(c) for c in cols]),
+        placeholders=sql.SQL(",").join([sql.Placeholder()] * len(cols)),
+        update_str=sql.SQL(",").join(
+            [
+                sql.SQL("{c} = EXCLUDED.{c}").format(c=sql.Identifier(c))
+                for c in cols[2:]
+            ]
+        ),
+    )
     with conn.cursor() as c:
-        c.execute(sql_str, (date, fund))
-        trade_ids = [tid for (tid,) in c]
-
-    portf = Portfolio([IRSwaption.from_tradeid(tid) for tid in trade_ids], trade_ids)
-    portf.value_date = date
-    return portf
+        for trade_id, trade in portf.items():
+            logger.info(f"marking IR swaption {trade_id}")
+            c.execute(
+                sql_str,
+                (
+                    trade.value_date,
+                    trade_id,
+                    trade.notional,
+                    trade.pv,
+                    trade.implied_vol,
+                    trade._vol_type.name,
+                    trade.DV01,
+                    trade.IRGamma1bp,
+                    trade.vega,
+                ),
+            )
+    conn.commit()