main.py

from flask import Flask, render_template, request, session, redirect
from flask_sqlalchemy import SQLAlchemy
from flask_mail import Mail
from werkzeug.utils import secure_filename
import json
import os
import math
from datetime import datetime

with open("config.json", "r") as c:
    params = json.load(c)["params"]

app = Flask(__name__)

app.secret_key = 'super-secret-key'

app.config['UPLOAD_FOLDER'] = params['upload_location']

app.config.update(
    MAIL_SERVER = 'smtp.gmail.com',
    MAIL_PORT = '465',
    MAIL_USE_SSL = True,
    MAIL_USERNAME = params['gmail-user'],
    MAIL_PASSWORD = params['gmail-pass']
)

mail = Mail(app)

local_server = True

if (local_server):
    app.config['SQLALCHEMY_DATABASE_URI'] = params["local_uri"]

else:
    app.config['SQLALCHEMY_DATABASE_URI'] = params["prod_uri"]

db = SQLAlchemy(app)


class Contacts(db.Model):
    """ sno, name, phone_num, msg, date, email
    """
    sno = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(80), nullable=False)
    phone_num = db.Column(db.String(12), nullable=False)
    msg = db.Column(db.String(120), nullable=False)
    date = db.Column(db.String(12), nullable=True)
    email = db.Column(db.String(20), nullable=False)


class Posts(db.Model):
    """ sno, name, phone_num, msg, date, email
    """
    sno = db.Column(db.Integer, primary_key=True)
    title = db.Column(db.String(80), nullable=False)
    slug = db.Column(db.String(21), nullable=False)
    content = db.Column(db.String(120), nullable=False)
    date = db.Column(db.String(12), nullable=True)
    img_file = db.Column(db.String(12), nullable=True)
    tagline = db.Column(db.String(12), nullable=True)


@app.route('/')
def home():
    posts = Posts.query.filter_by().all()
    last = math.ceil(len(posts) / int(params['no_of_post']))
    #[0: params['no_of_post']]
    page = request.args.get('page')
    if (not str(page).isnumeric()):
        page = 1
    page = int(page)
    posts = posts[(page - 1) * int(params['no_of_post'])
                  : (page - 1) * int(params['no_of_post']) +
                    int(params['no_of_post'])]

    if (page == 1):
        prev = '#'
        next = '/?page=' + str(page + 1)

    elif (page == last):
        prev = '/?page=' + str(page - 1)
        next = '#'

    else:
        prev = '/?page=' + str(page - 1)
        next = '/?page=' + str(page + 1)



    return  render_template('index.html', params = params, posts = posts , prev = prev, next = next)


@app.route('/post/<string:post_slug>', methods = ['GET'])
def post_route(post_slug):
    post = Posts.query.filter_by(slug = post_slug).first()
    return render_template('post.html',  params = params, post = post)


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


@app.route('/post')
def post():
    return render_template('post.html', params = params)


@app.route('/dashboard', methods = ['GET', 'POST'])
def dashboard():
    if  ('user' in session and session['user'] == params['admin_user']):
        posts = Posts.query.all()
        return render_template('dashboard.html', params = params, posts = posts)

    if request.method == 'POST':
        username = request.form.get('uname')
        userpass = request.form.get('pass')
        if (username == params['admin_user'] and userpass == params['admin_pass']):
            #set the session variable
            session['user'] = username
            posts = Posts.query.all()
            return render_template('dashboard.html', params = params, posts = posts)

    return render_template('login.html', params = params)


@app.route('/edit/<string:sno>', methods = ["GET", "POST"])
def edit(sno):
    if ('user' in session and session['user'] == params['admin_user']):
        if request.method == 'POST':
            box_title = request.form.get('title')
            tline = request.form.get('tline')
            slug = request.form.get('slug')
            content = request.form.get('content')
            img_file = request.form.get('img_file')
            date = datetime.now()

            if sno == '0':
                entry = Posts(title = box_title, tagline = tline, slug = slug, content = content, img_file = img_file, date = date)
                db.session.add(entry)
                db.session.commit()
            else:
                post = Posts.query.filter_by(sno = sno).first()
                post.title = box_title
                post.slug = slug
                post.content = content
                post.tagline = tline
                post.img_file = img_file
                post.date = date
                db.session.commit()
                return redirect('/edit/'+sno)
        post = Posts.query.filter_by(sno=sno).first()
        return render_template('edit.html', params = params, post = post)


@app.route('/uploader', methods = ["GET", "POST"])
def uploader():
    if ('user' in session and session['user'] == params['admin_user']):
        if request.method == 'POST':
            f = request.files['file1']
            f.save(os.path.join(app.config['UPLOAD_FOLDER'], secure_filename(f.filename) ))
            return "Uploaded succesfully"


@app.route('/logout')
def logout():
    session.pop('user')
    return redirect('/dashboard')

@app.route('/delete/<string:sno>', methods = ["GET", "POST"])
def delete(sno):
    post = Posts.query.filter_by(sno = sno).first()
    db.session.delete(post)
    db.session.commit()
    return redirect('/dashboard')

@app.route('/contact', methods = ["GET", "POST"])
def contact():
    if request.method == "POST":
        """
        ADD ENTRY TO THE DATABASE
        """
        name = request.form.get("name")
        email = request.form.get("email")
        phone = request.form.get("phone")
        message = request.form.get("message")
        """ 
        sno, name, phone_num, msg, date, email
        """
        entry = Contacts(name = name, email = email, phone_num = phone, msg = message, date = datetime.now())
        db.session.add(entry)
        db.session.commit()
        mail.send_message('New message from {}'.format(name),
                          sender = email,
                          recipients = [params['gmail-user']],
                          body = message + "\n" + phone
                          )
    return render_template('contact.html', params = params)

if __name__== '__main__':
    app.run(debug=True)