Allow users to add more elements

safehtmlform/fields.py

@@ -5,10 +5,16 @@
 class SafeHTMLField(forms.CharField):
     help_text=_("This field allows the use of the following HTML tags: %(tags)s." % {'tags': acceptable_elements})
     widget=forms.widgets.Textarea()
+
+    def __init__(self, *args, **kwargs):
+        self.acceptable_elements = acceptable_elements
+        additional_elements = kwargs.pop("additional_elements", None)
+        if additional_elements:
+            self.acceptable_elements += additional_elements
 
     def clean(self, value):
         """
         Cleans non-allowed HTML from the input.
         """
         value = super(SafeHTMLField, self).clean(value)
-        return sanitize_html(value)
+        return sanitize_html(value, elements=self.acceptable_elements)

safehtmlform/utils.py

@@ -22,13 +22,13 @@ def sanitize_attrs(attrs):
     return sanitized_attrs
 
 
-def sanitize_html(value):
+def sanitize_html(value, elements=acceptable_elements):
     soup = BeautifulSoup(value)
     for comment in soup.findAll(
         text=lambda text: isinstance(text, Comment)):
         comment.extract()
     for tag in soup.findAll(True):
-        if tag.name not in acceptable_elements:
+        if tag.name not in elements:
             tag.hidden = True
         tag.attrs = sanitize_attrs(tag.attrs)
     return soup.renderContents().decode('utf8').replace('javascript:', '')