diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/interest_statement.py | 20 |
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 |