open-law/app/views/vote.py

from flask import Blueprint, jsonify, request, url_for
from flask_login import login_required, current_user

from app import models as m, db
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
    book: m.Book = db.session.get(m.Book, interpretation.book.id)

    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 = request.json.get("positive") in ("true", True)

    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()

    # notifications
    if current_user.id != book.owner.id:
        redirect_url = url_for(
            "book.interpretation_view",
            book_id=book.id,
            section_id=interpretation.section_id,
        )
        notification_text = f"{current_user.username} voted your interpretation"
        m.Notification(
            link=redirect_url, text=notification_text, user_id=book.owner.id
        ).save()
        log(
            log.INFO,
            "Create notification for user with id [%s]",
            book.owner.id,
        )
    # -------------

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


@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

    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 = request.json.get("positive") in ("true", True)

    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,
            "current_user_vote": comment.current_user_vote,
        }
    )