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.media import Media, MediaForm
from app.controllers.crud_controller import CrudView

class MediaView(CrudView):
  model = Media
  form = MediaForm
  index_all = False
  use_limit = False

  def where(self, query, args):
    episode_id = args.get('episode_id', default=None)
    if episode_id is not None:
      query = query.filter(Media.episode_id == int(episode_id))
    return query

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

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