mirror of
https://github.com/TrentSPalmer/todo_app_flask.git
synced 2024-11-24 17:11:29 -08:00
75 lines
3.0 KiB
Python
75 lines
3.0 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import psycopg2
|
||
|
|
from flask import Blueprint, redirect, url_for, request, flash
|
||
|
|
from flask import current_app
|
||
|
|
from app.models import Task
|
||
|
|
from flask_login import current_user
|
||
|
|
from .task_action import make_move_up_down_bools
|
||
|
|
from .. import db
|
||
|
|
|
||
|
|
reorderp = Blueprint(
|
||
|
|
"reorderp", __name__, template_folder="templates"
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
@reorderp.route('/move-task/<int:taskid>')
|
||
|
|
def move_task(taskid):
|
||
|
|
task = Task.query.get(taskid)
|
||
|
|
if task is None or current_user.is_anonymous or task.contributor_id != current_user.id:
|
||
|
|
return(redirect(url_for('cats.index')))
|
||
|
|
make_move_up_down_bools(task)
|
||
|
|
priority = task.priority
|
||
|
|
if request.args['move'] == 'up' and task.can_move_up:
|
||
|
|
other_task = Task.query.filter(
|
||
|
|
Task.catid == task.catid,
|
||
|
|
Task.contributor_id == task.contributor_id,
|
||
|
|
Task.done == task.done, Task.priority > task.priority).order_by(Task.priority).first()
|
||
|
|
other_priority = other_task.priority
|
||
|
|
other_task.priority = task.priority
|
||
|
|
task.priority = other_priority
|
||
|
|
if request.args['move'] == 'down' and task.can_move_down:
|
||
|
|
other_task = Task.query.filter(
|
||
|
|
Task.catid == task.catid,
|
||
|
|
Task.contributor_id == task.contributor_id,
|
||
|
|
Task.done == task.done, Task.priority < task.priority).order_by(Task.priority.desc()).first()
|
||
|
|
other_priority = other_task.priority
|
||
|
|
other_task.priority = task.priority
|
||
|
|
task.priority = other_priority
|
||
|
|
if request.args['move'] == 'top' and task.can_move_top:
|
||
|
|
other_task = Task.query.filter(
|
||
|
|
Task.catid == task.catid,
|
||
|
|
Task.contributor_id == task.contributor_id,
|
||
|
|
Task.done == task.done).order_by(Task.priority.desc()).first()
|
||
|
|
task.priority = other_task.priority + 1
|
||
|
|
if request.args['move'] == 'end' and task.can_move_end:
|
||
|
|
task.priority = 0
|
||
|
|
db.session.commit()
|
||
|
|
if request.args['move'] == 'end' or request.args['move'] == 'top':
|
||
|
|
reorder_priorities(task.catid, task.contributor_id, task.done, current_app.config)
|
||
|
|
flash("task {} is moved {}".format(priority, request.args['move']))
|
||
|
|
ru = 'tsks.{}tasks'.format('hidden_' if task.done else '')
|
||
|
|
return(redirect(url_for(ru, category_id=task.catid)))
|
||
|
|
|
||
|
|
|
||
|
|
def reorder_priorities(catid, conid, done, app_config):
|
||
|
|
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 COUNT(id) FROM task WHERE catid=%s AND contributor_id=%s AND done=%s",
|
||
|
|
(catid, conid, done))
|
||
|
|
if cur.fetchone()[0] > 1:
|
||
|
|
cur.execute(
|
||
|
|
"SELECT id FROM task WHERE catid=%s AND contributor_id=%s AND done=%s ORDER BY priority",
|
||
|
|
(catid, conid, done))
|
||
|
|
ids = [x[0] for x in cur.fetchall()]
|
||
|
|
for i, task in enumerate(ids, 1):
|
||
|
|
cur.execute("UPDATE task SET priority=%s WHERE id=%s", (i, task))
|
||
|
|
conn.commit()
|
||
|
|
conn.close()
|