app.py

from flask import Flask, render_template, flash, redirect, url_for, session, logging, request
# from data import Articles
from flask_mysqldb import MySQL
from wtforms import Form, StringField, TextAreaField, PasswordField, validators
from passlib.hash import sha256_crypt
from functools import wraps


app = Flask(__name__)

#Config MySQL
app.config['MYSQL_HOST'] = 'localhost'
app.config['MYSQL_USER'] = 'root'
app.config['MYSQL_PASSWORD'] = '4915494500'
app.config['MYSQL_DB'] = 'myflaskapp'
app.config['MYSQL_CURSORCLASS'] = 'DictCursor'

# init MySQL
mysql = MySQL(app)

# Articles = Articles()

def is_logged_in(f):
    @wraps(f)
    def wrap(*args, **kwargs):
        if 'logged_in' in session:
            return f(*args, **kwargs)
        else:
            flash('Unauthorized, Please Login', 'danger')
            return redirect(url_for('login'))
    return wrap

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

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

@app.route('/articles')
def articles():
    #open DB connection
    cur = mysql.connection.cursor()

    result = cur.execute("SELECT * FROM articles")

    articles = cur.fetchall()

    if result > 0:
        return render_template('articles.html', articles = articles)
    else:
        msg= 'No Articles Found'
        return render_template('articles.html', msg=msg)

    #close DB connection
    cur.close()


@app.route('/article/<string:id>')
def article(id):
    #open DB connection
    cur = mysql.connection.cursor()

    result = cur.execute("SELECT * FROM articles WHERE id=%s", [id])
    article = cur.fetchone()

    if result > 0:
        return render_template('article.html', article=article)
    else:
        flash("No Article Exist", "danger")
        return redirect(url_for('home'))


class RegisterForm(Form):
    name = StringField('Name', [validators.Length(min=1, max=50)])
    username = StringField('Username', [validators.Length(min=4, max=25)])
    email = StringField('Email', [validators.Length(min=6, max=50)])
    password = PasswordField('Password', [
        validators.DataRequired(),
        validators.EqualTo('confirm', message="Password do not match"),
    ])
    confirm = PasswordField('Confirm Password')

@app.route('/register', methods=['GET', 'POST'])
def register():
    form = RegisterForm(request.form)
    if request.method == 'POST' and form.validate():
        name = form.name.data
        email = form.email.data
        username = form.username.data
        password = sha256_crypt.encrypt(str(form.password.data))

        #create the cursor

        cur = mysql.connection.cursor()
        cur.execute("INSERT INTO users(name, email, username, password) VALUES(%s, %s, %s, %s)", (name, email, username, password))

        #commit to db

        mysql.connection.commit()

        #close connection

        cur.close()

        flash('You are now registered and can log in', 'success')

        return redirect(url_for('login'))
    return render_template('register.html', form=form)

#User login

@app.route('/login', methods=['GET', 'POST'])
def login():
    if request.method == 'POST':
        #Get Form Fields
        username = request.form['username']
        password_candidate = request.form['password']

        # Create cursor
        cur = mysql.connection.cursor()

        # Get user by username

        result = cur.execute("SELECT * FROM users WHERE username  = %s", [username])

        if result > 0:
            #Get the stored hash
            data = cur.fetchone()
            password = data['password']

            #compare password
            if sha256_crypt.verify(password_candidate, password):
                session['logged_in'] = True
                session['username'] = username

                flash('You are now logged in', 'success')

                return redirect(url_for('dashboard'))
            else:
                error = "Invalid Password"
                return render_template('login.html', error=error)
            #cursor closed
            cur.closed()
        else:
            error = "Invalid Username"
            return render_template('login.html', error=error)


    return render_template('login.html')

#check to see if the user is logged in


@app.route('/dashboard')
@is_logged_in
def dashboard():
    #opening DB connection
    cur = mysql.connection.cursor()

    #get articles
    result = cur.execute("SELECT * FROM articles")

    articles = cur.fetchall()

    if result > 0:
        return render_template('dashboard.html', articles=articles)
    else:
        msg= 'No Articles Found'
        return render_template('dashboard.html', msg=msg)

    #closing DB connection
    cur.close()



@app.route('/logout')
@is_logged_in
def logout():
    session.clear()
    flash('You have logged out', 'success')

    return redirect(url_for('index'))

#Adding and Editing Article form
class ArticleForm(Form):
    title=StringField('Title', [validators.Length(min=1,max=200)])
    body=TextAreaField('Body', [validators.Length(min=30)])
#Add article
@app.route('/add_article', methods=["GET", "POST"])
@is_logged_in
def add_article():
    form = ArticleForm(request.form)
    if request.method == "POST" and form.validate():
        title = form.title.data
        body = form.body.data

        #create cursor
        cur = mysql.connection.cursor()

        cur.execute("INSERT INTO articles(title, body, author) VALUES(%s, %s, %s)",(title, body, session['username']))

        #commit to DB

        mysql.connection.commit()

        #close connection
        cur.close()

        flash('Article Created', 'success')

        return redirect(url_for('dashboard'))

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

@app.route('/edit_article/<string:id>', methods=["GET", "POST"])
@is_logged_in
def edit_article(id):
    #open DB cursor

    cur = mysql.connection.cursor()

    #get article by its id

    result = cur.execute("SELECT * FROM articles WHERE id =%s", [id])
    article = cur.fetchone()

    form = ArticleForm(request.form)

    #populate form fields

    form.title.data = article['title']
    form.body.data = article['body']

    if request.method == "POST" and form.validate():
        title = request.form['title']
        body = request.form['body']

        #create cursor
        cur = mysql.connection.cursor()

        cur.execute("UPDATE articles SET title=%s, body=%s WHERE id = %s", (title, body, id))

        #commit to DB

        mysql.connection.commit()

        #close connection
        cur.close()

        flash('Article Updated', 'success')

        return redirect(url_for('dashboard'))

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

@app.route('/delete_article/<string:id>', methods=["POST"])
@is_logged_in
def delete_article(id):
    cur = mysql.connection.cursor()

    cur.execute("DELETE FROM articles WHERE id = %s", [id])

    #commit to DB

    mysql.connection.commit()

    #close connection
    cur.close()

    flash('Article Deleted', 'success')

    return redirect(url_for('dashboard'))




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