How to use the mailsuite.imap.IMAPClient function in mailsuite

def idle_callback(connection):
        res = get_dmarc_reports_from_inbox(connection=connection,
                                           reports_folder=reports_folder,
                                           archive_folder=archive_folder,
                                           delete=delete,
                                           test=test,
                                           offline=offline,
                                           nameservers=nameservers,
                                           dns_timeout=dns_timeout,
                                           strip_attachment_payloads=sa)
        callback(res)

    while True:
        try:
            IMAPClient(host=host, username=username, password=password,
                       port=port, ssl=ssl, verify=verify,
                       initial_folder=reports_folder,
                       idle_callback=idle_callback,
                       idle_timeout=idle_timeout)
        except timeout:
            logger.warning("IMAP connection timeout. Reconnecting...")

aggregate_reports = []
    forensic_reports = []
    aggregate_report_msg_uids = []
    forensic_report_msg_uids = []
    aggregate_reports_folder = "{0}/Aggregate".format(archive_folder)
    forensic_reports_folder = "{0}/Forensic".format(archive_folder)
    invalid_reports_folder = "{0}/Invalid".format(archive_folder)

    if results:
        aggregate_reports = results["aggregate_reports"].copy()
        forensic_reports = results["forensic_reports"].copy()

    if connection:
        server = connection
    else:
        server = IMAPClient(host, user, password, port=port,
                            ssl=ssl, verify=verify,
                            timeout=timeout,
                            max_retries=max_retries,
                            initial_folder=reports_folder)

    server.create_folder(archive_folder)
    server.create_folder(aggregate_reports_folder)
    server.create_folder(forensic_reports_folder)
    server.create_folder(invalid_reports_folder)

    messages = server.search()
    total_messages = len(messages)
    logger.debug("Found {0} messages in {1}".format(len(messages),
                                                    reports_folder))
    for i in range(len(messages)):
        msg_uid = messages[i]