"""
from taggit-templatetags. This app is very useful but does not
"""

import logging

from django import template
from django.contrib.contenttypes.models import ContentType
from django.db.models import Count
from django.utils.safestring import SafeText
from taggit_templatetags import settings
from templatetag_sugar.parser import Name, Variable, Constant, Optional
from templatetag_sugar.register import tag
from django.apps import apps


T_MAX = getattr(settings, 'TAGCLOUD_MAX', 6.0)
T_MIN = getattr(settings, 'TAGCLOUD_MIN', 1.0)

register = template.Library()

logger = logging.getLogger(__name__)

def get_queryset(forvar=None):
    TaggedItem = apps.get_model('taggit', 'TaggedItem')
    Tag = apps.get_model('taggit', 'Tag')
    if None == forvar:
        # get all tags
        queryset = Tag.objects.all()
    else:
        if isinstance(forvar, str) or isinstance(forvar, SafeText):
            # extract app label and model name
            _, applabel, model = None, None, None
            try:
                _, applabel, model = forvar.rsplit('.', 2)
            except ValueError:
                try:
                    applabel, model = forvar.rsplit('.', 1)
                except ValueError:
                    applabel = forvar
            
            # filter tagged items
            if applabel:
                queryset = TaggedItem.objects.filter(content_type__app_label=applabel.lower())
            if model:
                queryset = queryset.filter(content_type__model=model.lower())
            tag_ids = queryset.values_list('tag_id', flat=True)
            queryset = Tag.objects.filter(id__in=tag_ids)
            
            if applabel and not model :
                return queryset.extra(select={ "num_times": 'SELECT count(*) FROM "taggit_taggeditem" U0 INNER JOIN "django_content_type" U1  ON ( U0."content_type_id" = U1."id" ) WHERE ( U1."app_label" = \'' + applabel.lower() + '\' ) AND ("taggit_tag"."id" = U0."tag_id")' })
            elif applabel and model:
                return queryset.extra(select={ "num_times": 'SELECT count(*) FROM "taggit_taggeditem" U0 INNER JOIN "django_content_type" U1  ON ( U0."content_type_id" = U1."id" ) WHERE (U1."app_label" = \'' + applabel.lower() + '\'  AND U1."model" = \'' + model.lower() + '\' ) AND ("taggit_tag"."id" = U0."tag_id")' })
        
        else:
            # forvar is a query set
            content_type = ContentType.objects.get_for_model(forvar[0])
            queryset = TaggedItem.objects.filter(content_type=content_type, object_id__in=[o.pk for o in forvar])
            # get tags
            tag_ids = queryset.values_list('tag_id', flat=True)
            queryset = Tag.objects.filter(id__in=tag_ids)
            return queryset.annotate(num_times=Count('taggit_taggeditem_items'))
    
    return queryset.annotate(num_times=Count('taggit_taggeditem_items'))
    

def get_weight_fun(t_min, t_max, f_min, f_max):
    def weight_fun(f_i, t_min=t_min, t_max=t_max, f_min=f_min, f_max=f_max):
        # Prevent a division by zero here, found to occur under some
        # pathological but nevertheless actually occurring circumstances.
        if f_max == f_min:
            mult_fac = 1.0
        else:
            mult_fac = float(t_max-t_min)/float(f_max-f_min)
            
        return t_max - (f_max-f_i)*mult_fac
    return weight_fun

@tag(register, [Constant('as'), Name(), Optional([Constant('for'), Variable()])])
def get_taglist(context, asvar, forvar=None):
    queryset = get_queryset(forvar)         
    queryset = queryset.order_by('-num_times')        
    context[asvar] = queryset
    return ''

@tag(register, [Constant('as'), Name(), Optional([Constant('for'), Variable()])])
def get_tagcloud(context, asvar, forvar=None):
    queryset = get_queryset(forvar)
    num_times = queryset.values_list('num_times', flat=True)
    if(len(num_times) == 0):
        context[asvar] = queryset
        return ''
    weight_fun = get_weight_fun(T_MIN, T_MAX, min(num_times), max(num_times))
    queryset = queryset.order_by('name')
    for tag in queryset:
        tag.weight = weight_fun(tag.num_times)
    context[asvar] = queryset
    return ''
    
def include_tagcloud(forvar=None):
    return {'forvar': forvar}

def include_taglist(forvar=None):
    return {'forvar': forvar}
  
register.inclusion_tag('taggit_templatetags/taglist_include.html')(include_taglist)
register.inclusion_tag('taggit_templatetags/tagcloud_include.html')(include_tagcloud)
