Manage specific CSS for areas inside admin

1 files changed, 6 insertions, 1 deletions

diff --git a/chimere/models.py b/chimere/models.py
index 96430ff..f9a1e65 100644
--- a/chimere/models.py
+++ b/chimere/models.py
@@ -172,16 +172,19 @@ class SubCategory(models.Model):
             subcategories = subcategories.filter(available=True)
         else:
             subcategories = subcategories.filter(item_type__in=item_types)
+        selected_cats = []
         if area_name:
             area = Area.objects.get(urn=area_name)
             # if there some restrictions with categories limit them
             if area.subcategories.count():
                 sub_ids = [sub.id for sub in area.subcategories.all()]
                 subcategories = subcategories.filter(id__in=sub_ids)
+            selected_cats = [subcat.pk
+                             for subcat in area.default_subcategories.all()]
         for sub_category in subcategories:
             if sub_category.category not in sub_categories:
                 sub_categories[sub_category.category] = []
-            if sub_category.id in settings.CHIMERE_DEFAULT_CATEGORIES:
+            if sub_category.id in selected_cats:
                 sub_category.selected = True
                 sub_category.category.selected = True
             sub_categories[sub_category.category].append(sub_category)
@@ -801,6 +804,8 @@ class Area(models.Model, SimpleArea):
        blank=True, db_table='chimere_subcategory_areas',
        verbose_name=_(u"Restricted to theses subcategories"),
        help_text=_(u"If no subcategory is set all subcategories are available"))
+    external_css = models.URLField(_(u"Link to an external CSS"), blank=True,
+                                   null=True)
     objects = models.GeoManager()
 
     def __unicode__(self):