from django import template
from datetime import timedelta
from availability.models import Availability

register = template.Library()

@register.simple_tag(takes_context=True)
def ratesroom(context, rates, room, start, end):
    ratessalida = []

    avs =  Availability.objects.filter(date__range=[start, end],
                                                room_id=room)
    condicionesEncontradas = []
    for av in avs:
       if av.conditionday and av.conditionday != '' and av.conditionday != '-':
           if av.conditionday not in condicionesEncontradas:
              condicionesEncontradas.append(av.conditionday)
    strMasRestrictiva = ''
    if len(condicionesEncontradas)>0:
       if 'NR' in condicionesEncontradas:
          strMasRestrictiva = 'NR'
       elif 'GE' in condicionesEncontradas:
          strMasRestrictiva = 'GE'
       elif 'G' in condicionesEncontradas:
          strMasRestrictiva = 'G'
       if strMasRestrictiva != '':
          for r in rates:
             raux = {'name': r.name + ' *Cond.: (' + strMasRestrictiva + ')','id':r.id}
             ratessalida.append(raux)
    else:
      for r in rates:
          raux = {'name': r,'id':r.id}
          ratessalida.append(raux)
    context['ratessalida'] = ratessalida
    return ''