from flask import request, jsonify, redirect
from flask_classful import route
from werkzeug.datastructures import MultiDict

from app.sql.common import db, Session
from app.sql.models.annotation import Annotation, AnnotationForm
from app.controllers.crud_controller import CrudView

class AnnotationView(CrudView):
  model = Annotation
  form = AnnotationForm
  default_sort = "start_ts"

  def on_create(self, session, form, item):
    if 'paragraph_id' in form:
      item.paragraph_id = form['paragraph_id']
    if 'settings' in form:
      item.settings = form['settings']

  def on_update(self, session, form, item):
    if 'paragraph_id' in form:
      item.paragraph_id = form['paragraph_id']
    if 'settings' in form:
      item.settings = form['settings']