# -*- coding: utf-8 -*- # vi:si:et:sw=4:sts=4:ts=4 from __future__ import division from django.db.models import Max, Min import ox from ox.utils import json from ox.django.decorators import login_required_json from ox.django.shortcuts import render_to_json_response, get_object_or_404_json, json_response from api.actions import actions from item import utils import models import tasks @login_required_json def addPlace(request): #FIXME: require admin ''' param data { name: "", alternativeNames: [], geoname: "", countryCode: '', south: float, west: float, north: float, east: float, lat: float, lng: float, area: float, type: "" } ''' data = json.loads(request.POST['data']) exists = False names = data.pop('name') for name in [names] + data.get('alternativeNames', []): if models.Place.objects.filter(name_find__icontains=u'|%s|'%name).count() != 0: exists = True if not exists: place = models.Place() place.user = request.user place.name = names place.alternativeNames = tuple(data.pop('alternativeNames', [])) for key in data: value = data[key] if isinstance(value, list): value = tuple(value) setattr(place, key, value) place.save() #tasks.update_matches.delay(place.id) response = json_response(place.json()) else: response = json_response(status=403, text='place name exists') return render_to_json_response(response) actions.register(addPlace, cache=False) @login_required_json def editPlace(request): ''' param data { 'id': placeid, 'name': ... 'north': 0... } can contain any of the allowed keys for place ''' data = json.loads(request.POST['data']) place = get_object_or_404_json(models.Place, pk=ox.from32(data['id'])) names = data.get('name', []) if isinstance(names, basestring): names = [names] if place.editable(request.user): conflict = False for name in names + data.get('alternativeNames', []): if models.Place.objects.filter(name_find__icontains=u'|%s|'%name).exclude(id=place.id).count() != 0: conflict = True if 'geoname' in data: if models.Place.objects.filter(geoname=data['geoname']).exclude(id=place.id).count() != 0: conflict = True if not conflict: for key in data: if key != 'id': value = data[key] if isinstance(value, list): value = tuple(value) setattr(place, key, value) place.save() #tasks.update_matches.delay(place.id) response = json_response(place.json()) else: response = json_response(status=403, text='place name/geoname conflict') else: response = json_response(status=403, text='permission denied') return render_to_json_response(response) actions.register(editPlace, cache=False) @login_required_json def removePlace(request): ''' param data { 'id': placeid, } ''' data = json.loads(request.POST['data']) if isinstance(data, dict): data = data['id'] place = get_object_or_404_json(models.Place, pk=ox.from32(data)) if place.editable(request.user): place.delete() response = json_response(status=200, text='deleted') else: response = json_response(status=403, text='permission denied') return render_to_json_response(response) actions.register(removePlace, cache=False) def parse_query(data, user): query = {} query['range'] = [0, 100] query['sort'] = [{'key':'name', 'operator':'+'}] for key in ('keys', 'group', 'list', 'range', 'position', 'positions', 'sort', 'query'): if key in data: query[key] = data[key] query['qs'] = models.Place.objects.find(query, user) if 'itemQuery' in data: item_query = models.Item.objects.find({'query': data['itemQuery']}, user) query['qs'] = query['qs'].filter(items__in=item_query) return query def order_query(qs, sort): order_by = [] for e in sort: operator = e['operator'] if operator != '-': operator = '' key = { 'name': 'name_sort', 'geoname': 'geoname_sort', }.get(e['key'], e['key']) order = '%s%s' % (operator, key) order_by.append(order) if order_by: qs = qs.order_by(*order_by, nulls_last=True) return qs def findPlaces(request): ''' param data { query: { conditions: [ { key: 'user', value: 'something', operator: '=' } ] operator: "," }, itemQuery: { //see find request }, sort: [{key: 'name', operator: '+'}], range: [0, 100] keys: [] } possible query keys: name, geoname, user itemQuery can be used to limit the resuts to matches in those items. Uses the same query syntax as used in the find request. possible keys: name, geoname, user return { status: { code: int, text: string }, data: { items: [ {name:, user:, featured:, public...} ] } } param data {'query': query, 'sort': array, 'range': array} query: query object, more on query syntax at https://wiki.0x2620.org/wiki/pandora/QuerySyntax sort: array of key, operator dics [ { key: "year", operator: "-" }, { key: "director", operator: "" } ] range: result range, array [from, to] with keys, items is list of dicts with requested properties: return {'status': {'code': int, 'text': string}, 'data': {items: array}} Positions param data {'query': query, 'positions': []} query: query object, more on query syntax at https://wiki.0x2620.org/wiki/pandora/QuerySyntax positions: ids of places for which positions are required ''' data = json.loads(request.POST['data']) response = json_response() query = parse_query(data, request.user) qs = order_query(query['qs'], query['sort']) if 'keys' in data: qs = qs[query['range'][0]:query['range'][1]] response['data']['items'] = [p.json(request.user) for p in qs] elif 'position' in query: ids = [i.get_id() for i in qs] data['conditions'] = data['conditions'] + { 'value': query['position'], 'key': query['sort'][0]['key'], 'operator': '^' } query = parse_query(data, request.user) qs = order_query(query['qs'], query['sort']) if qs.count() > 0: response['data']['position'] = utils.get_positions(ids, [qs[0].itemId])[0] elif 'positions' in data: ids = [i.get_id() for i in qs] response['data']['positions'] = utils.get_positions(ids, query['positions']) else: response['data']['items'] = qs.count() response['data']['area'] = qs.aggregate( south=Min('south'), west=Min('west'), north=Max('north'), east=Max('east'), ) return render_to_json_response(response) actions.register(findPlaces)