open-law/app/views/vote.py

from flask import (
    Blueprint,
    jsonify,
)
from flask_login import login_required, current_user

from app import models as m, db, forms as f
from app.logger import log

bp = Blueprint("vote", __name__, url_prefix="/vote")


@bp.route(
    "/interpretation/<int:interpretation_id>",
    methods=["POST"],
)
@login_required
def vote_interpretation(interpretation_id: int):
    interpretation: m.Interpretation = db.session.get(
        m.Interpretation, interpretation_id
    )
    if not interpretation:
        log(log.WARNING, "Interpretation with id [%s] not found", interpretation_id)
        return jsonify({"message": "Interpretation not found"}), 404

    form = f.VoteForm()
    if form.validate_on_submit():
        vote: m.InterpretationVote = m.InterpretationVote.query.filter_by(
            user_id=current_user.id, interpretation_id=interpretation_id
        ).first()
        if vote:
            db.session.delete(vote)

        positive = form.positive.data and "True" in form.positive.raw_data
        if not vote or vote.positive != positive:
            vote: m.InterpretationVote = m.InterpretationVote(
                user_id=current_user.id,
                interpretation_id=interpretation_id,
                positive=positive,
            )
            log(
                log.INFO,
                "User [%s]. [%s] vote interpretation: [%s]",
                current_user,
                "Positive" if positive else "Negative",
                interpretation,
            )
            vote.save(False)
        else:
            log(
                log.INFO,
                "User [%s]. Remove [%s] vote for interpretation: [%s]",
                current_user,
                "positive" if positive else "negative",
                interpretation,
            )
        db.session.commit()

        return jsonify({"vote_count": interpretation.vote_count})

    log(
        log.CRITICAL,
        "Unexpected error: User [%s]. Vote for interpretation: [%s]",
        current_user,
        interpretation,
    )
    return jsonify({"message": "Unexpected error"}), 400


@bp.route(
    "/comment/<int:comment_id>",
    methods=["POST"],
)
@login_required
def vote_comment(comment_id: int):
    comment: m.Comment = db.session.get(m.Comment, comment_id)
    if not comment:
        log(log.WARNING, "Comment with id [%s] not found", comment_id)
        return jsonify({"message": "Comment not found"}), 404

    form = f.VoteForm()
    if form.validate_on_submit():
        vote: m.CommentVote = m.CommentVote.query.filter_by(
            user_id=current_user.id, comment_id=comment_id
        ).first()
        if vote:
            db.session.delete(vote)

        positive = form.positive.data and "True" in form.positive.raw_data
        if not vote or vote.positive != positive:
            vote: m.CommentVote = m.CommentVote(
                user_id=current_user.id,
                comment_id=comment_id,
                positive=positive,
            )
            log(
                log.INFO,
                "User [%s]. [%s] vote comment: [%s]",
                current_user,
                "Positive" if positive else "Negative",
                comment,
            )
            vote.save(False)
        else:
            log(
                log.INFO,
                "User [%s]. Remove [%s] vote for comment: [%s]",
                current_user,
                "positive" if positive else "negative",
                comment,
            )
        db.session.commit()

        return jsonify({"vote_count": comment.vote_count})

    log(
        log.CRITICAL,
        "Unexpected error: User [%s]. Vote for comment: [%s]",
        current_user,
        comment,
    )
    return jsonify({"message": "Unexpected error"}), 400