74 lines
2.5 KiB
Python
74 lines
2.5 KiB
Python
#!/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')
|