bootstrap3: teach add_css_class to prepend classes

Also update tests

Author: xrmx

bootstrap3/html.py

@@ -17,14 +17,16 @@ def split_css_classes(css_classes):
     return [c for c in classes_list if c]
 
 
-def add_css_class(css_classes, css_class):
+def add_css_class(css_classes, css_class, prepend=False):
     """
     Add a CSS class to a string of CSS classes
     """
     classes_list = split_css_classes(css_classes)
-    for c in split_css_classes(css_class):
-        if c not in classes_list:
-            classes_list.append(c)
+    classes_to_add = [c for c in split_css_classes(css_class) if c not in classes_list]
+    if prepend:
+        classes_list = classes_to_add + classes_list
+    else:
+        classes_list += classes_to_add
     return ' '.join(classes_list)
 
 

bootstrap3/tests.py

@@ -310,7 +310,10 @@ def __str__(self):
 
 class HtmlTest(TestCase):
     def test_add_css_class(self):
-        css_classes = "hey hi"
-        css_class = "there"
+        css_classes = "one two"
+        css_class = "three four"
         classes = add_css_class(css_classes, css_class)
-        self.assertEqual(classes, "hey hi there")
+        self.assertEqual(classes, "one two three four")
+
+        classes = add_css_class(css_classes, css_class, prepend=True)
+        self.assertEqual(classes, "three four one two")