Admin: add multimedia and picture formsets to polygon and route

1 files changed, 20 insertions, 2 deletions

diff --git a/chimere/admin.py b/chimere/admin.py
index 55bc972..333db6d 100644
--- a/chimere/admin.py
+++ b/chimere/admin.py
@@ -396,6 +396,15 @@ class MarkerAdmin(admin.ModelAdmin):
         return form
 
 
+class PictureRouteInline(PictureMarkerInline):
+    exclude = ('thumbnailfile', 'thumbnailfile_height', 'thumbnailfile_width',
+               'polygon', 'marker')
+
+
+class MultimediaRouteInline(MultimediaMarkerInline):
+    exclude = ('polygon', 'marker')
+
+
 class RouteAdmin(MarkerAdmin):
     """
     Specialized the Route field.
@@ -409,7 +418,7 @@ class RouteAdmin(MarkerAdmin):
     actions = [validate, disable, managed_modified, export_to_kml,
                export_to_shapefile, export_to_csv]
     fieldsets = ROUTE_FIELDSETS
-    inlines = []
+    inlines = [MultimediaRouteInline, PictureRouteInline]
     has_properties = False
     geo_type = 'route'
 
@@ -429,6 +438,15 @@ class RouteAdmin(MarkerAdmin):
                                 Route.objects.filter(pk=item_id))
 
 
+class PicturePolygonInline(PictureMarkerInline):
+    exclude = ('thumbnailfile', 'thumbnailfile_height', 'thumbnailfile_width',
+               'route', 'marker')
+
+
+class MultimediaPolygonInline(MultimediaMarkerInline):
+    exclude = ('route', 'marker')
+
+
 class PolygonAdmin(MarkerAdmin):
     """
     Specialized the Polygon field.
@@ -442,7 +460,7 @@ class PolygonAdmin(MarkerAdmin):
         'submiter_name', 'ref_item', 'modified_since_import']
     exclude = ['submiter_session_key', 'import_key', 'import_version',
                'ref_item']
-    inlines = []
+    inlines = [MultimediaPolygonInline, PicturePolygonInline]
     fieldsets = POLYGON_FIELDSETS
     geo_type = 'polygon'