#!/usr/bin/env python

# requires Python >= 2.5
import smtplib
from email.mime.multipart import MIMEMultipart
from email.mime.base import MIMEBase
from email.mime.text import MIMEText
from email.mime.audio import MIMEAudio
from email.mime.image import MIMEImage
from email.encoders import encode_base64
from mimetypes import guess_type
from os.path import basename

class GmailMail():
    def __init__(self, gmail_user, gmail_pwd):
        """
        Prepares an instance with basic authentication

        """
        self.gmail_user = gmail_user
        self.gmail_pwd = gmail_pwd

    def getAttachment(self, path, charset='ASCII'):
        contentType, encoding = guess_type(path)

        if contentType is None or encoding is not None:
            contentType = 'application/octet-stream'

        mainType, subType = contentType.split('/', 1)
        _file = open(path, 'rb')

        if mainType == 'text':
            attachment = MIMEText(_file.read(), subType, charset)
        elif mainType == 'message':
            attachment = email.message_from_file(_file)
        elif mainType == 'image':
            attachment = MIMEImage(_file.read(), _subType=subType)
        elif mainType == 'audio':
            attachment = MIMEAudio(_file.read(), _subType=subType)
        else:
            attachment = MIMEBase(mainType, subType)
            attachment.set_payload(_file.read())
            encode_base64(attachment)

        _file.close()

        attachment.add_header('Content-Disposition', 'attachment',
            filename=basename(path))

        return attachment

    def send(self, to, subject, text=u"", html=None, attachments=None, charset="iso-8859-15"):
        """
        Sends an email through Gmail using the authentication
        given to this instance.

        If given, attachments must be a list of paths pointing
        to the files we want to include.

        This script does not embed inline content (multipart/related)

        """
        if charset in ['utf8','utf-8']: #bug?
            from email.charset import add_charset, SHORTEST
            add_charset('utf-8', SHORTEST, None, None)

        if isinstance(text, unicode):
            text = text.encode(charset, 'replace')

        if isinstance(html, unicode):
            html = html.encode(charset, 'replace')

        if attachments is None:
            attachments = []

        if text: plain_part = MIMEText(text, 'plain', charset)
        if html: html_part = MIMEText(html, 'html', charset)

        is_alternative = html and text
        layers = []
        if attachments or is_alternative:
            msg = MIMEMultipart() #mixed
            msg.set_charset(charset)
            msg.preamble = 'This is a multi-part message in MIME format.'
            msg.epilogue = ''
            layers.append(msg)

            if is_alternative:
                msgAlternative = MIMEMultipart('alternative')
                msg.attach(msgAlternative)
                layers.append(msgAlternative)

            if text:
                layers[-1].attach(plain_part)
            if html:
                layers[-1].attach(html_part)

        elif text:
            msg = plain_part
        else: #html only
            msg = html_part

        for path in attachments:
            msg.attach(self.getAttachment(path, charset))

        msg['From'] = self.gmail_user
        msg['To'] = to
        msg['Subject'] = subject

        mailServer = smtplib.SMTP("smtp.gmail.com", 587)
        mailServer.ehlo()
        mailServer.starttls()
        mailServer.ehlo()
        mailServer.login(self.gmail_user, self.gmail_pwd)
        mailServer.sendmail(self.gmail_user, to, msg.as_string())
        # Should be mailServer.quit(), but that crashes...
        mailServer.close()