Weighted markers: model

1 files changed, 8 insertions, 0 deletions

diff --git a/chimere/models.py b/chimere/models.py
index 545d7e0..3400118 100644
--- a/chimere/models.py
+++ b/chimere/models.py
@@ -213,6 +213,8 @@ class SubCategory(models.Model):
     available = models.BooleanField(_(u"Available"), default=True)
     submission = models.BooleanField(_(u"Available for submission"),
                                      default=True)
+    weighted = models.BooleanField(_(u"Has an associated quantity"),
+                                   default=False)
     TYPE = (('M', _(u'Marker')),
             ('R', _(u'Route')),
             ('B', _(u'Both')),)
@@ -465,6 +467,8 @@ class Marker(GeographicItem):
                                           null=True) # used by feeds
     route = models.ForeignKey(u"Route", blank=True, null=True,
                               related_name='associated_marker')
+    weight = models.IntegerField(_(u"Quantity"), blank=True, null=True,
+                                 default=0)
     description = models.TextField(_(u"Description"), blank=True, null=True)
     is_front_page = models.NullBooleanField(_(u"Is front page"), blank=True,
                                             null=True)
@@ -534,6 +538,10 @@ class Marker(GeographicItem):
     def geom_attr(self):
         return 'point'
 
+    @property
+    def has_weight(self):
+        return bool(self.categories.filter(weighted=True).count())
+
     class Meta:
         ordering = ('status', 'name')
         verbose_name = _(u"Point of interest")