1 files changed, 14 insertions, 6 deletions
diff --git a/python/interest_statement.py b/python/interest_statement.py
index 38f5d312..716381ad 100644
--- a/python/interest_statement.py
+++ b/python/interest_statement.py
@@ -49,9 +49,17 @@ def f(df_balances, df_rates, broker, start_date, end_date):
         )
 
 
-dfs = {}
-for cp in ("GS", "MS", "BAML_ISDA", "CITI", "CS", "BNP"):
-    dfs[cp] = f(df_balances, df_rates, cp, "2020-12-01", "2020-12-31")
-df = pd.concat(dfs, names=["broker"])
-df = df[df.amount != 0.0]
-df.amount *= -1.0
+if __name__ == "__main__":
+    import argparse
+
+    parser = argparse.ArgumentParser()
+    parser.add_argument("start_date")
+    parser.add_argument("end_date")
+    args = parser.parse_args()
+
+    dfs = {}
+    for cp in ("GS", "MS", "BAML_ISDA", "CITI", "CS", "BNP"):
+        dfs[cp] = f(df_balances, df_rates, cp, args.start_date, args.end_date)
+    df = pd.concat(dfs, names=["broker", "folder"])
+    df = df[df.amount != 0.0]
+    df.amount *= -1.0