import imaplib, email, re, sys

def get_code(addr, password):
    m = imaplib.IMAP4_SSL("imap.mmmail.local", 993)
    m.login(addr, password)
    m.select("INBOX")
    typ, ids = m.search(None, "ALL")
    if not ids[0]:
        return None
    last = ids[0].split()[-1]
    typ, data = m.fetch(last, "(RFC822)")
    msg = email.message_from_bytes(data[0][1])
    body = msg.get_payload(decode=True) or b""
    body = body.decode(errors="ignore")
    m = re.search(r"\b\d{6}\b", body)
    return m.group(0) if m else None

if __name__ == "__main__":
    addr, password = sys.argv[1], sys.argv[2]
    print(get_code(addr, password))