PyCS/pycs/frontend/endpoints/results/ConfirmResult.py

from flask import make_response, request, abort
from flask.views import View

from pycs.database.Database import Database
from pycs.frontend.notifications.NotificationManager import NotificationManager


class ConfirmResult(View):
    """
    confirm a result (change its origin to user)
    """
    # pylint: disable=arguments-differ
    methods = ['POST']

    def __init__(self, db: Database, nm: NotificationManager):
        # pylint: disable=invalid-name
        self.db = db
        self.nm = nm

    def dispatch_request(self, result_id: int):
        # extract request data
        data = request.get_json(force=True)

        if 'confirm' not in data or data['confirm'] is not True:
            return abort(400)

        # find result
        result = self.db.result(result_id)
        if result is None:
            return abort(404)

        # start transaction
        with self.db:
            result.set_origin('user')

        self.nm.edit_result(result)
        return make_response()