from django import template
from django.shortcuts import get_object_or_404
from jarvis.people.models import Gymnast

register = template.Library()


@register.inclusion_tag("gymnasts/gymnast_level_chart_bar.html")
def generate_level_chart_bar(gymnast_id):
    """
    Tag permettant l'affichage des statistiques d'un gymnaste : le nombre de saut qu'il sait faire
    (total, par niveau, par rank, …), calcule la complétude, …
    """

    gymnast = get_object_or_404(Gymnast, pk=gymnast_id)
    context = gymnast.get_informations_from_type("level")
    return context