Manage multimedia type tree

1 files changed, 8 insertions, 2 deletions

diff --git a/chimere/forms.py b/chimere/forms.py
index 4933c3a..430cb0c 100644
--- a/chimere/forms.py
+++ b/chimere/forms.py
@@ -1,6 +1,6 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
-# Copyright (C) 2008-2011  Étienne Loks  <etienne.loks_AT_peacefrogsDOTnet>
+# Copyright (C) 2008-2012  Étienne Loks  <etienne.loks_AT_peacefrogsDOTnet>
 
 # This program is free software: you can redistribute it and/or modify
 # it under the terms of the GNU General Public License as
@@ -31,7 +31,8 @@ from django.contrib.admin.widgets import AdminDateWidget
 from django.core.mail import EmailMessage, BadHeaderError
 
 from chimere.models import Marker, Route, PropertyModel, Property, Area,\
-     News, Category, SubCategory, RouteFile, MultimediaFile, PictureFile
+     News, Category, SubCategory, RouteFile, MultimediaFile, MultimediaType, \
+     PictureFile
 from chimere.widgets import AreaField, PointField, TextareaWidget, \
                             DatePickerWidget
 
@@ -317,6 +318,11 @@ class MultimediaFileForm(BaseFileForm):
         model = MultimediaFile
         exclude = ('order',)
 
+    def __init__(self, *args, **kwargs):
+        super(MultimediaFileForm, self).__init__(*args, **kwargs)
+        self.fields['multimedia_type'].widget.choices = \
+                                              MultimediaType.get_tuples()
+
 MultimediaFileFormSet = formset_factory(MultimediaFileForm, can_delete=True)
 
 class PictureFileForm(BaseFileForm):