Initial commit.

django_mailbox/models.py

@@ -0,0 +1,105 @@
+import email
+import rfc822
+import urllib
+import urlparse
+
+from django.db import models
+from django_mailbox.transports import PopMailEnumerator, ImapMailEnumerator
+from django_mailbox.signals import message_received
+
+class Mailbox(models.Model):
+    name = models.CharField(max_length=255)
+    uri = models.CharField(
+            max_length=255,
+            help_text="""
+                Start your URI with imap:// or pop3://.
+                <br />
+                <br />
+                Example: imap://myusername:mypassword@someserver
+                <br />
+                <br />
+                Be sure to urlencode your username and password should they 
+                contain illegal characters (like @, :, etc).
+                """
+            )
+
+    @property
+    def _protocol_info(self):
+        return urlparse.urlparse(self.uri)
+
+    @property
+    def _domain(self):
+        return self._protocol_info.hostname
+
+    @property
+    def port(self):
+        return self._protocol_info.port
+
+    @property
+    def username(self):
+        return urllib.unquote(self._protocol_info.username)
+
+    @property
+    def password(self):
+        return urllib.unquote(self._protocol_info.password)
+
+    @property
+    def location(self):
+        return self._domain if self._domain else '' + self._protocol_info.path
+
+    @property
+    def type(self):
+        return self._protocol_info.scheme.lower()
+
+    def get_connection(self):
+        if self.type == 'imap':
+            conn = ImapMailEnumerator(
+                        self.location,
+                        self.port if self.port else 143
+                    )
+        elif self.type == 'pop3':
+            conn = PopMailEnumerator(
+                        self.location,
+                        self.port if self.port else 110
+                    )
+        conn.connect(self.username, self.password)
+        return conn
+
+    def get_new_mail(self):
+        connection = self.get_connection()
+        new_mail = []
+        for message in connection.get_message():
+            msg = Message()
+            msg.mailbox = self
+            msg.subject = message['subject']
+            msg.message_id = message['message-id']
+            msg.from_address = rfc822.parseaddr(message['from'])[1]
+            msg.body = message.as_string()
+            msg.save()
+            new_mail.append(msg)
+            message_received.send(sender=self, message=msg)
+        return new_mail
+
+    def __unicode__(self):
+        return self.name
+
+    class Meta:
+        verbose_name_plural = "Mailboxes"
+
+class Message(models.Model):
+    mailbox = models.ForeignKey(Mailbox, related_name='messages')
+    subject = models.CharField(max_length=255)
+    message_id = models.CharField(max_length=255)
+    from_address = models.CharField(max_length=255)
+
+    body = models.TextField()
+
+    received = models.DateTimeField(
+            auto_now_add=True
+            )
+
+    def get_email_object(self):
+        return email.message_from_string(self.body)
+
+    def __unicode__(self):
+        return self.subject