forked from 0x2620/pandora
110 lines
3.8 KiB
Python
110 lines
3.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
# vi:si:et:sw=4:sts=4:ts=4
|
|
from __future__ import division, with_statement
|
|
from django.db import models
|
|
|
|
import ox
|
|
from ox.django import fields
|
|
from django.contrib.auth.models import User, Group
|
|
from django.db.models import Q
|
|
|
|
import managers
|
|
from annotation.models import Annotation
|
|
from item.models import Item
|
|
|
|
|
|
class Place(models.Model):
|
|
'''
|
|
Places are named locations, they should have geographical information attached to them.
|
|
'''
|
|
created = models.DateTimeField(auto_now_add=True)
|
|
modified = models.DateTimeField(auto_now=True)
|
|
user = models.ForeignKey(User, null=True, related_name='places')
|
|
|
|
name = models.CharField(max_length=1024)
|
|
alternativeNames = fields.TupleField(default=[])
|
|
name_sort = models.CharField(max_length=200)
|
|
name_find = models.TextField(default='', editable=False)
|
|
|
|
geoname = models.CharField(max_length=1024, unique=True)
|
|
geoname_sort = models.CharField(max_length=1024, unique=True)
|
|
countryCode = models.CharField(max_length=16, default='')
|
|
|
|
wikipediaId = models.CharField(max_length=1000, blank=True)
|
|
type = models.CharField(max_length=1000, default='')
|
|
|
|
south = models.FloatField(default=0)
|
|
west = models.FloatField(default=0)
|
|
north = models.FloatField(default=0)
|
|
east = models.FloatField(default=0)
|
|
lat = models.FloatField(default=0)
|
|
lng = models.FloatField(default=0)
|
|
area = models.FloatField(default=0)
|
|
|
|
matches = models.IntegerField(default=0)
|
|
items = models.ManyToManyField(Item, blank=True, related_name='places')
|
|
annotations = models.ManyToManyField(Annotation, blank=True, related_name='places')
|
|
|
|
objects = managers.PlaceManager()
|
|
|
|
class Meta:
|
|
ordering = ('name_sort', )
|
|
|
|
def __unicode__(self):
|
|
return self.name
|
|
|
|
def editable(self, user):
|
|
if user and user.is_staff or self.user == user:
|
|
return True
|
|
return False
|
|
|
|
def get_id(self):
|
|
return ox.to32(self.id)
|
|
|
|
def json(self, user=None):
|
|
j = {
|
|
'id': self.get_id(),
|
|
'user': self.user.username,
|
|
'editable': self.editable(user)
|
|
}
|
|
for key in ('created', 'modified',
|
|
'name', 'alternativeNames', 'geoname', 'countryCode',
|
|
'south', 'west', 'north', 'east',
|
|
'lat', 'lng', 'area', 'matches', 'type'):
|
|
j[key] = getattr(self, key)
|
|
return j
|
|
|
|
def get_matches(self):
|
|
q = Q(value__contains=" " + self.name)|Q(value__startswith=self.name)
|
|
for name in self.alternativeNames:
|
|
q = q|Q(value__contains=" " + name)|Q(value__startswith=name)
|
|
return Annotation.objects.filter(q)
|
|
|
|
def update_matches(self):
|
|
matches = self.get_matches()
|
|
self.matches = matches.count()
|
|
for i in self.annotations.exclude(id__in=matches):
|
|
self.annotations.remove(i)
|
|
for i in matches.exclude(id__in=self.annotations.all()):
|
|
self.annotations.add(i)
|
|
ids = list(set([a.item.id for a in matches]))
|
|
for i in self.items.exclude(id__in=ids):
|
|
self.items.remove(i)
|
|
for i in Item.objects.filter(id__in=ids).exclude(id__in=self.items.all()):
|
|
self.items.add(i)
|
|
self.save()
|
|
|
|
def save(self, *args, **kwargs):
|
|
if not self.name_sort:
|
|
self.name_sort = self.name #', '.join(self.name)
|
|
self.geoname_sort = ', '.join(reversed(self.geoname.split(', ')))
|
|
self.name_find = '|%s|'%'|'.join([self.name]+list(self.alternativeNames))
|
|
|
|
#update center
|
|
#self.lat = ox.location.center(self.south, self.north)
|
|
#self.lng = ox.location.center(self.east, self.west)
|
|
|
|
#update area
|
|
#self.area= ox.location.area(self.south, self.west, self.north, self.east)
|
|
|
|
super(Place, self).save(*args, **kwargs)
|