Coverage for bookie.lib.message : 77%

"""Create and send messages to users 

""" 

import logging 

import os 

import smtplib 

from email.mime.multipart import MIMEMultipart 

from email.mime.text import MIMEText 

from pyramid.settings import asbool 

LOG = logging.getLogger(__name__) 

# notification statuses 

# might have pending, sent, failed 

MSG_STATUS = { 

    'pending': 0, 

    'sent': 1, 

    'failed': 2, 

    'not_sent': 3, 

    'error': 4, 

} 

def sendmail(to, from_addr, subject, body): 

    sendmail_location = "/usr/sbin/sendmail" 

    p = os.popen("{0} -t".format(sendmail_location), "w") 

    p.write("From: {0}\n".format(from_addr)) 

    p.write("To: {0}\n".format(to)) 

    p.write("Subject: {0}\n".format(subject)) 

    p.write("\n")  # blank line separating headers from body 

    p.write(body) 

    status = p.close() 

    if status != 0: 

        LOG.debug("SENDMAIL FAIL: " + str(status)) 

        return False 

    else: 

        return True 

class Message(object): 

    """This is a base email message we can then tweak""" 

    def __init__(self, to, subject, settings): 

        """Start up a basic message""" 

        self.to = to 

        self.subject = subject 

        self.settings = settings 

        self.from_addr = settings.get('email.from', None) 

        # need ot setup/override in the extending classes 

        self.message_file = None 

    def _get_message_body(self, template_file, message_data): 

        """Return the completed message template body 

        """ 

        return "Test email message from bookie" 

        # lookup = config['pylons.app_globals'].mako_lookup 

        # template = lookup.get_template(template_file) 

        # # template vars are a combo of the obj dict and the extra dict 

        # template_vars = {'data': message_data} 

        # return template.render(**template_vars) 

    def send(self, message_data=None): 

        """Send the message with the given subject 

        body can be sent as part of send() or it can be set to the object as 

        msg.body = "xxx" 

        """ 

        self.body = self._get_message_body(self.message_file, message_data) 

        msg = MIMEMultipart('related') 

        msg['Subject'] = self.subject 

        msg['From'] = self.from_addr 

        msg['To'] = self.to 

        plain_text = MIMEText(self.body, 'plain', _charset="UTF-8") 

        msg.attach(plain_text) 

        is_live = asbool(self.settings.get('email.enable', False)) 

        is_live = True 

        if not is_live: 

            print msg.as_string() 

            return MSG_STATUS['sent'] 

        else: 

            try: 

                all_emails = msg['To'] 

                smtp_server = self.settings.get('email.host') 

                if smtp_server == 'sendmail': 

                    sendmail(msg['To'], msg['From'], msg['Subject'], self.body) 

                else: 

                    mail_server = smtplib.SMTP(smtp_server) 

                    mail_server.sendmail(msg['From'], 

                                         all_emails, 

                                         msg.as_string()) 

                    mail_server.quit() 

                return MSG_STATUS['sent'] 

            except smtplib.SMTPException: 

                LOG.error( 

                    "SMTP Error sending notice for: {0} ".format( 

                        str(msg))) 

                return MSG_STATUS['error'] 

class ReactivateMsg(Message): 

    """Send an email for a reactivation email""" 

    def _get_message_body(self, template_file, message_data): 

        """Return the completed message template body 

        """ 

        return """ 

Hello {username}: 

Please activate your Bookie account by clicking on the following url: 

{url} 

--- 

The Bookie Team""".format(**message_data) 

        # lookup = config['pylons.app_globals'].mako_lookup 

        # template = lookup.get_template(template_file) 

        # # template vars are a combo of the obj dict and the extra dict 

        # template_vars = {'data': message_data} 

        # return template.render(**template_vars) 

class InvitationMsg(Message): 

    """Send an email that you've been invited to the system""" 

    def _get_message_body(self, template_file, message_data): 

        """Return the completed message template body 

        """ 

        return """ 

You've been invited to Bookie, a web service for managing your bookmarks! 

Please click the link below to activate your account. 

{0} 

We currently support importing from Google Bookmarks and Delicious exports. 

Importing from a Chrome or Firefox export does work, however it reads the 

folder names in as tags. So be aware of that. 

Get the Chrome extension from the Chrome web store: 

https://chrome.google.com/webstore/detail/knnbmilfpmbmlglpeemajjkelcbaaega 

If you have any issues feel free to join #bookie on freenode.net or report 

the issue or idea on http://github.com/mitechie/Bookie/issues. 

We also encourage you to sign up for our mailing list at: 

https://groups.google.com/forum/#!forum/bookie_bookmarks 

and our Twitter account: 

http://twitter.com/BookieBmarks 

Bookie is open source. Check out the source at: 

https://github.com/mitechie/Bookie 

--- 

The Bookie Team""".format(message_data) 

class ImportFailureMessage(Message): 

    """Send an email that the import has failed.""" 

    def _get_message_body(self, template_file, message_data): 

        """Build the email message body.""" 

        msg = """ 

The import for user {username} has failed to import. The path to the import 

is: 

{file_path} 

Error: 

{exc} 

""".format(**message_data) 

        return msg 

class UserImportFailureMessage(Message): 

    """Send an email to the user their import has failed.""" 

    def _get_message_body(self, template_file, message_data): 

        """Build the email message body.""" 

        msg = """ 

Your import has failed. The error is listed below. Please file a bug at 

https://github.com/mitechie/bookie/issues if this error continues. You may 

also join #bookie on freenode irc if you wish to aid in debugging the issue. 

If the error pertains to a specific bookmark in your import file you might try 

removing it and importing the file again. 

Error 

---------- 

{exc} 

A copy of this error has been logged and will be looked at. 

--- 

The Bookie Team""".format(**message_data) 

        return msg 

class UserImportSuccessMessage(Message): 

    """Send an email to the user after a successful import.""" 

    def _get_message_body(self, template_file, message_data): 

        """Build the email message body.""" 

        msg = """ 

Your bookmark import is complete! We've begun processing your bookmarks to 

load their page contents and fulltext index them. This process might take a 

while if you have a large number of bookmarks. Check out your imported 

bookmarks at https://bmark.us/{username}/recent. 

--- 

The Bookie Team""".format(**message_data) 

        return msg