Ticket #3297: 4549-with-tests-and-imagefield.patch

django/db/models/fields/__init__.py

@@ -658 +658 @@
             else:
                 func(new_data[upload_field_name]["filename"], new_data[upload_field_name]["content"])
 
+    def formfield_save_file(self, field_name, new_data, new_object):
+        func = getattr(new_object, 'save_%s_file' % self.name)
+        func(new_data[field_name][0]["filename"], new_data[field_name][0]["content"])
+
     def get_directory_name(self):
         return os.path.normpath(datetime.datetime.now().strftime(self.upload_to))
 
@@ -666 +670 @@
         f = os.path.join(self.get_directory_name(), get_valid_filename(os.path.basename(filename)))
         return os.path.normpath(f)
 
+    def formfield(self, **kwargs):
+        defaults = {'required': not self.blank, 'label': capfirst(self.verbose_name), 'widget': forms.FileInput}
+        defaults.update(kwargs)
+        return forms.FileField(**defaults)
+
 class FilePathField(Field):
     def __init__(self, verbose_name=None, name=None, path='', match=None, recursive=False, **kwargs):
         self.path, self.match, self.recursive = path, match, recursive
@@ -712 +721 @@
                 setattr(new_object, self.height_field, getattr(original_object, self.height_field))
             new_object.save()
 
+    def formfield(self, **kwargs):
+        defaults = {'required': not self.blank, 'label': capfirst(self.verbose_name), 'widget': forms.FileInput}
+        defaults.update(kwargs)
+        return forms.ImageField(**defaults)
+
 class IntegerField(Field):
     empty_strings_allowed = False
     def get_manipulator_field_objs(self):

django/newforms/models.py

@@ -16 +16 @@
 
     This method is created for any form_for_model Form.
     """
+    from django.db import models
     if self.errors:
         raise ValueError("The %s could not be created because the data didn't validate." % self._model._meta.object_name)
     return save_instance(self, self._model(), commit)
@@ -36 +37 @@
     for f in opts.fields:
         if not f.editable or isinstance(f, models.AutoField):
             continue
+        if isinstance(f,models.FileField):
+            continue
         setattr(instance, f.name, clean_data[f.name])
     if commit:
         instance.save()
         for f in opts.many_to_many:
             setattr(instance, f.attname, clean_data[f.name])
+        for f in opts.fields:
+            if isinstance(f, models.FileField):
+                f.formfield_save_file(f.name, clean_data, instance)
     # GOTCHA: If many-to-many data is given and commit=False, the many-to-many
     # data will be lost. This happens because a many-to-many options cannot be
     # set on an object until after it's saved. Maybe we should raise an

django/newforms/fields.py

@@ -10 +10 @@
 import time
 
 __all__ = (
-    'Field', 'CharField', 'IntegerField',
+    'Field', 'CharField', 'FileField', 'ImageField', 'IntegerField',
     'DEFAULT_DATE_INPUT_FORMATS', 'DateField',
     'DEFAULT_TIME_INPUT_FORMATS', 'TimeField',
     'DEFAULT_DATETIME_INPUT_FORMATS', 'DateTimeField',
@@ -107 +107 @@
         if self.max_length is not None and isinstance(widget, (TextInput, PasswordInput)):
             return {'maxlength': str(self.max_length)}
 
+class FileField(Field):
+
+    prev_error = False
+ 
+    def clean(self, value):
+        super(FileField, self).clean(value)
+
+        if self.required:
+            if value[0]:
+                if not len(value[0]['content'])>0:
+                    self.prev_error = True
+                    raise ValidationError(gettext(u'This field is required.'))
+            else:
+                if not value[1]:
+                    self.prev_error = True
+                    raise ValidationError(gettext(u'This field is required.'))
+        return value
+
+class ImageField(FileField):
+    
+    def clean(self, value):
+        super(ImageField, self).clean(value)
+        if value[0] and not value[1] and not self.prev_error:
+            from PIL import Image
+            from cStringIO import StringIO
+            try:
+                Image.open(StringIO(value[0]['content']))
+            except IOError: # Python Imaging Library doesn't recognize it as an image
+                raise ValidationError, gettext(u'Upload a valid image. The file you uploaded was either not an image or a corrupted image.')
+        return value
+
 class IntegerField(Field):
     def __init__(self, max_value=None, min_value=None, *args, **kwargs):
         self.max_value, self.min_value = max_value, min_value

django/newforms/widgets.py

@@ -115 +115 @@
 
 class FileInput(Input):
     input_type = 'file'
+    
+    def render(self,name,value,attrs=None,choices=()):
+        if value is None: value = ''
+        file_attrs = self.build_attrs(attrs, type='file', name=name+'_file')
+        final_attrs = self.build_attrs(attrs, type='hidden', name=name)
+        if value != '': final_attrs['value'] = smart_unicode(value) # only add the value attribute if a value is non-empty
+        
+        if value != '':
+            currently = u'Currently: %s<br/>Change: ' % smart_unicode(value)
+            current = u' <input%s />' % flatatt(final_attrs)
+        else:
+            currently = u''
+            current = u''
+         
+        return u'%s<input%s />%s' % (currently,flatatt(file_attrs),current)
+    
+    def value_from_datadict(self,data,name):
+        return [data.get(name+'_file',None),data.get(name,None)]
 
 class Textarea(Widget):
     def render(self, name, value, attrs=None):

tests/modeltests/model_forms/models.py

@@ -54 +54 @@
     def __str__(self):
         return self.headline
 
+class File(models.Model):
+    description = models.CharField(maxlength=50)
+    file = models.FileField(upload_to='files')
+
+    def __str__(self):
+        return self.description
+
 __test__ = {'API_TESTS': """
 >>> from django.newforms import form_for_model, form_for_instance, save_instance, BaseForm, Form, CharField
 >>> import datetime
@@ -370 +377 @@
 Traceback (most recent call last):
 ...
 ValidationError: [u'Select a valid choice. 5 is not one of the available choices.']
+
+Test for filefield
+
+>>> form_data = {'description': 'FileField test', 'file_file': {'content': 'FileField test', 'filename': 'filetest.txt'} }
+
+>>> TestFileForm = form_for_model(File)
+>>> empty_fileform = TestFileForm(auto_id=False)
+>>> print empty_fileform.as_ul()
+<li>Description: <input type="text" name="description" maxlength="50" /></li>
+<li>File: <input type="file" name="file_file" /></li>
+
+Test with data
+
+>>> filled_fileform= TestFileForm(form_data, auto_id=False)
+>>> filled_fileform.is_valid()
+True
+
+>>> instance = filled_fileform.save()
+>>> print instance.file
+files/filetest.txt
+>>> instance.delete()
 """}

tests/regressiontests/forms/tests.py

@@ -165 +165 @@
 # FileInput Widget ############################################################
 
 >>> w = FileInput()
->>> w.render('email', '')
-u'<input type="file" name="email" />'
->>> w.render('email', None)
-u'<input type="file" name="email" />'
->>> w.render('email', 'test@example.com')
-u'<input type="file" name="email" value="test@example.com" />'
->>> w.render('email', 'some "quoted" & ampersanded value')
-u'<input type="file" name="email" value="some &quot;quoted&quot; &amp; ampersanded value" />'
->>> w.render('email', 'test@example.com', attrs={'class': 'fun'})
-u'<input type="file" name="email" value="test@example.com" class="fun" />'
+>>> w.render('file', '')
+u'<input type="file" name="file_file" />'
+>>> w.render('file', 'file/filetest.txt')
+u'Currently: file/filetest.txt<br/>Change: <input type="file" name="file_file" /> <input type="hidden" name="file" value="file/filetest.txt" />'
 
-You can also pass 'attrs' to the constructor:
->>> w = FileInput(attrs={'class': 'fun'})
->>> w.render('email', '')
-u'<input type="file" class="fun" name="email" />'
->>> w.render('email', 'foo@example.com')
-u'<input type="file" class="fun" value="foo@example.com" name="email" />'
-
->>> w.render('email', 'ŠĐĆŽćžšđ', attrs={'class': 'fun'})
-u'<input type="file" class="fun" value="\u0160\u0110\u0106\u017d\u0107\u017e\u0161\u0111" name="email" />'
-
 # Textarea Widget #############################################################
 
 >>> w = Textarea()