# -*- coding: utf-8 -*-

'''

Created on Jun 13, 2013

@author: ymh

'''

from core.models import Thesaurus, Term, TERM_URL_STATUS_CHOICES_TRANS

from django.forms import Form, fields, ModelChoiceField

from django.forms.util import flatatt

from django.forms.widgets import Widget

from django.utils import formats

from django.utils.encoding import force_text

from django.utils.html import format_html, escape

from django.utils.safestring import mark_safe

from django.utils.translation import ugettext as _

import json

class ThesaurusTreeWidget(Widget):

    def _format_value(self, value):

        if self.is_localized:

            return formats.localize_input(value)

        return value

    def render(self, name, value, attrs=None):

        term = None

        if isinstance(value, Term):

            term = value

        elif value and isinstance(value, (int, basestring)):

            terms = list(Term.objects.filter(id=int(value)))  # @UndefinedVariable

            if terms:

                term = terms[0]

        final_attrs = self.build_attrs(attrs, type="hidden", name=name)

        if term is not None:

            # Only add the 'value' attribute if a value is non-empty.

            final_attrs['value'] = force_text(self._format_value(term.id))

            final_attrs['data-term_tree_node'] = mark_safe(json.dumps({'id': term.id, 'label': escape(term.label)}).replace("\"", "'"));

        input_res = format_html('<input{0} />', flatatt(final_attrs))

        if term is not None:

            dialog_text = term.label

        else:

            dialog_text = _("Open Dialog")

        dialog_res = "<div id=\"dialog-link-container\" class=\"ui-state-default ui-corner-all\"><a href=\"#\" id=\"dialog-link\" title=\"%s\">%s</a><span class=\"ui-icon ui-icon-closethick\" id=\"dialog-deselect\"></span></div>" % (_("Open Dialog"),dialog_text)

        return input_res + dialog_res

class ValidateTermForm(Form):

    term_id = fields.IntegerField(required=True)

    validation_val = fields.BooleanField(required=False)

class WikipediaEditionForm(Form):

    term_id = fields.IntegerField(required=True)

    wikipedia_edition = fields.BooleanField(required=False)

class ModifyWpLinkForm(Form):

    term_id = fields.IntegerField(required=True)

    label = fields.CharField(required=True, min_length=1)

class LinkSemanticLevelForm(Form):

    term_id = fields.IntegerField(required=True)

    link_semantic_level = fields.IntegerField(required=True)

class RemoveWpLinkForm(Form):

    term_id = fields.IntegerField(required=True)

def validated_to_bool(val):

    if isinstance(val, basestring) and len(val)==0:

        return None

    else:

        return bool(int(val))

class TermFilterForm(Form):

    thesaurus = ModelChoiceField(label=_("thesaurus"), required=False, queryset=Thesaurus.objects.all().order_by('label'))

    thesaurus_tree = ModelChoiceField(label=_("thesaurus tree"), queryset=Term.objects.all(), required=False, widget=ThesaurusTreeWidget)  # @UndefinedVariable

    label = fields.CharField(label=_("label"), required=False)

    link_status = fields.TypedChoiceField(label=_("link_status"), required=False, empty_value=-1, coerce=int, choices=tuple([(-1,'---------')]+[(v, l) for v,l in TERM_URL_STATUS_CHOICES_TRANS]))

    validated = fields.TypedChoiceField(label=_("validated"), required=False, empty_value=None, coerce=validated_to_bool, choices=(("", "---"),("1", _("yes")), ("0", _("no"))))

    order_by = fields.ChoiceField(label=_("order_by"), required=False, choices=(('normalized_label',_('label')),('nb_notice',_('nb notice')),('level',_('level'))))

    order_dir = fields.ChoiceField(label=_("order_dir"), required=False, choices=(('asc',_('asc')), ('desc',_('desc'))))

    def get_filter_qs(self, base_qs=None):

        qs = base_qs

        if qs is None:

            qs = Term.objects.all()  # @UndefinedVariable

        thes = self.cleaned_data.get('thesaurus',None)

        if thes:

            qs = qs.filter(thesaurus=thes)

        thes_tree = self.cleaned_data.get('thesaurus_tree',None)

        if thes_tree:

            qs = qs & thes_tree.get_descendants()  # @UndefinedVariable

        lk_status = self.cleaned_data.get('link_status',-1)

        if lk_status>=0:

            qs = qs.filter(url_status=lk_status)

        validated = self.cleaned_data.get('validated', None)

        if validated is not None:

            qs = qs.filter(validated=validated)

        label_regexp = self.cleaned_data.get('label', None)

        if label_regexp:

            qs = qs.filter(label__iregex=label_regexp)

        order_by = self.cleaned_data.get('order_by', 'normalized_label') or 'normalized_label'

        order_dir = self.cleaned_data.get('order_dir', 'asc') or 'asc'

        if order_dir == 'desc':

            order_by = "-"+order_by

        if order_by == "normalized_label" or order_by == "label":

            order_by = [order_by, 'nb_notice', 'id']

        else:

            order_by = [order_by, 'normalized_label', 'id']

        qs = qs.order_by(*order_by)

        return qs