initial commit

app/tasks/new_task.py

@@ -0,0 +1,73 @@
+#!/usr/bin/env python3
+
+import psycopg2
+from datetime import datetime
+from flask import Blueprint, redirect, url_for, flash, render_template
+from flask import current_app as app
+from flask_login import current_user
+from flask_wtf import FlaskForm
+from wtforms import SubmitField, TextAreaField
+from wtforms.validators import DataRequired
+from app.models import Task, Category
+
+newtask = Blueprint(
+    "newtask", __name__, template_folder="templates"
+)
+
+
+def insert_task(task):
+    conn = psycopg2.connect(
+        dbname=app.config['DATABASE_NAME'],
+        user=app.config['DATABASE_USER'],
+        host=app.config['DATABASE_HOST'],
+        password=app.config['DATABASE_PASSWORD']
+    )
+    cur = conn.cursor()
+    cur.execute("SELECT setval('task_id_seq', (SELECT MAX(id) FROM task))")
+    conn.commit()
+    cur.execute(
+        "SELECT MAX(priority) FROM task WHERE catid=%s AND contributor_id=%s AND done=%s",
+        (task.catid, task.contributor_id, False)
+    )
+    max_priority = cur.fetchone()[0]
+    task.priority = 1 if max_priority is None else max_priority + 1
+    cur.execute(
+        "SELECT count(id) FROM task WHERE content=%s AND contributor_id=%s",
+        (task.content, task.contributor_id)
+    )
+    if cur.fetchone()[0] == 0:
+        cur.execute(
+            "INSERT INTO task(content, contributor_id, catid, priority, timestamp) VALUES(%s,%s,%s,%s,%s)",
+            (task.content, task.contributor_id, task.catid, task.priority, task.timestamp)
+        )
+        conn.commit()
+    conn.close
+
+
+@newtask.route("/new-task/<int:category_id>", methods=["GET", "POST"])
+def new_task(category_id):
+    if current_user.is_anonymous:
+        return(redirect(url_for('cats.index')))
+    form = NewTask()
+    category = Category.query.get(category_id)
+    nl = (('cancel', url_for('tsks.tasks', category_id=category_id)), )
+    if form.validate_on_submit():
+        task = Task(
+            content=form.content.data,
+            contributor_id=current_user.id,
+            catid=category_id,
+            timestamp=str(datetime.utcnow()),
+        )
+        insert_task(task)
+        flash("Thanks for the new task!")
+        return(redirect(url_for('tsks.tasks', category_id=category_id)))
+    return render_template('new_task.html', title='New Task', form=form, navbar_links=nl, category=category)
+
+
+class NewTask(FlaskForm):
+    content = TextAreaField(
+        'New Task - Markdown Supported',
+        validators=[DataRequired(), ],
+        render_kw={'autofocus': True}
+    )
+    submit = SubmitField('Save')