Ticket #8306: admin_field_overrides.diff

django/contrib/admin/options.py

@@ -125 +125 @@
     radio_fields = {}
     prepopulated_fields = {}
 
+    field_type_overrides = { models.DateTimeField: { 'form_class': forms.SplitDateTimeField,
+                                                     'widget': widgets.AdminSplitDateTime },
+                             models.DateField: { 'widget': widgets.AdminDateWidget },
+                             models.TimeField: { 'widget': widgets.AdminTimeWidget },
+                             models.TextField: { 'widget': widgets.AdminTextareaWidget },
+                             models.URLField: { 'widget': widgets.AdminURLFieldWidget },
+                             models.IntegerField: { 'widget': widgets.AdminIntegerFieldWidget },
+                             models.CharField: { 'widget': widgets.AdminTextInputWidget },
+                             models.ImageField: { 'widget': widgets.AdminFileWidget },
+                             models.FileField: { 'widget': widgets.AdminFileWidget },
+                             }
+
     def formfield_for_dbfield(self, db_field, **kwargs):
         """
         Hook for specifying the form Field instance for a given database Field
@@ -150 +162 @@
                 # Otherwise, use the default select widget.
                 return db_field.formfield(**kwargs)
 
-        # For DateTimeFields, use a special field and widget.
-        if isinstance(db_field, models.DateTimeField):
-            kwargs['form_class'] = forms.SplitDateTimeField
-            kwargs['widget'] = widgets.AdminSplitDateTime()
-            return db_field.formfield(**kwargs)
+        if db_field.__class__ in self.field_type_overrides:
+            return dbfield.formfield(**dict(kwargs, **self.field_type_overrides[db_field.__class__]))
 
-        # For DateFields, add a custom CSS class.
-        if isinstance(db_field, models.DateField):
-            kwargs['widget'] = widgets.AdminDateWidget
-            return db_field.formfield(**kwargs)
-
-        # For TimeFields, add a custom CSS class.
-        if isinstance(db_field, models.TimeField):
-            kwargs['widget'] = widgets.AdminTimeWidget
-            return db_field.formfield(**kwargs)
-        
-        # For TextFields, add a custom CSS class.
-        if isinstance(db_field, models.TextField):
-            kwargs['widget'] = widgets.AdminTextareaWidget
-            return db_field.formfield(**kwargs)
-        
-        # For URLFields, add a custom CSS class.
-        if isinstance(db_field, models.URLField):
-            kwargs['widget'] = widgets.AdminURLFieldWidget
-            return db_field.formfield(**kwargs)
-        
-        # For IntegerFields, add a custom CSS class.
-        if isinstance(db_field, models.IntegerField):
-            kwargs['widget'] = widgets.AdminIntegerFieldWidget
-            return db_field.formfield(**kwargs)
-
-        # For TextInputs, add a custom CSS class.
-        if isinstance(db_field, models.CharField):
-            kwargs['widget'] = widgets.AdminTextInputWidget
-            return db_field.formfield(**kwargs)
-    
-        # For FileFields and ImageFields add a link to the current file.
-        if isinstance(db_field, models.ImageField) or isinstance(db_field, models.FileField):
-            kwargs['widget'] = widgets.AdminFileWidget
-            return db_field.formfield(**kwargs)
-
         # For ForeignKey or ManyToManyFields, use a special widget.
         if isinstance(db_field, (models.ForeignKey, models.ManyToManyField)):
             if isinstance(db_field, models.ForeignKey) and db_field.name in self.raw_id_fields: