1 files changed, 10 insertions, 17 deletions
diff --git a/python/globeop_reports.py b/python/globeop_reports.py
index 4175a501..8385b61f 100644
--- a/python/globeop_reports.py
+++ b/python/globeop_reports.py
@@ -31,24 +31,17 @@ def get_monthly_pnl(group_by=["identifier"], fund="SERCGMAST"):
     ].sum()
 
 
-def get_portfolio(report_date=None):
+def get_portfolio(report_date=None, fund="SERCGMAST"):
+    sql_string = "SELECT * FROM valuation_reports"
+    df = pd.read_sql_query(
+        sql_string,
+        dawn_engine,
+        parse_dates=["periodenddate"],
+        index_col=["periodenddate"],
+    )
+    df = df[df.fund == fund]
     if report_date is not None:
-        sql_string = "SELECT * FROM valuation_reports where periodenddate = %s"
-        df = pd.read_sql_query(
-            sql_string,
-            dawn_engine,
-            parse_dates=["periodenddate"],
-            index_col=["periodenddate"],
-            params=[report_date],
-        )
-    else:
-        sql_string = "SELECT * FROM valuation_reports"
-        df = pd.read_sql_query(
-            sql_string,
-            dawn_engine,
-            parse_dates=["periodenddate"],
-            index_col=["periodenddate"],
-        )
+        df = df[df.date == report_date]
     df["identifier"] = df.invid.str.replace("_A$", "", regex=True)
     return df