from flask import redirect, url_for, abort

import screens

from core.flask_app import app


@app.route("/game/<slug>/elements/<int:element_id>/tabs/add", methods=["POST"])
def element_tab_add(slug, element_id):
    onglets = screens.get_element(slug, element_id)
    if not onglets:
        abort(404)
    screens.add_tab(slug, element_id, onglets["screen_id"])
    return redirect(url_for("screen_edit", slug=slug, screen_id=onglets["screen_id"], selected=element_id))