from django import template
from django.middleware.csrf import get_token
from django.utils.html import format_html
from django.utils.safestring import mark_safe

register = template.Library()


@register.simple_tag(takes_context=True)
def csrf_input(context):
    """Compatível com templates portados do Flask — use {% csrf_token %} em código novo."""
    request = context["request"]
    token = get_token(request)
    return format_html(
        '<input type="hidden" name="csrfmiddlewaretoken" value="{}">',
        token,
    )


@register.simple_tag
def submit_btn(label, css_class="btn btn-parsifal"):
    return format_html('<button type="submit" class="{}">{}</button>', css_class, label)