Django

Code

Ticket #5894: diff.9.diff

File diff.9.diff, 5.5 kB (added by Alex, 6 months ago)

Refixed the tests, and made the patch more PEP8(thanks brosner)

  • django/db/models/fields/__init__.py

    old new  
    814814        self.path, self.match, self.recursive = path, match, recursive 
    815815        kwargs['max_length'] = kwargs.get('max_length', 100) 
    816816        Field.__init__(self, verbose_name, name, **kwargs) 
     817     
     818    def formfield(self, **kwargs): 
     819        defaults = { 
     820            'path': self.path, 
     821            'match': self.match, 
     822            'recursive': self.recursive, 
     823            'form_class': forms.FilePathField, 
     824        } 
     825        defaults.update(kwargs) 
     826        return super(FilePathField, self).formfield(**defaults) 
    817827 
    818828    def get_manipulator_field_objs(self): 
    819829        return [curry(oldforms.FilePathField, path=self.path, match=self.match, recursive=self.recursive)] 
  • django/newforms/fields.py

    old new  
    44 
    55import copy 
    66import datetime 
     7import os 
    78import re 
    89import time 
    910# Python 2.3 fallbacks 
     
    3132    'RegexField', 'EmailField', 'FileField', 'ImageField', 'URLField', 
    3233    'BooleanField', 'NullBooleanField', 'ChoiceField', 'MultipleChoiceField', 
    3334    'ComboField', 'MultiValueField', 'FloatField', 'DecimalField', 
    34     'SplitDateTimeField', 'IPAddressField', 
     35    'SplitDateTimeField', 'IPAddressField', 'FilePathField', 
    3536) 
    3637 
    3738# These values, if given to to_python(), will trigger the self.required check. 
     
    479480            raise ValidationError(self.error_messages['invalid_image']) 
    480481        return f 
    481482 
     483class FilePathField(ChoiceField): 
     484    def __init__(self, path, match=None, recursive=False, required=True, 
     485                 widget=Select, label=None, initial=None, help_text=None, 
     486                 *args, **kwargs): 
     487        self.path, self.match, self.recursive = path, match, recursive 
     488        super(FilePathField, self).__init__(choices=(), required=required, 
     489            widget=widget, label=label, initial=initial, help_text=help_text, 
     490            *args, **kwargs) 
     491        self.choices = [] 
     492        if self.match is not None: 
     493            self.match_re = re.compile(self.match) 
     494        if recursive: 
     495            for root, dirs, files in os.walk(self.path): 
     496                for f in files: 
     497                    if self.match is None or self.match_re.search(f): 
     498                        f = os.path.join(root, f) 
     499                        self.choices.append(f, f.replace(self.path, "", 1)) 
     500        else: 
     501            try: 
     502                for f in os.listdir(self.path): 
     503                    full_file = os.path.join(self.path, f) 
     504                    if os.path.isfile(full_file) and (self.match is None or self.match_re.search(f)): 
     505                        self.choices.append((full_file, f)) 
     506            except OSError: 
     507                pass 
     508        self.widget.choices = self.choices 
     509 
    482510url_re = re.compile( 
    483511    r'^https?://' # http:// or https:// 
    484512    r'(?:(?:[A-Z0-9-]+\.)+[A-Z]{2,6}|' #domain... 
  • tests/regressiontests/forms/fields.py

    old new  
    11061106>>> f.clean(None) 
    11071107u'' 
    11081108 
     1109# FilePathField ############################################################### 
     1110 
     1111>>> import re 
     1112>>> from django import newforms as forms 
     1113>>> loc = forms.__file__ 
     1114>>> m = re.search(r'(.*?)/__init__\..*$', loc) 
     1115>>> loc = m.group(1) + '/' 
     1116>>> loc 
     1117'.../django/newforms/' 
     1118>>> f = forms.FilePathField(path=loc) 
     1119>>> f.choices.sort() 
     1120>>> f.choices 
     1121[('.../django/newforms/__init__.py', '__init__.py'), ('.../django/newforms/__init__.pyc', '__init__.pyc'), ('.../django/newforms/fields.py', 'fields.py'), ('.../django/newforms/fields.pyc', 'fields.pyc'), ('.../django/newforms/fields.py~', 'fields.py~'), ('.../django/newforms/forms.py', 'forms.py'), ('.../django/newforms/forms.pyc', 'forms.pyc'), ('.../django/newforms/models.py', 'models.py'), ('.../django/newforms/models.pyc', 'models.pyc'), ('.../django/newforms/util.py', 'util.py'), ('.../django/newforms/util.pyc', 'util.pyc'), ('.../django/newforms/widgets.py', 'widgets.py'), ('.../django/newforms/widgets.pyc', 'widgets.pyc')] 
     1122>>> f.clean('fields.py') 
     1123Traceback (most recent call last): 
     1124... 
     1125ValidationError: [u'Select a valid choice. That choice is not one of the available choices.'] 
     1126>>> f.clean(loc+'fields.py') 
     1127u'.../django/newforms/fields.py' 
     1128 
    11091129# SplitDateTimeField ########################################################## 
    11101130 
    11111131>>> f = SplitDateTimeField() 
  • docs/newforms.txt

    old new  
    13451345 
    13461346.. _`bind the file data to the form`: `Binding uploaded files to a form`_ 
    13471347 
     1348``FilePathField`` 
     1349~~~~~~~~~~~~~~ 
     1350 
     1351**New in Django development version** 
     1352 
     1353    * Default widget: ``Select`` 
     1354    * Empty value: ``None`` 
     1355    * Normalizes to: A unicode object 
     1356    * Validates that the selected choice exists in the list of choices. 
     1357    * Error message keys: ``required``, ``invalid_choice`` 
     1358 
     1359The choices are generated from the filesystem.  They can be forced to match a specific regulat expression(this only matches the file, not the entire path).  It can also be recursive. 
     1360 
    13481361``ImageField`` 
    13491362~~~~~~~~~~~~~~ 
    13501363