doneth.at/app/main.py

from . import timeutils
from flask import Blueprint, render_template, redirect, url_for, abort, request
from flask_login import current_user, login_required
from flask_wtf import FlaskForm
from wtforms import StringField, SubmitField, IntegerField
from wtforms.validators import DataRequired, Length, NumberRange
from .db import db, Accomplishment
from datetime import datetime, timedelta
import time

main = Blueprint('main', __name__)


class NewAccomplishementForm(FlaskForm):
    text = StringField(
        'Accomplishment',
        validators=[
            DataRequired(), Length(max=256)])
    submit_5 = SubmitField('5 XP')
    submit_10 = SubmitField('10 XP')
    submit_15 = SubmitField('15 XP')


def handle_accomplishment_submission(form):
    accomplishment = Accomplishment()
    accomplishment.user_id = current_user.id
    accomplishment.text = form.text.data
    accomplishment.difficulty = 5
    if form.submit_10.data:
        accomplishment.difficulty = 10
    elif form.submit_15.data:
        accomplishment.difficulty = 15
    # the timestamp should be set by the database
    db.session.add(accomplishment)
    db.session.commit()
    return redirect(url_for('main.index'))


def parse_day(day_string):
    day_datetime = None
    if day_string == "today":
        day_datetime = timeutils.today()
    else:
        day_datetime = timeutils.from_str(day_string)

    day_string_clean = timeutils.as_str(day_datetime)
    return {
        "datetime": day_datetime,
        "string": day_string_clean,
        "fancy": timeutils.as_fancy_str(day_datetime),
        "is_today": timeutils.is_today(day_datetime)
    }


def get_day_template_data(day_string):
    day = parse_day(day_string)
    day_datetime = day["datetime"]

    accomplishments = list(reversed(
        Accomplishment.get_day(current_user.id, day_datetime)))
    total = sum(a.difficulty for a in accomplishments)

    yesterday = timeutils.day_before(day_datetime)
    tomorrow = timeutils.day_after(day_datetime)
    if timeutils.is_future(tomorrow):
        tomorrow = None

    return {
        "day": day,
        "links": {
            "yesterday": url_for('main.index', day=timeutils.as_str(yesterday)),
            "tomorrow": url_for('main.index', day=timeutils.as_str(tomorrow)) if tomorrow is not None else None
        },
        "accomplishments": accomplishments,
        "total_xp": sum(a.difficulty for a in accomplishments),
        "ts": int(time.time()),  # timestamp for cachebusting
    }


@main.route('/', defaults={'day': 'today'}, methods=['GET', 'POST'])
@main.route('/day/<day>')
def index(day):
    if not current_user.is_authenticated:
        return render_template('index.html')

    form = NewAccomplishementForm()
    if form.validate_on_submit():
        return handle_accomplishment_submission(form)

    return render_template(
        'main/app.html',
        form=form,
        **get_day_template_data(day)
    )


@main.route('/day/<day>/edit')
@login_required
def edit_day(day):
    form = NewAccomplishementForm()
    if form.validate_on_submit():
        return handle_accomplishment_submission(form)

    return render_template(
        'main/app.html',
        form=form,
        edit=True,
        **get_day_template_data(day)
    )


class DeleteForm(FlaskForm):
    submit = SubmitField('Delete')


@main.route('/accomplishment/<accomplishment_id>/delete', methods=['GET', 'POST'])
@login_required
def delete_accomplishment(accomplishment_id):
    a = Accomplishment.query.get_or_404(accomplishment_id)
    if a.user_id != current_user.id:
        abort(403)

    back_url = url_for(
        'main.edit_day', day=timeutils.as_str(timeutils.day(a.time)))

    form = DeleteForm()
    if form.validate_on_submit():
        db.session.delete(a)
        db.session.commit()
        return redirect(back_url)

    return render_template(
        'main/delete.html',
        form=form,
        accomplishment=a,
        cancel=back_url)


class EditForm(FlaskForm):
    text = StringField(
        'Accomplishment',
        validators=[DataRequired(), Length(max=256)]
    )
    difficulty = IntegerField(
        'Difficulty (XP)',
        validators=[DataRequired(), NumberRange(max=100, min=-100)]
    )
    submit = SubmitField('Save')


@main.route('/accomplishment/<accomplishment_id>/edit', methods=['GET', 'POST'])
@login_required
def edit_accomplishment(accomplishment_id):
    a = Accomplishment.query.get_or_404(accomplishment_id)
    if a.user_id != current_user.id:
        abort(403)

    back_url = url_for(
        'main.edit_day', day=timeutils.as_str(timeutils.day(a.time)))

    form = EditForm(obj=a)
    if form.validate_on_submit():
        a.text = form.text.data
        a.difficulty = form.difficulty.data
        db.session.commit()
        return redirect(back_url)

    return render_template('main/edit.html', form=form, cancel=back_url)


@main.route('/day/<day>/add', methods=['GET', 'POST'])
@login_required
def add_day(day):
    day_parsed = parse_day(day)
    form = EditForm()

    back_url = ""

    from_top = ("from" in request.args) and ("top" in request.args["from"])
    back_to_day = url_for('main.index', day=day_parsed["string"])
    back_to_edit = url_for('main.edit_day', day=day_parsed["string"])

    if form.validate_on_submit():
        accomplishment = Accomplishment()
        accomplishment.user_id = current_user.id
        accomplishment.text = form.text.data
        accomplishment.difficulty = form.difficulty.data
        accomplishment.time = timeutils.from_str(day)
        db.session.add(accomplishment)
        db.session.commit()
        return redirect(back_to_day)

    return render_template(
        'main/edit.html',
        day=day_parsed,
        form=form,
        edit=True,
        cancel=back_to_day if from_top else back_to_edit
    )