from django import template

from program.models import BroadcastFormat, MusicFocus, ShowInformation, ShowTopic, Language

register = template.Library()


# Legacy Tags for Homepage until 2021

@register.inclusion_tag('boxes/broadcastformat.html')
def broadcastformat():
    return {'broadcastformat_list': BroadcastFormat.objects.filter(enabled=True)}


@register.inclusion_tag('boxes/musicfocus.html')
def musicfocus():
    return {'musicfocus_list': MusicFocus.objects.all()}


@register.inclusion_tag('boxes/showinformation.html')
def showinformation():
    return {'showinformation_list': ShowInformation.objects.all()}


@register.inclusion_tag('boxes/showtopic.html')
def showtopic():
    return {'showtopic_list': ShowTopic.objects.all()}


@register.inclusion_tag('boxes/language.html')
def language():
    return {'language_list': Language.objects.all()}


# V2 Tags added for new Homepage 2021

@register.inclusion_tag('v2/boxes/broadcastformat.html')
def broadcastformatV2():
    return {'broadcastformat_list': BroadcastFormat.objects.filter(enabled=True)}


@register.inclusion_tag('v2/boxes/musicfocus.html')
def musicfocusV2():
    return {'musicfocus_list': MusicFocus.objects.all()}


@register.inclusion_tag('v2/boxes/showinformation.html')
def showinformationV2():
    return {'showinformation_list': ShowInformation.objects.all()}


@register.inclusion_tag('v2/boxes/showtopic.html')
def showtopicV2():
    return {'showtopic_list': ShowTopic.objects.all()}


@register.inclusion_tag('v2/boxes/language.html')
def languageV2():
    return {'language_list': Language.objects.all()}