import json, os

from django import template

from interface.models import Transcripts

register = template.Library()


def sub_vtt(value): 
    if value is None:
        return {}

    tr_obj = Transcripts.objects.get(id_transcript=value)

    VTT_RESPONSE = {
        'file': tr_obj.file_org.file_s3_url,
        'es': ((str(os.path.basename(tr_obj.path_srt_es)).replace('.srt','.vtt') if "mp4" not in  tr_obj.path_srt_es else f"{(os.path.splitext(os.path.basename(tr_obj.path_srt_es))[0]).split('.')[0]}.vtt") if tr_obj.path_srt_es is not None else None ),
        'en': ((str(os.path.basename(tr_obj.path_srt_en)).replace('.srt','.vtt') if "mp4" not in tr_obj.path_srt_en else f"{(os.path.splitext(os.path.basename(tr_obj.path_srt_en))[0]).split('.')[0]}.vtt") if tr_obj.path_srt_en is not None else None ),
    }
    return json.dumps(VTT_RESPONSE)

register.filter('sub_vtt', sub_vtt)