diff -r fdf808d7d374 -r e3ebe3545f72 web/hdalab/views/ajax.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/web/hdalab/views/ajax.py Thu Feb 16 21:48:40 2012 +0100 @@ -0,0 +1,234 @@ +# -*- coding: utf-8 -*- +''' +Created on Jan 31, 2012 + +@author: ymh +''' +from django.conf import settings +from django.db import connection +from django.db.models import Q, Count, Sum +from django.http import HttpResponse +from hdabo.models import Tag, Datasheet, TaggedSheet +from hdalab.models import TagLinks, HdaSession, CountryCode, TagYears +import django.utils.simplejson as json +import hmac +import uuid + +def taginfo(request): + label = request.GET.get('label', None) + + resobj = {'requested_label' : label} + + resobj["content_count"] = Datasheet.objects.filter(taggedsheet__tag__label__iexact = label).distinct().count() + + res = Tag.objects.select_related('dbpedia_fields').filter(label__iexact = label).order_by('-dbpedia_uri')[0:1] + if len(res) == 1: + restag = res.get() + resobj["dbpedia_uri"] = restag.dbpedia_uri + if resobj["dbpedia_uri"] is not None and restag.dbpedia_fields is not None: + dbfield = restag.dbpedia_fields + resobj["abstract"] = dbfield.abstract + resobj["dbpedia_label"] = dbfield.label + resobj["thumbnail"] = dbfield.thumbnail + res = Tag.objects.filter(label__iexact = label).order_by('-wikipedia_url')[0:1] + if len(res) == 1: + resobj["wikipedia_url"] = res.get().wikipedia_url + + resobj["links"] = [{'subject':tl.subject.label, 'object':tl.object.label} for tl in TagLinks.objects.select_related().filter(Q(subject__label__iexact = label) | Q(object__label__iexact = label))] + + return HttpResponse(content=json.dumps(resobj), mimetype='application/json') + + +def sessioninfo(request): + + data = json.loads(request.GET.get('data', "{}")) + write = False + + if 'sessionid' in request.GET: + request.session['sessionid'] = request.GET['sessionid'] + if 'sessionkey' in request.GET: + request.session['sessionkey'] = request.GET['sessionkey'] + + if 'sessionid' in request.session: + sessionid = request.session['sessionid'] + + if HdaSession.objects.filter(sessionid=sessionid).count() == 1: + sessionkey = request.session.get('sessionkey',None) + hm = hmac.new(settings.SECRET_KEY, sessionid) + if hm.hexdigest() == sessionkey: + write = True + else: + del request.session['sessionid'] + + if 'sessionid' not in request.session: + sessionid = unicode(uuid.uuid1()) + HdaSession.objects.create(sessionid=sessionid, data=json.dumps({})) + write = True + request.session['sessionid'] = sessionid + request.session['sessionkey'] = hmac.new(settings.SECRET_KEY, sessionid).hexdigest() + + if write and data: + HdaSession.objects.filter(sessionid=sessionid).update(data=json.dumps(data)) + else: + data = HdaSession.objects.get(sessionid=sessionid).data + data = json.loads(data) if data else {} + + resobj = {'data': data, "write_allowed" : write, "sessionid": sessionid } + if write: + resobj['sessionkey'] = request.session['sessionkey'] + + return HttpResponse(content=json.dumps(resobj), mimetype='application/json') + + +def tagsearch(request): + + q = request.GET.get('term',None) + qs = Tag.objects.filter(label__icontains = q) if q else Tag.objects.all() + res = qs.annotate(nb=Count('datasheet')).order_by('-nb')[:20] + + return HttpResponse(content=json.dumps([{'label':t.label,'nb':t.nb} for t in res]), mimetype='application/json') + + +def filter(request): + periode = request.GET.get('period',None) + label = request.GET.get('label', None) + contentlist = request.GET.get('contentlist', None) + max_tag_order = request.GET.get('mto', 12) + content_count = request.GET.get('contentcount', 12) + tag_count = request.GET.get('tagcount', 30) + cont_count = 0; + contenus = {} + + + if label or periode or contentlist : + + qs = Datasheet.objects.filter(validated=True) + + #$globalsql = "SELECT id, title, description, url FROM hdabo_datasheet E"; + + globalids = [] + #$globalfilters = array(); + + if periode: + years = periode.split(",") + start_year = int(years[0]) + end_year = int(years[0:2][-1]) + delta = max(1, (end_year-start_year)/2) + minstart = start_year - delta; + maxend = end_year + delta; + + tagqs = Tag.objects.filter(years__end_year__gte = start_year, + years__start_year__lte = end_year, + years__end_year__lte = maxend, + years__start_year__gte = minstart) + + globalids += [t.id for t in tagqs] + + qs = qs.filter(taggedsheet__tag__in = tagqs) + + if label: + for txtlbl in label.split(","): + taglblqs = Tag.objects.filter(label__iexact = txtlbl) + globalids += [t.id for t in taglblqs if t.id not in globalids] + qs = qs.filter(taggedsheet__tag__in = taglblqs) + + if label or periode: + qs = qs.filter(taggedsheet__order__lte = max_tag_order) + + if contentlist: + qs = qs.filter(id__in = contentlist.split(",")) + + if contentlist is None: + qs = qs.order_by('?') + + + for content in qs: + cont_count += 1 + contenus[content.id] = {'score' : 0, 'tags' : [], 'id':content.id, 'title': content.title, 'description': content.description, 'url': content.url} + + + countries = dict([ (cc.label, {'isocode':cc.isocode, 'score':0}) for cc in CountryCode.objects.all() ]) + disciplines = dict([ (d.label, {'label':d.label, 'score':0}) for d in Tag.objects.filter(category__label = "Discipline artistique")]) + + tags = {} + + contentids = contenus.keys() + qs = TaggedSheet.objects.select_related('tag').filter(datasheet__in = contentids, order__lte = max_tag_order).order_by('order') + for ts in qs: + match_tag = ts.tag.id in globalids + contenus[ts.datasheet.id]['tags'].append({'id': ts.tag.id, 'label':ts.tag.label, 'order':ts.order, 'match': match_tag}) + tagscore = 2*max_tag_order - ts.order + if ts.tag.id not in tags: + tags[ts.tag.id] = {'id':ts.tag.id,'label':ts.tag.label, 'score':0} + tags[ts.tag.id]['score'] += tagscore + if match_tag: + contenus[ts.datasheet.id]['score'] += tagscore + tags[ts.tag.id]['match'] = True + if ts.tag.label in countries: + countries[ts.tag.label]['score'] += tagscore + if ts.tag.label in disciplines: + disciplines[ts.tag.label]['score'] += tagscore + + content_count = content_count if not contentlist else len(contenus) + contenus = sorted(contenus.values(),key=lambda e: e.get('score', 0))[0:content_count] + countries = countries.values() + tags = sorted(tags.values(), key=lambda e: e.get('score', 0))[0:tag_count] + disciplines = sorted(disciplines.values(), key=lambda e: e.get('score', 0))[0:10] + years = {} + if contentids: + qs = TagYears.objects.values('start_year', 'end_year').annotate(order_count=Count('tag__taggedsheet'), order_sum=Sum("tag__taggedsheet__order")).filter(tag__taggedsheet__order__lte=max_tag_order, tag__taggedsheet__datasheet__in = contentids) + for ty in qs: + for year in range(ty['start_year'], ty['end_year']): + years[year] = (2*max_tag_order*ty['order_count']-ty['order_sum'])/(ty['end_year']-ty['start_year']) + years[year] if year in years else 0 +# $rq = pg_query("SELECT U.start_year, U.end_year, SUM(".(2*$max_tag_order)." - V.order)/(U.end_year + 1 - U.start_year) score FROM hdaviz_years U, hdabo_taggedsheet V WHERE U.tag_id = V.tag_id AND V.order <= $max_tag_order AND V.datasheet_id IN ($contentids) GROUP BY U.start_year, U.end_year"); +# while($ligne = pg_fetch_row($rq)) { +# foreach(range($ligne[0], $ligne[1]) as $year) { +# $years[$year] = $ligne[2] + ( isset($years[$year]) ? $years[$year] : 0 ); + + else: + + for ds in Datasheet.objects.order_by("?")[:content_count]: + contenus[ds.id] = {'id':ds.id, 'title':ds.title, 'description':ds.description, 'url':ds.url, 'tags':[]} + cont_count = Datasheet.objects.count() + + qs = TaggedSheet.objects.select_related('tag','datasheet').filter(datasheet__id__in = contenus.keys(), order__lte = max_tag_order).order_by("order").only('order','tag__label','tag__id','datasheet__id') + for ts in qs: + contenus[ts.datasheet.id]['tags'].append({'id': ts.tag.id, 'label':ts.tag.label, 'order':ts.order}) + contenus = contenus.values() + + cursor = connection.cursor() #@UndefinedVariable + try: + cursor.execute("select t.id, t.label, sum(2*%s-ts.order) as score from hdabo_tag as t join hdabo_taggedsheet as ts on t.id = ts.tag_id where ts.order < %s group by t.id, t.label order by score limit %s",[max_tag_order, max_tag_order, tag_count]) + tags = [{'id': t[0], 'label':t[1], 'score':t[2]} for t in cursor.fetchall()] + #tags = sorted([{'id':tag.id,'label':tag.label,'score':2*max_tag_order*tag.count_score - tag.sum_score} for tag in qs], key=lambda t:t['score'])[:tag_count] + finally: + cursor.close() + cursor = connection.cursor() #@UndefinedVariable + try: + cursor.execute("select c.isocode as isocode, sum(2*%s-ts.order) as score from hdabo_tag as t join hdabo_taggedsheet as ts on t.id = ts.tag_id join hdalab_countrycode as c on t.label = c.label group by c.isocode", [max_tag_order]) + countries = [{'isocode': r[0], 'score':r[1]} for r in cursor.fetchall()] + + finally: + cursor.close() + + qs = Tag.objects.annotate(count_score=Count('taggedsheet'),sum_score=Sum('taggedsheet__order')).filter(taggedsheet__order__lte = max_tag_order, category__label__iexact = u"Discipline") + disciplines = sorted([{'label':tag.label,'score':2*max_tag_order*tag.count_score - tag.sum_score} for tag in qs], key=lambda t:t.score, reverse=True)[:10] + + years = {} + qs = TagYears.objects.values('start_year', 'end_year').annotate(order_count=Count('tag__taggedsheet'), order_sum=Sum("tag__taggedsheet__order")).filter(tag__taggedsheet__order__lte=max_tag_order) + for ty in qs: + for year in range(ty['start_year'], ty['end_year']): + years[year] = (2*max_tag_order*ty['order_count']-ty['order_sum'])/(ty['end_year']-ty['start_year']) + years[year] if year in years else 0 + + yearchange = [] + for year in sorted(years.keys()): + score = years[year] + if year < 2011: + if year-1 not in years or years[year-1] != score: + yearchange.append({'year': year, 'score': score}) + if year+1 not in years and year != -1: + yearchange.append({'year': year+1, 'score': 0}) + + output = {'count': cont_count, 'contents': contenus, 'tags':tags, 'sparkline':yearchange, 'countries':countries, 'disciplines':disciplines} + + return HttpResponse(content=json.dumps(output), mimetype='application/json')