import datetime
from serenitas.utils.db import dbconn
from citco_ops.utils import PaymentSettlement

if __name__ == "__main__":
    import argparse

    parser = argparse.ArgumentParser()
    parser.add_argument(
        "date",
        nargs="?",
        type=datetime.date.fromisoformat,
        default=datetime.date.today(),
        help="working date",
    )
    args = parser.parse_args()
    conn = dbconn("dawndb")
    sql_str = "SELECT settle_date, currency, sum(payment_amount) as payment_amount FROM payment_settlements ps2 WHERE fund=%s AND asset_class in ('bond', 'spot') AND settle_date >= %s group by settle_date, currency;"
    with conn.cursor() as c:
        for fund in ("ISOSEL",):
            c.execute(sql_str, (fund, args.date))
            PaymentSettlement.stage_payment(c)
            PaymentSettlement.email_innocap(args.date)