diff options
Diffstat (limited to 'python/monthly_interest.py')
| -rw-r--r-- | python/monthly_interest.py | 36 |
1 files changed, 16 insertions, 20 deletions
diff --git a/python/monthly_interest.py b/python/monthly_interest.py index b6a3b3b0..72987b58 100644 --- a/python/monthly_interest.py +++ b/python/monthly_interest.py @@ -93,11 +93,7 @@ def get_CITI(path): def get_GS(g): for e in g: - if "TOTAL INTEREST" in e.text: - breakpoint() - print(next(g).text) - print(next(g).text) - next(g) + if "due to" in e.text: return float(next(g).text.replace("USD", "").replace(",", "")) @@ -113,8 +109,8 @@ def get_BAML(g): def get_JPM(g): - breakpoint() for e in g: + print(e.text) if "Page" in e.text: return float(value) value = e.text @@ -147,6 +143,7 @@ def get_interest(delete=False): ).glob("*.pdf"): g = iter(load_pdf(file)) amount = func(g) + print(cp, ":", amount) interest_amounts[cp] = interest_amounts[cp] + amount if delete: try: @@ -168,7 +165,7 @@ parser.add_argument("start") parser.add_argument("end", default=datetime.date.today()) parser.add_argument("--auto", action="store_true", help="for automation or for monthly") parser.add_argument( - "--delete", action="store_true", default=True, help="for automation or for monthly" + "--delete", action="store_true", default=False, help="for automation or for monthly" ) args = parser.parse_args() @@ -178,16 +175,15 @@ for cp in counterparties: else: download_messages(em, cp, args.start, args.end) -if args.auto: - df = get_interest(delete=args.delete) - start, end = start_end( - datetime.datetime.today().year, datetime.datetime.today().month - 1 - ) - new_df = pd.merge( - export_data(start, end).groupby("broker").sum(), - df, - how="outer", - left_index=True, - right_index=True, - ) - new_df["difference"] = new_df["amount"] - new_df["monthly_statement"] +df = get_interest(delete=args.delete) +start, end = start_end( + datetime.datetime.today().year, datetime.datetime.today().month - 1 +) +new_df = pd.merge( + export_data(start, end).groupby("broker").sum(), + df, + how="outer", + left_index=True, + right_index=True, +) +new_df["difference"] = new_df["amount"] - new_df["monthly_statement"] |
