Edit form: disable a subcat for submission only

1 files changed, 6 insertions, 2 deletions

diff --git a/chimere/models.py b/chimere/models.py
index 99fa073..1ca5db9 100644
--- a/chimere/models.py
+++ b/chimere/models.py
@@ -148,7 +148,9 @@ class SubCategory(models.Model):
     '''
     category = models.ForeignKey(Category, verbose_name=_(u"Category"))
     name = models.CharField(_(u"Name"), max_length=150)
-    available = models.BooleanField(_(u"Available"))
+    available = models.BooleanField(_(u"Available"), default=True)
+    submission = models.BooleanField(_(u"Available for submission"),
+                                     default=True)
     icon = models.ForeignKey(Icon, verbose_name=_(u"Icon"))
     color_theme = models.ForeignKey(ColorTheme, verbose_name=_(u"Color theme"),
                                     blank=True, null=True)
@@ -165,7 +167,7 @@ class SubCategory(models.Model):
         verbose_name_plural = _(u"Sub-categories")
 
     @classmethod
-    def getAvailable(cls, item_types=None, area_name=None):
+    def getAvailable(cls, item_types=None, area_name=None, public=False):
         '''Get list of tuples with first the category and second the associated
         subcategories
         '''
@@ -175,6 +177,8 @@ class SubCategory(models.Model):
             subcategories = subcategories.filter(available=True)
         else:
             subcategories = subcategories.filter(item_type__in=item_types)
+        if public:
+            subcategories = subcategories.filter(submission=True)
         selected_cats = []
         if area_name:
             area = Area.objects.get(urn=area_name)