1 files changed, 163 insertions, 0 deletions
diff --git a/main/models.py b/main/models.py
new file mode 100644
index 0000000..6d673d8
--- /dev/null
+++ b/main/models.py
@@ -0,0 +1,163 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+Models description
+"""
+from django.utils.translation import ugettext_lazy as _
+
+from django.contrib.gis.db import models
+from django.contrib import admin
+
+from chimere import settings
+from chimere.main.widgets import PointField
+
+class Category(models.Model):
+    """Category of Point Of Interest (POI)
+    """
+    name = models.CharField(_("Name"), max_length=150)
+    available = models.BooleanField(_("Available"))
+    order = models.IntegerField(_("Order"))
+    def __unicode__(self):
+        return self.name
+    class Meta:
+        ordering = ["order"]
+        verbose_name = _("Category")
+
+class Icon(models.Model):
+    '''Icon
+    '''
+    name = models.CharField(_("Name"), max_length=150)
+    image = models.ImageField(_("Image"), upload_to='icons')
+    def __unicode__(self):
+        return self.name
+    class Meta:
+        verbose_name = _("Icon")
+
+class SubCategory(models.Model):
+    '''Sub-category
+    '''
+    category = models.ForeignKey(Category, verbose_name=_("Category"))
+    name = models.CharField(_("Name"), max_length=150)
+    available = models.BooleanField(_("Available"))
+    icon = models.ForeignKey(Icon, verbose_name=_("Icon"))
+    order = models.IntegerField(_("Order"))
+    def __unicode__(self):
+        return u"%s / %s" % (self.category.name, self.name)
+    class Meta:
+        ordering = ["category", "order"]
+        verbose_name = _("Subcategory")
+
+    def getAvailable():
+        '''Get list of tuples with first the category and second the associated
+        subcategories
+        '''
+        sub_categories = {}
+        for sub_category in SubCategory.objects.filter(category__available=True,
+                                                       available=True):
+            if sub_category.category not in sub_categories:
+                sub_categories[sub_category.category] = []
+            sub_categories[sub_category.category].append(sub_category)
+        return [(category, sub_cats) for category, sub_cats \
+                                         in sub_categories.items()]
+    getAvailable = staticmethod(getAvailable)
+
+class Marker(models.Model):
+    '''Marker for a POI
+    '''
+    name = models.CharField(_("Name"), max_length=150)
+    subcategory = models.ForeignKey(SubCategory, verbose_name=_("Subcategory"))
+    point = PointField(_("Localisation"))
+    picture = models.ImageField(_("Image"), upload_to='upload', blank=True)
+    STATUS = (('S', _('Submited')),
+              ('A', _('Available')),
+              ('D', _('Disabled')),)
+    STATUS_DCT = {}
+    for key, label in STATUS:
+        STATUS_DCT[key] = label
+    status = models.CharField(_("Status"), max_length=1, choices=STATUS)
+    objects = models.GeoManager()
+
+    def __unicode__(self):
+        return self.name
+
+    class Meta:
+        ordering = ('subcategory__category', 'subcategory', 'status', 'name')
+        verbose_name = _("Point of interest")
+
+    def getLatitude(self):
+        '''Return the latitude
+        '''
+        return self.point.y
+
+    def getLongitude(self):
+        '''Return the longitude
+        '''
+        return self.point.x
+
+    def getProperty(self, propertymodel, safe=None):
+        """Get the property of an associated property model.
+        If safe set to True, verify if the property is available
+        """
+        if safe and not propertymodel.available:
+            return
+        try:
+            property = Property.objects.get(propertymodel=propertymodel,
+                                            marker=self)
+        except Property.DoesNotExist:
+            return
+        return property
+
+    def getProperties(self):
+        """Get all the property availables
+        """
+        properties = []
+        for pm in PropertyModel.objects.filter(available=True):
+            property = self.getProperty(pm)
+            if property:
+                properties.append(property)
+        return properties
+
+    def getGeoJSON(self):
+        '''Return a GeoJSON string
+        '''
+        return """{"type":"Feature", "geometry":{"type":"Point", "crs": "EPSG:%(epsg)d", "coordinates":[%(longitude)s, %(latitude)s]}, "properties":{"pk": %(id)d, "name": "%(name)s", "icon_path":"%(icon_path)s"}}""" \
+ % {'id':self.id, 'name':self.name, 'icon_path':self.subcategory.icon.image,
+'latitude':self.getLatitude(), 'longitude':self.getLongitude(),
+'epsg':settings.EPSG_PROJECTION}
+
+class PropertyModel(models.Model):
+    '''Model for a property
+    '''
+    name = models.CharField(_("Name"), max_length=150)
+    order = models.IntegerField(_("Order"))
+    available = models.BooleanField(_("Available"))
+    TYPE = (('T', _('Text')),
+            ('L', _('Long text')),
+            ('P', _('Password')))
+    TYPE_WIDGET = {'T':'TextInput',
+                   'L':'Textarea',
+                   'P':'PasswordInput'}
+    type = models.CharField(_("Type"), max_length=1, choices=TYPE)
+    def __unicode__(self):
+        return self.name
+    class Meta:
+        ordering = ('order',)
+        verbose_name = _("Property model")
+
+    def getNamedId(self):
+        '''Get the name used as named id (easily sortable)
+        '''
+        return 'property_%d_%d' % (self.order, self.id)
+
+class Property(models.Model):
+    '''Property for a POI
+    '''
+    marker = models.ForeignKey(Marker, verbose_name=_("Point of interest"))
+    propertymodel = models.ForeignKey(PropertyModel,
+                                      verbose_name=_("Property model"))
+    value = models.CharField(_("Value"), max_length=1000)
+    def __unicode__(self):
+        return "%s : %s" % (str(self.propertymodel), self.value)
+    class Meta:
+        verbose_name = _("Property")
+