diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/monthend_interest_recon.py | 18 |
1 files changed, 8 insertions, 10 deletions
diff --git a/python/monthend_interest_recon.py b/python/monthend_interest_recon.py index efec80c1..59859bde 100644 --- a/python/monthend_interest_recon.py +++ b/python/monthend_interest_recon.py @@ -154,8 +154,8 @@ def get_interest(counterparties, save=False): if __name__ == "__main__": em = ExchangeMessage() counterparties = ["BNP", "CITI", "CS", "GS", "MS", "BAML", "JPM"] - - parser = argparse.ArgumentParser(description="determine sender destination") + # First run --save, then run --accept if it looks good + parser = argparse.ArgumentParser(description="Save, accept within tolerances") parser.add_argument( "date", type=datetime.date.fromisoformat, @@ -180,22 +180,20 @@ if __name__ == "__main__": for cp in counterparties: download_messages(em, cp, start, args.date, args.save) - df = get_interest(counterparties, save=args.save) + interest_amounts = get_interest(counterparties, save=args.save) start, end = start_end(args.date) - global interest_recon - interest_recon = pd.merge( + global df + df = pd.merge( export_data(start, end).groupby("broker").sum(), - df, + interest_amounts, how="outer", left_index=True, right_index=True, ) - interest_recon["difference"] = ( - interest_recon["amount"] - interest_recon["monthly_statement"] - ) + df["difference"] = df["amount"] - df["monthly_statement"] if args.accept: df = export_data(start, end) - for k, v in interest_recon["difference"].items(): + for k, v in df["difference"].items(): df.loc[k].loc["CSH_CASH"] -= v from io import StringIO from serenitas.utils.exchange import ExchangeMessage, FileAttachment |