Manage auto-detection of multimedia type by extension - improvement on alternate multimedia widget

Conflicts:
	chimere/admin.py
	chimere/forms.py
	chimere/models.py
	chimere/templates/chimere/blocks/alternate_multimedia.html
	chimere/templatetags/chimere_tags.py

1 files changed, 28 insertions, 3 deletions

diff --git a/chimere/forms.py b/chimere/forms.py
index efea3dd..813f0f8 100644
--- a/chimere/forms.py
+++ b/chimere/forms.py
@@ -32,8 +32,9 @@ 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, MultimediaType, \
-    PictureFile, Importer, IMPORTER_CHOICES
+   News, Category, SubCategory, RouteFile, MultimediaFile, MultimediaType, \
+   PictureFile, Importer, IMPORTER_CHOICES, PropertyModelChoice, IFRAME_LINKS,\
+   MultimediaExtension
 from chimere.widgets import AreaField, PointField, TextareaWidget, \
     ImportFiltrWidget, TextareaAdminWidget, DatePickerWidget, \
     ImporterChoicesWidget, RE_XAPI
@@ -425,7 +426,31 @@ class MultimediaFileForm(BaseFileForm):
     def __init__(self, *args, **kwargs):
         super(MultimediaFileForm, self).__init__(*args, **kwargs)
         self.fields['multimedia_type'].widget.choices = \
-                                              MultimediaType.get_tuples()
+                                                MultimediaType.get_tuples()
+
+    def clean(self):
+        if not self.cleaned_data.get('multimedia_type') or\
+           not self.cleaned_data.get('url'):
+            return self.cleaned_data
+        if self.cleaned_data['multimedia_type'].name != 'auto':
+            return self.cleaned_data
+        url = self.cleaned_data['url']
+        for mm_type in IFRAME_LINKS:
+            res, embeded_url = IFRAME_LINKS[mm_type]
+            if [r for r in res if r.search(url)]:
+                multimedia_type = MultimediaType.objects.get(
+                                    name__iexact=mm_type)
+                self.cleaned_data['multimedia_type'] = multimedia_type
+                return self.cleaned_data
+        ext = url.split(".")[-1]
+        q = MultimediaExtension.objects.filter(name__iendswith=ext)
+        if q.count():
+            self.cleaned_data['multimedia_type'] = q.all()[0].multimedia_type
+        else:
+            # default to an iframe
+            self.cleaned_data['multimedia_type'] = \
+                MultimediaType.objects.filter(name__iexact='iframe').all()[0]
+        return self.cleaned_data
 
 MultimediaFileFormSet = formset_factory(MultimediaFileForm, can_delete=True)