from flask import redirect, url_for

import screens

from core.flask_app import app


@app.route("/game/<slug>/screens/<int:screen_id>/delete", methods=["POST"])
def screen_delete(slug, screen_id):
    referencing = screens.screens_referencing(slug, screen_id)
    if referencing:
        return redirect(url_for("screens_list", slug=slug, blocked_screen=screen_id))
    screens.delete_screen(slug, screen_id)
    return redirect(url_for("screens_list", slug=slug))