import datetime
from mtm_status import get_latest_file


def contact_novation_counterparties(date):
    df = get_latest_file(date)
    df = df[
        (df["SwapType"].isin(["ASGM"]))
        & (df["ThirdPartyStatus.1"].isin(["Pending Consent"]))
    ]
    df = (
        df.groupby(["ExecutingBroker", "RemainingBroker", "CurrencyCode"])[
            "Novation Consent ID"
        ]
        .apply(list)
        .reset_index()
    )
    for row in df.itertuples():
        print(row.ExecutingBroker, row.RemainingBroker, row.CurrencyCode, row[4])


if __name__ == "__main__":
    contact_novation_counterparties(datetime.date(2023, 3, 14))