main.py

from flask import render_template, redirect, g, Flask, request, url_for, session
from sqlite3 import dbapi2 as sqlite3
from os import path
app = Flask(__name__)


app.config.from_pyfile('config.py')


def get_db():
    db = getattr(g, '_database', None)
    if db is None:
        db_file = app.config["DATABASE"]
        exists = path.isfile(db_file)
        db = g._database = sqlite3.connect(app.config['DATABASE'])
        if not exists:
            qry = open('schema.sql', 'r').read()
            c = db.cursor()
            c.executescript(qry)
            db.commit()
            c.close()

    # Make sure database has tables
    return db


def cursor_to_dict_array(cur):
    columns = [column[0] for column in cur.description]
    return [dict(zip(columns, row)) for row in cur.fetchall()]


@app.teardown_appcontext
def close_connection(exception):
    db = getattr(g, '_database', None)
    if db is not None:
        db.close()


@app.route('/', methods=["GET"])
def base():
    if not session.get('logged_in'):
        return redirect("/login")
    return redirect("/home")


@app.route('/login', methods=["GET", "POST"])
def login():
    error = None
    if request.method == "POST":
        if request.form['username'] != app.config['USERNAME']:
            error = "Invalid Username"
        elif request.form['password'] != app.config['PASSWORD']:
            error = "Invalid Username"
        else:
            session['logged_in'] = True
            return redirect(url_for('home'))
    return render_template('login.html', error=error)


@app.route('/logout')
def logout():
    session.pop('logged_in', None)
    return redirect("/login")


@app.route('/home')
def home():
    return redirect("/home/active")


@app.route('/home/active')
def home_active():
    if not session.get('logged_in'):
        return redirect("/login")
    # load completed projects
    db = get_db()
    cur = db.execute('select * from projects where is_complete=0')
    entries = cursor_to_dict_array(cur)
    # render home page with completed projects
    return render_template('home.html', type="active", entries=entries)


@app.route('/home/completed')
def home_completed():
    if not session.get('logged_in'):
        return redirect("/login")
    # load completed projects
    db = get_db()
    cur = db.execute('select * from projects where is_complete=1')
    entries = cursor_to_dict_array(cur)
    # render home page with completed projects
    return render_template('home.html', type="completed", entries=entries)


@app.route("/projects", methods=["POST"])
def add_project():
    if not session.get('logged_in'):
        return redirect("/login")
    db = get_db()
    db.execute(
        'insert into projects (contact_email,contact_name,contract_type,quote_dollars,is_complete) values (?,?,?, ?, ?)',
        (
            request.form.get('contact_email', type=str),
            request.form.get('contact_name', type=str),
            'hourly' if request.form.get(
                'contract_type') == "hourly" else "milestone",
            request.form.get('quote_dollars', type=int),
            0 if not request.form.get('is_complete') else 1
        )
    )
    db.commit()
    db.close()
    return redirect(url_for('home'))


@app.route("/home/projects/<int:project_id>", methods=["GET"])
def edit_project_page(project_id):
    if not session.get('logged_in'):
        return redirect("/login")
    # Load particular project
    db = get_db()
    cur = db.execute("select * from projects where id=?", (int(project_id),))
    entries = cursor_to_dict_array(cur)
    if len(entries) == 0:
        return redirect('/home')

    # Return page
    return render_template('edit_form.html', project=entries[0])


@app.route("/projects/edit/<int:project_id>", methods=["POST"])
def edit_project(project_id):
    if not session.get('logged_in'):
        return redirect("/login")
    # Save the information into the database
    db = get_db()
    db.execute(
        '''UPDATE projects SET contact_email=?, contact_name=?, contract_type=?, quote_dollars=?, is_complete=? where id=?''',
        (
            request.form.get('contact_email', type=str),
            request.form.get('contact_name', type=str),
            'hourly' if request.form.get(
                'contract_type') == "hourly" else "milestone",
            request.form.get('quote_dollars', type=int),
            0 if not request.form.get('is_complete') else 1,
            project_id
        )
    )
    db.commit()
    db.close()

    return redirect(url_for('home'))