SecureInbox.py

   1 #!/bin/env python
   2 # Created:20080603
   3 # By Jeff Connelly
   4 #
   5 # Communicate with mail servers
   6 #
   7 import imaplib
   8 import getpass
   9 import email
  10
  11 class SecureInbox(object):
  12     def __init__(self, username, password):
  13         # IMAP SSL for incoming messages
  14         self.mail_in = imaplib.IMAP4_SSL("imap.gmail.com", 993)
  15
  16         try:
  17             typ, data = self.mail_in.login(username, password)
  18         except imaplib.error, e:
  19             raise "Login failure: %s" % (e,)
  20         else:
  21             assert typ == "OK", "imap login returned: %s %s" % (status, message)
  22             print "Logged in:", typ, data
  23
  24         # Always have "Secure" mailbox selected
  25         typ, num_msgs = self.mail_in.select(mailbox="Secure")
  26         if typ != "OK":
  27             raise ("imap select failure: %s %s" % (typ, num_msgs) +
  28                 "The 'Secure' tag doesn't exist. Tag some of your EMOTP messages" +
  29                 "using a new label, named 'Secure'")
  30
  31     def get_messages(self):
  32         msgs = []
  33
  34         try:
  35             typ, all_msgs_string = self.mail_in.search(None, 'ALL')
  36         except imaplib.error, e:
  37             raise "imap search failed: %s" % (e,)
  38
  39         all_msgs = all_msgs_string[0].split()
  40         for num in all_msgs:
  41             typ, body = self.mail_in.fetch(num, "(BODY[])")
  42             msg = email.message_from_string(body[0][1])
  43
  44             body = str(msg.get_payload())
  45             subject = msg.get("Subject")
  46             sender = msg.get("From")
  47             id = msg.get("Message-ID")
  48
  49             #print 'Message %s\n%s\n' % (num, data[0][1])
  50             if "--EMOTP_BEGIN--" not in body:
  51                 continue
  52
  53             msgs.append({"body": body,
  54                 "subject": subject,
  55                 "sender": sender,
  56                 "id": id,
  57                 "num": num})
  58
  59         return msgs
  60
  61     def __del__(self):
  62         self.mail_in.close()
  63         self.mail_in.logout()
  64
  65 def main():
  66     ms = SecureInbox("shellreef", getpass.getpass())
  67
  68     for m in ms.get_messages():
  69         print m["sender"], m["subject"]
  70
  71 if __name__ == "__main__":
  72     main()
  73