app.py

"""
KULDEEP SINGH
"""

from flask import Flask, request, g, render_template, Response, jsonify
from flask import session, flash, redirect, url_for, make_response

from flask_pymongo import PyMongo
from flask_mail import Mail, Message
import config_ext
import os

from celery import Celery

app = Flask(__name__)

# --- MongoDB configs. --- #

app.config['MONGO_DBNAME'] = config_ext.mongo_dbname
app.config['MONGO_URI'] = config_ext.mongo_uri

mongo = PyMongo(app)

# Flask-Mail configuration
app.config['MAIL_SERVER'] = 'smtp.googlemail.com'
app.config['MAIL_PORT'] = 587
app.config['MAIL_USE_TLS'] = True
app.config['MAIL_USERNAME'] = os.environ.get('MAIL_USERNAME')
app.config['MAIL_PASSWORD'] = os.environ.get('MAIL_PASSWORD')
app.config['MAIL_DEFAULT_SENDER'] = 'testskuldeep@gmail.com'

mail = Mail(app)

# Celery configuration
app.config['CELERY_BROKER_URL'] = 'redis://localhost:6379/0'
app.config['CELERY_RESULT_BACKEND'] = 'redis://localhost:6379/0'

celery = Celery(app.name, broker=app.config['CELERY_BROKER_URL'])
celery.conf.update(app.config)


@app.route('/')
def index():
    '''
    Entry point.
    '''
    message = ""
    return render_template("index.html", message=message)


@app.errorhandler(404)
def not_found(error):
    '''
    Handles if 404 occurs.
    '''
    return render_template("404.html")


@celery.task()
def send_async_emails(title, sender, recipients, body):
    '''
    Method finds subscribed users and sends them a mail about the
    changes in the database.
    '''
    with app.app_context():
        msg = Message(title, sender=sender,
                      recipients=recipients, body=body)

        mail.send(msg)


@app.route('/subscribed', methods=['GET', 'POST'])
def subscribe():
    '''
    Subscription to the email notifications.
    '''
    if request.method == 'POST':
        user_name = request.form['user_name']
        user_email = request.form['user_email']

        if request.form.getlist('activate_subscription'):
            user_activated_subscription = True
        else:
            user_activated_subscription = False

        db = mongo.db.MongoNotify
        db.create_index("user_email", unique=True)

        try:
            db.insert(
                {
                    "user_name": user_name,
                    "user_email": user_email,
                    "user_activated_subscription": user_activated_subscription
                })

            subscribers = db.find({"user_activated_subscription": True})
            recipients = []
            for s in subscribers:
                user_info = list(s.values())
                user_email = user_info[2]
                recipients.append(user_email)

            if recipients:
                title = "Hi subscribed user"
                sender = 'testskuldeep@gmail.com'
                body = "There was a new entry in the database."
                send_async_emails.delay(title, sender, recipients, body)

            print (user_activated_subscription)

            return render_template("subscribed.html", user_name=user_name,
                                   user_activated_subscription=user_activated_subscription)
        except:
            message = "Email already taken."
            return render_template("index.html", message=message)

    return render_template("index.html")


@app.route('/unsubscribe', methods=['GET', 'POST'])
def unsubscribe():
    '''
    Unsubscribe from the notifications.
    '''

    return render_template("unsubscribe.html")


if __name__ == "__main__":
    app.run()