Moved notifications module.

2 files changed, 169 insertions, 0 deletions

diff --git a/src/notifications/emails.py b/src/notifications/emails.py
new file mode 100644
index 00000000..5fb5ce01
--- /dev/null
+++ b/src/notifications/emails.py
@@ -0,0 +1,102 @@
+#! /usr/bin/env python
+# -*- coding: utf-8 -*-
+
+# JARR - A Web based news aggregator.
+# Copyright (C) 2010-2016  Cédric Bonhomme - https://www.cedricbonhomme.org
+#
+# For more information : https://github.com/JARR-aggregator/JARR/
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as
+# published by the Free Software Foundation, either version 3 of the
+# License, or (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+import logging
+import smtplib
+from email.mime.multipart import MIMEMultipart
+from email.mime.text import MIMEText
+
+from postmark import PMMail
+
+import conf
+from web.decorators import async
+
+logger = logging.getLogger(__name__)
+
+
+@async
+def send_async_email(mfrom, mto, msg):
+    try:
+        s = smtplib.SMTP(conf.NOTIFICATION_HOST)
+        s.login(conf.NOTIFICATION_USERNAME, conf.NOTIFICATION_PASSWORD)
+    except Exception:
+        logger.exception('send_async_email raised:')
+    else:
+        s.sendmail(mfrom, mto, msg.as_string())
+        s.quit()
+
+def send(*args, **kwargs):
+    """
+    This functions enables to send email through Postmark
+    or a SMTP server.
+    """
+    if conf.ON_HEROKU:
+        send_postmark(**kwargs)
+    else:
+        send_smtp(**kwargs)
+
+def send_smtp(to="", bcc="", subject="", plaintext="", html=""):
+    """
+    Send an email.
+    """
+    # Create message container - the correct MIME type is multipart/alternative.
+    msg = MIMEMultipart('alternative')
+    msg['Subject'] = subject
+    msg['From'] = conf.NOTIFICATION_EMAIL
+    msg['To'] = to
+    msg['BCC'] = bcc
+
+    # Record the MIME types of both parts - text/plain and text/html.
+    part1 = MIMEText(plaintext, 'plain', 'utf-8')
+    part2 = MIMEText(html, 'html', 'utf-8')
+
+    # Attach parts into message container.
+    # According to RFC 2046, the last part of a multipart message, in this case
+    # the HTML message, is best and preferred.
+    msg.attach(part1)
+    msg.attach(part2)
+
+    try:
+        s = smtplib.SMTP(conf.NOTIFICATION_HOST)
+        s.login(conf.NOTIFICATION_USERNAME, conf.NOTIFICATION_PASSWORD)
+    except Exception:
+        logger.exception("send_smtp raised:")
+    else:
+        s.sendmail(conf.NOTIFICATION_EMAIL, msg['To'] + ", " + msg['BCC'], msg.as_string())
+        s.quit()
+
+def send_postmark(to="", bcc="", subject="", plaintext=""):
+    """
+    Send an email via Postmark. Used when the application is deployed on
+    Heroku.
+    """
+    try:
+        message = PMMail(api_key = conf.POSTMARK_API_KEY,
+                        subject = subject,
+                        sender = conf.NOTIFICATION_EMAIL,
+                        text_body = plaintext)
+        message.to = to
+        if bcc != "":
+            message.bcc = bcc
+        message.send()
+    except Exception as e:
+        logger.exception("send_postmark raised:")
+        raise e
diff --git a/src/notifications/notifications.py b/src/notifications/notifications.py
new file mode 100644
index 00000000..86486074
--- /dev/null
+++ b/src/notifications/notifications.py
@@ -0,0 +1,67 @@
+#! /usr/bin/env python
+# -*- coding: utf-8 -*-
+
+# JARR - A Web based news aggregator.
+# Copyright (C) 2010-2016  Cédric Bonhomme - https://www.cedricbonhomme.org
+#
+# For more information : https://github.com/JARR-aggregator/JARR/
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as
+# published by the Free Software Foundation, either version 3 of the
+# License, or (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+import datetime
+from flask import render_template
+import conf
+from notifications import emails
+from web.lib.user_utils import generate_confirmation_token
+
+
+def information_message(subject, plaintext):
+    """
+    Send an information message to the users of the platform.
+    """
+    from web.models import User
+    users = User.query.all()
+    # Only send email for activated accounts.
+    user_emails = [user.email for user in users if user.enabled]
+    # Postmark has a limit of twenty recipients per message in total.
+    for i in xrange(0, len(user_emails), 19):
+        emails.send(to=conf.NOTIFICATION_EMAIL,
+                    bcc=", ".join(user_emails[i:i+19]),
+                    subject=subject, plaintext=plaintext)
+
+def new_account_notification(user):
+    """
+    Account creation notification.
+    """
+    token = generate_confirmation_token(user.email)
+    expire_time = datetime.datetime.now() + \
+                    datetime.timedelta(seconds=conf.TOKEN_VALIDITY_PERIOD)
+
+    plaintext = render_template('emails/account_activation.txt',
+                                    user=user, platform_url=conf.PLATFORM_URL,
+                                    token=token,
+                                    expire_time=expire_time)
+
+    emails.send(to=user.email, bcc=conf.NOTIFICATION_EMAIL,
+                subject="[JARR] Account creation", plaintext=plaintext)
+
+def new_password_notification(user, password):
+    """
+    New password notification.
+    """
+    plaintext = render_template('emails/new_password.txt',
+                                    user=user, password=password)
+    emails.send(to=user.email,
+                bcc=conf.NOTIFICATION_EMAIL,
+                subject="[JARR] New password", plaintext=plaintext)