Ticket #589: FilePathField.diff

trunk/django/core/meta/fields.py

@@ -427 +427 @@
         f = os.path.join(self.get_directory_name(), get_valid_filename(os.path.basename(filename)))
         return os.path.normpath(f)
 
+class FilePathField(Field):
+    def __init__(self, verbose_name=None, name=None, path='', match='', recursive=False, **kwargs):
+        self.path = path
+        self.match = match
+        self.recursive = recursive
+        Field.__init__(self, verbose_name, name, **kwargs)
+    
+    def get_manipulator_fields(self, opts, manipulator, change, name_prefix='', rel=False):
+        if self.match != '':
+            import re
+            p = re.compile(self.match)
+        files = os.listdir(self.path)
+        theFileList = []
+        if self.recursive:
+            for root, dirs, files in os.walk(self.path):
+                for f in files:
+                    if self.match != '':
+                        m = p.match(f)
+                        if m:
+                            theFileList.append(tuple([ os.path.join(self.path, f), f]))
+                    else:
+                        theFileList.append(tuple([ os.path.join(self.path, f), f]))
+        else:
+            for f in files:
+                if os.path.isfile(os.path.join(self.path,f)):
+                    if self.match != '':
+                        m = p.match(f)
+                        if m:
+                                theFileList.append(tuple([ os.path.join(self.path,f), f]))
+                    else:
+                        theFileList.append(tuple([ os.path.join(self.path,f), f]))
+        self.choices = tuple(theFileList)
+        return Field.get_manipulator_fields(self, opts, manipulator, change, name_prefix, rel)
+    
+    def get_manipulator_field_objs(self):
+        return [formfields.TextField]
+
 class FloatField(Field):
     empty_strings_allowed = False
     def __init__(self, verbose_name=None, name=None, max_digits=None, decimal_places=None, **kwargs):