| 
 | 1 | +from datetime import datetime  | 
 | 2 | + | 
 | 3 | +from flask import Flask, render_template, request, redirect, url_for, session  | 
 | 4 | +from passlib.hash import pbkdf2_sha256  | 
 | 5 | + | 
 | 6 | +from model import Task, User  | 
 | 7 | + | 
 | 8 | +app = Flask(__name__)  | 
 | 9 | + | 
 | 10 | + | 
 | 11 | +@app.route('/all')  | 
 | 12 | +def all_tasks():  | 
 | 13 | +    return render_template('all.jinja2', tasks=Task.select())  | 
 | 14 | + | 
 | 15 | + | 
 | 16 | +@app.route('/incomplete', methods=['GET', 'POST'])  | 
 | 17 | +def incomplete_tasks():  | 
 | 18 | +    if 'username' not in session:  | 
 | 19 | +        return redirect(url_for('login'))  | 
 | 20 | + | 
 | 21 | +    if request.method == 'POST':  | 
 | 22 | +        user = User.select().where(User.name == session['username']).get()  | 
 | 23 | + | 
 | 24 | +        Task.update(performed=datetime.now(), performed_by=user)\  | 
 | 25 | +            .where(Task.id == request.form['task_id'])\  | 
 | 26 | +            .execute()  | 
 | 27 | + | 
 | 28 | +    return render_template('incomplete.jinja2', tasks=Task.select().where(Task.performed.is_null()))  | 
 | 29 | + | 
 | 30 | + | 
 | 31 | +@app.route('/create', methods=['GET', 'POST'])  | 
 | 32 | +def create():  | 
 | 33 | +    if 'username' not in session:  | 
 | 34 | +        return redirect(url_for('login'))  | 
 | 35 | + | 
 | 36 | +    if request.method == 'POST':  | 
 | 37 | +        task = Task(name=request.form['name'])  | 
 | 38 | +        task.save()  | 
 | 39 | + | 
 | 40 | +        return redirect(url_for('all_tasks'))  | 
 | 41 | +    else:  | 
 | 42 | +        return render_template('create.jinja2')  | 
 | 43 | + | 
 | 44 | + | 
 | 45 | +@app.route('/login', methods=['GET', 'POST'])  | 
 | 46 | +def login():  | 
 | 47 | +    if request.method == 'POST':  | 
 | 48 | +        user = User.select().where(User.name == request.form['name']).get()  | 
 | 49 | + | 
 | 50 | +        if user and pbkdf2_sha256.verify(request.form['password'], user.password):  | 
 | 51 | +            session['username'] = request.form['name']  | 
 | 52 | +            return redirect(url_for('all_tasks'))  | 
 | 53 | + | 
 | 54 | +        return render_template('login.jinja2', error="Incorrect username or password.")  | 
 | 55 | + | 
 | 56 | +    else:  | 
 | 57 | +        return render_template('login.jinja2')  | 
 | 58 | + | 
 | 59 | + | 
 | 60 | +@app.route('/logout')  | 
 | 61 | +def logout():  | 
 | 62 | +    session.pop('username', None)  | 
 | 63 | +    return redirect(url_for('all_tasks'))  | 
 | 64 | + | 
 | 65 | +app.secret_key = b'\x9d\xb1u\x08%\xe0\xd0p\x9bEL\xf8JC\xa3\xf4J(hAh\xa4\xcdw\x12S*,u\xec\xb8\xb8'  | 
 | 66 | + | 
 | 67 | + | 
 | 68 | +if __name__ == "__main__":  | 
 | 69 | +    app.run()  | 
0 commit comments