gistfile1.pyw

import oauth2.clients.imap as imaplib
import email


class Mailbox(object):
    def __init__(self, conn):                                                   
        self.conn = conn
        self.total = self.conn.select('INBOX')[1][0]
        self.current = 1

    def __iter__(self):
        return self                                                             

    def next(self):
        if self.current > self.total:                                           
            raise StopIteration()                                               
    
        typ, data = self.conn.fetch(self.current, '(RFC822)')                   
        self.current += 1

        return email.message_from_string(data[0][1])

    def __del__(self):
        self.conn.close()
        self.conn.logout()