from django.conf import settings

from blinkster.ldt.models import Segment

from blinkster.ldt.utils import LdtSearch

import re

import urllib2

def serialize_queryset_to_json(response, obj_list, extra=None):

    if extra:

        objs = dict(objs, **extra)

    results = []

    for roi_dict in [roi.serialize_to_dict() for roi in Roi.objects.all()]:

        roi_dict["poi_list_url"] = unicode(settings.WEB_URL.rstrip('/') + reverse('blinkster.views.poi_list',args=[roi_dict["sid"]]))

        results.append(roi_dict)

    return serialize_queryset_to_json(response, results)

    results = []

    for poi_dict in [poi.serialize_to_dict() for     poi in Poi.objects.filter(roi=roi)]:

        poi_dict["segment_search_url"] = unicode(settings.WEB_URL.rstrip('/') + reverse('blinkster.views.segment_search_get',args=[u"tags",u"or",urllib2.quote(poi_dict["tags"].encode("utf-8"))]))

        results.append(poi_dict)

    return serialize_queryset_to_json(response, results)

# Display segment detail.

# This will have to be moved to ldt module

def segment_detail(request, iri_id, ensemble_id, cutting_id, element_id, project_id = None):

    segment = get_object_or_404(Segment, project_id=project_id, iri_id=iri_id, ensemble_id=ensemble_id, cutting_id=cutting_id, element_id=element_id)

    return render_to_response("segment_detail.html",{'segment':segment}, context_instance=RequestContext(request))

def segment_search_post(request):

    query = request.POST["query"]

    field = request.POST["field"]

    operator = request.POST["operator"]

    return segment_search_get(request, query, field, operator)

def segment_search_get(request, query, field = u"tags", operator=u"or"):

    content_type = request.GET.get("content-type", "application/json; charset=utf-8")

    response = HttpResponse(content_type=str(content_type))

    # transform tag1, tag2, tag3 in "tag1" OR "tag2" OR "tag3"

    query_str = (u" " + operator.upper() + u" ").join(["\"" + t.strip() + "\"" for t in re.split("\,|\;",query)])

    searcher = blinkster.ldt.utils.LdtSearch()

    index_results = searcher.query(field, query_str)

    results = []

    for index_res in index_results:

        res = dict(index_res)

        segs = Segment.objects.filter(iri_id=index_res["iri_id"], element_id=index_res["element_id"], cutting_id=index_res["decoupage_id"], ensemble_id=index_res["ensemble_id"]).values() 

        for seg in segs:

            del seg["project_obj_id"]

            del seg["content_id"]

            del seg["id"]

            res = dict(res, **seg)

        res["segment_detail_url"] = unicode(settings.WEB_URL.rstrip('/') + reverse('blinkster.views.segment_detail',args=[res["iri_id"], res["ensemble_id"], res["cutting_id"], res["element_id"]]))

        results.append(res)

    return serialize_queryset_to_json(response, results, {u"query" : query, u"field": field, u"operator":operator, u"final_query": query_str})