from django.template.loader import TemplateDoesNotExist, get_template

from wagtail.admin.modal_workflow import render_modal_workflow
from wagtail.snippets.models import get_snippet_models

from .utils import get_snippet_frontend_template


def choose_snippet_model(request):
    snippet_model_opts = []

    # Only display those snippet models which have snippet frontend template
    for snippet_model in get_snippet_models():
        snippet_frontend_template = get_snippet_frontend_template(
            snippet_model._meta.app_label, snippet_model._meta.model_name
        )
        try:
            get_template(snippet_frontend_template)
            snippet_model_opts.append(snippet_model._meta)
        except TemplateDoesNotExist:
            pass

    return render_modal_workflow(
        request,
        "wagtail_draftail_snippet/choose_snippet_model.html",
        None,
        {"snippet_model_opts": snippet_model_opts},
        json_data={"step": "choose"},
    )