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

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

from pycs import db
from pycs.database.File import File
from pycs.frontend.notifications.NotificationManager import NotificationManager


class CreateResult(View):
    """
    create a result for a file
    """
    # pylint: disable=arguments-differ
    methods = ['POST']

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

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

        if request_data.get('type') not in ['labeled-image', 'bounding-box']:
            return abort(400)

        rtype = request_data['type']

        if 'label' in request_data and request_data['label']:
            label = request_data['label']

        elif request_data['type'] == 'labeled-image':
            return abort(400, "label missing for the labeled-image annotation")

        else:
            label = None

        if request_data.get('data'):
            data = request_data['data']

        elif request_data['type'] == 'bounding-box':
            return abort(400, "data missing for the bounding box annotation")

        else:
            data = {}

        # find file
        file = File.query.get(file_id)
        if file is None:
            return abort(404)


        removed = []
        # start transaction
        with db.session.begin_nested():
            # find full-image labels and remove them
            for result in file.results.all():
                if result.type == 'labeled-image':
                    removed.append(result.serialize())
                    result.remove(commit=True)

            # insert into database
            new_result = file.create_result('user', rtype, label, data,
                commit=False)

        for result in removed:
            self.nm.remove_result(result.serialize())

        self.nm.create_result(new_result.id)
        return jsonify(new_result)