diff options
Diffstat (limited to 'python/monthend_interest_recon.py')
| -rw-r--r-- | python/monthend_interest_recon.py | 14 |
1 files changed, 8 insertions, 6 deletions
diff --git a/python/monthend_interest_recon.py b/python/monthend_interest_recon.py index 4e392952..b2f29309 100644 --- a/python/monthend_interest_recon.py +++ b/python/monthend_interest_recon.py @@ -16,6 +16,7 @@ import shutil from interest_statement import export_data from dateutil.relativedelta import relativedelta + logger = logging.getLogger(__name__) @@ -113,8 +114,9 @@ def get_JPM(g): value = e.text -def start_end(year, month): - start = datetime.date(year, month, 1) +def start_end(date): + date = date - relativedelta(months=1) + start = datetime.date(date.year, date.month, 1) end = start + relativedelta(months=1) end -= datetime.timedelta(days=1) return start, end @@ -151,9 +153,8 @@ def main(): parser = argparse.ArgumentParser(description="determine sender destination") parser.add_argument( - "start", default=(datetime.date(2021, datetime.date.today().month, 1)) + "date", default=datetime.date.today() ) - parser.add_argument("end", default=datetime.date.today()) parser.add_argument( "--save", action="store_true", @@ -167,13 +168,14 @@ def main(): help="for automation or for monthly", ) args = parser.parse_args() + start = datetime.date(args.date.year, args.date.month, 1) for cp in counterparties: - download_messages(em, cp, args.start, args.end, args.save) + download_messages(em, cp, start, args.date, args.save) df = get_interest(counterparties, save=args.save) start, end = start_end( - datetime.datetime.today().year, datetime.datetime.today().month - 1 + args.date ) global interest_recon interest_recon = pd.merge( |