app.py

import os
from flask import Flask, render_template, redirect, request, url_for, \
    session, flash
from flask_pymongo import PyMongo, pymongo
from bson.objectid import ObjectId
import bcrypt
from datetime import datetime
from flask_mail import Mail, Message
from forms import ContactForm
from flask_wtf import Form

app = Flask(__name__)
# EMAIL SETTINGS

app.config.update(
    DEBUG=True,
    MAIL_SERVER='smtp.gmail.com',
    MAIL_PORT=465,
    MAIL_USE_SSL=True,
    MAIL_USERNAME=os.environ.get('MAIL_USERNAME'),
    MAIL_PASSWORD=os.environ.get('MAIL_PASSWORD'),
    )
mail = Mail(app)

app.config['SECRET_KEY'] = os.environ.get('SECRET_KEY')
app.config['MONGO_DBNAME'] = os.environ.get('MONGO_DBNAME')
app.config['MONGO_URI'] = os.environ.get('MONGO_URI')
hash_salt = os.environ.get('HASH_SALT')

date = datetime.now().strftime('%Y-%m-%d  %H:%M:%S')

mongo = PyMongo(app)

# MongoDB collections

users_collection = mongo.db.users
posts_collection = mongo.db.posts
categories_collection = mongo.db.categories
# Routes


@app.route('/', methods=['POST', 'GET'])
def home():
    categories = list(categories_collection.find())
    posts = posts_collection.find().sort('_id', pymongo.DESCENDING)
    return render_template('home.html', posts=posts, categories=categories)


@app.route('/login', methods=['POST', 'GET'])
def login():
    if request.method == 'POST':
        users_collection = mongo.db.users
        login_user = users_collection.find_one({'name': request.form[
            'username']})

        if login_user:
            if bcrypt.hashpw(request.form['password'].encode('utf-8'),
               hash_salt.encode('utf-8')) == login_user['password']:
                session['username'] = request.form['username']
                return redirect(url_for('home'))
        return render_template('signup.html')
    return render_template('login.html')


@app.route('/register', methods=['POST', 'GET'])
def signup():
    if request.method == 'POST':
        users_collection = mongo.db.users
        existing_user = users_collection.find_one({'name': request.form[
            'username']})

        if existing_user is None:
            hashpass = bcrypt.hashpw(request.form[
                'password'].encode('utf-8'), hash_salt.encode('utf-8'))
            users_collection.insert({'name': request.form[
                'username'], 'password': hashpass})
            session['username'] = request.form.get('username')
            return redirect('/loggedin/' + session['username'])
    return render_template('signup.html')


@app.route('/loggedin/<username>', methods=['GET', 'POST'])
def loggedin(username):
    posts = posts_collection.find({'added_by': session[
        'username']}).sort('_id', pymongo.DESCENDING)
    return render_template('profile.html', username=session[
        'username'], posts=posts)


@app.route('/logout')
def logout():
    session.clear()
    return redirect(url_for('home'))


@app.route('/add_post')
def add_post():
    return render_template('addpost.html',
                           categories=categories_collection.find())


@app.route('/insert_post', methods=['POST'])
def insert_post():
    posts = posts_collection
    posts.insert_one({
        'category': request.form.get('category_name'),
        'name': request.form.get('name'),
        'make': request.form.get('makes'),
        'year_modal': request.form.get('year'),
        'image': request.form.get('image'),
        'quantity': request.form.get('quantity'),
        'technics': request.form.getlist('technic'),
        'contact': request.form.get('contact'),
        'condition': request.form.get('condition'),
        'price': request.form.get('price'),
        'added_by': session['username'],
        'date': datetime.utcnow().strftime('%Y-%m-%d  %H:%M:%S'),
        })
    return redirect(url_for('loggedin', username=session['username']))


@app.route('/update_post/<post_id>', methods=['POST'])
def update_post(post_id):
    posts_collection.update({'_id': ObjectId(post_id)}, {
        'category': request.form.get('category_name'),
        'name': request.form.get('name'),
        'make': request.form.get('makes'),
        'year_modal': request.form.get('year'),
        'image': request.form.get('image'),
        'quantity': request.form.get('quantity'),
        'technics': request.form.getlist('technic'),
        'contact': request.form.get('contact'),
        'condition': request.form.get('condition'),
        'price': request.form.get('price'),
        'added_by': session['username'],
        'date': datetime.utcnow().strftime(' %Y-%m-%d  %H:%M:%S'),
        })
    return redirect(url_for('loggedin', username=session['username']))


@app.route('/delete_post/<post_id>')
def delete_post(post_id):
    posts_collection.remove({'_id': ObjectId(post_id)})
    return redirect(url_for('loggedin', username=session['username']))


@app.route('/post/<post_id>')
def post(post_id):
    post = posts_collection.find_one({'_id': ObjectId(post_id)})
    return render_template('post.html', post=post)


@app.route('/<category_name>', methods=['GET'])
def filter_list(category_name):
    categories = list(categories_collection.find())
    category_name = \
        categories_collection.find_one({'category_name': category_name})
    posts = posts_collection.find()
    return render_template('filter.html', categories=categories,
                           category_name=category_name, posts=posts)


@app.route('/categories')
def categories():
    categories = categories_collection.find().sort('_id', pymongo.ASCENDING)
    return render_template('categories.html', categories=categories)


@app.route('/add_category')
def add_category():
    return render_template('addcategory.html')


@app.route('/insert_category', methods=['POST'])
def insert_category():
    category_doc = {'category_name': request.form['category_name']}
    categories_collection.insert_one(category_doc)
    return redirect(url_for('categories'))


@app.route('/delete_category/<category_id>')
def delete_category(category_id):
    categories_collection.remove({'_id': ObjectId(category_id)})
    return redirect(url_for('categories'))


@app.route('/edit_post/<post_id>')
def edit_post(post_id):
    post = posts_collection.find_one({'_id': ObjectId(post_id)})
    _categories = categories_collection.find()
    category_list = [category for category in _categories]
    return render_template('editpost.html', post=post,
                           categories=category_list)


@app.route('/edit_category/<category_id>')
def edit_category(category_id):
    return render_template('editcategory.html',
                           category=categories_collection.find_one({
                               '_id': ObjectId(category_id)}))


@app.route('/update_category/<category_id>', methods=['POST'])
def update_category(category_id):
    categories_collection.update({'_id': ObjectId(category_id)},
                                 {'category_name': request.form[
                                     'category_name']})
    return redirect(url_for('categories'))


@app.route('/about')
def about():
    return render_template('about.html')


@app.route('/contact', methods=['GET', 'POST'])
def contact():
    form = ContactForm()

    if request.method == 'POST':
        if form.validate() is False:
            flash('All fields are required.')
            return render_template('contact.html', form=form)
        else:
            msg = Message(form.subject.data,
                          sender='contact@example.com',
                          recipients=['frankmasabo55@gmail.com'])
            msg.body = """
    From: %s <%s>
    %s
    """ \
                % (form.name.data, form.email.data, form.message.data)
            mail.send(msg)

            return render_template('contact.html', success=True)
    elif request.method == 'GET':

        return render_template('contact.html', form=form)


@app.template_filter('formatdatetime')
def format_datetime(value, format='%d %b %Y %I:%M '):
    if value is None:
        return ''
    return value.strftime(format)



if __name__ == '__main__':
    app.run(
        host=os.environ.get('IP'),
        port=int(os.environ.get('PORT')),
        debug=False)