Ticket #3515: fields.3.diff

django/newforms/fields.py

@@ -35 +35 @@
     # Tracks each time a Field instance is created. Used to retain order.
     creation_counter = 0
 
-    def __init__(self, required=True, widget=None, label=None, initial=None, help_text=None):
+    def __init__(self, required=True, widget=None, label=None, initial=None, help_text=None, css=None):
         # required -- Boolean that specifies whether the field is required.
         #             True by default.
         # widget -- A Widget class, or instance of a Widget class, that should be
@@ -48 +48 @@
         # initial -- A value to use in this Field's initial display. This value is
         #            *not* used as a fallback if data isn't given.
         # help_text -- An optional string to use as "help text" for this Field.
+        # css -- An optional string to set the CSS class name(s) for this Field.
         if label is not None:
             label = smart_unicode(label)
         self.required, self.label, self.initial = required, label, initial
@@ -60 +61 @@
         extra_attrs = self.widget_attrs(widget)
         if extra_attrs:
             widget.attrs.update(extra_attrs)
+
+        # Set initial CSS class name(s)
+        if widget.attrs.has_key('class'):
+            widget.attrs.remove('class')
+        if css or required:
+            if required: css = (css or '') + 'required'
+            widget.attrs.update({'class': css.lstrip()})
 
         self.widget = widget