1 files changed, 15 insertions, 0 deletions
diff --git a/python/risk/trs.py b/python/risk/trs.py
new file mode 100644
index 00000000..59b9ee35
--- /dev/null
+++ b/python/risk/trs.py
@@ -0,0 +1,15 @@
+from serenitas.analytics.api import Portfolio, TRS
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+def get_trs_portfolio(date, conn, fund="SERCGMAST", **kwargs):
+    sql_str = "SELECT dealid FROM trs WHERE trade_date <= %s AND maturity_date > %s AND fund = %s"
+    with conn.cursor() as c:
+        c.execute(sql_str, (date, fund))
+        trade_ids = [tid for (tid,) in c]
+
+    portf = Portfolio([TRS.from_tradeid(tid) for tid in trade_ids], trade_ids)
+    portf.value_date = date
+    return portf