labertasche/labertasche/blueprints/bp_dashboard/comments.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# /**********************************************************************************
#  * _author  : Domeniko Gentner
#  * _mail    : code@tuxstash.de
#  * _repo    : https://git.tuxstash.de/gothseidank/labertasche
#  * _license : This project is under MIT License
#  *********************************************************************************/
from . import bp_dashboard
from flask import render_template, request, redirect, url_for
from flask_login import login_required
from flask_cors import cross_origin
from labertasche.database import labertasche_db as db
from labertasche.models import TLocation, TComments
from labertasche.helper import export_location, get_id_from_project_name


@cross_origin
@bp_dashboard.route('/<project>/manage-comments/', methods=["GET"])
@login_required
def dashboard_manage_regular_comments(project: str):
    location_id = 0
    proj_id = get_id_from_project_name(project)
    all_locations = db.session.query(TLocation)\
        .filter(TLocation.project_id == proj_id)\
        .all()

    # Check if there is a comment, otherwise don't show on management page
    # This can happen when the last comment was deleted, the location
    # won't be removed.
    tmp_list = list()
    for each in all_locations:
        comment_count = db.session.query(TComments.comments_id)\
                                  .filter(TComments.location_id == each.id_location)\
                                  .filter(TComments.is_spam == False) \
                                  .count()
        if comment_count > 0:
            tmp_list.append(each)

    all_locations = tmp_list

    # Project does not exist, error code is used by Javascript, not Flask
    if proj_id == -1:
        return redirect(url_for("bp_dashboard.dashboard_project_list", error=404))

    if request.args.get('location'):
        location_id = request.args.get('location')

    # no parameters found
    if location_id is None:
        return render_template("manage-comments.html", locations=all_locations,
                               selected=location_id, title="Manage Comments",
                               action="comments")

    try:
        if int(location_id) >= 1:
            spam_comments = db.session.query(TComments).filter(TComments.location_id == location_id) \
                .filter(TComments.is_spam == False)
            return render_template("manage-comments.html", locations=all_locations, selected=location_id,
                                   spam_comments=spam_comments, project=project,
                                   title="Manage Comments", action="comments")
    except ValueError:
        pass

    return render_template("manage-comments.html", locations=all_locations,
                           selected=location_id, project=project, title="Manage Comments",
                           action="comments")