update xss filter

2017-12-24 15:34:22 +08:00
parent 03cf356bab
commit a5fea0c653
4 changed files with 29 additions and 18 deletions
--- a/utils/models.py
+++ b/utils/models.py
@@ -1,14 +1,10 @@
 from django.contrib.postgres.fields import JSONField  # NOQA
 from django.db import models

-from utils.xss_filter import XssHtml
+from utils.xss_filter import XSSHtml


 class RichTextField(models.TextField):
    def get_prep_value(self, value):
-        if not value:
-            value = ""
-        parser = XssHtml()
-        parser.feed(value)
-        parser.close()
-        return parser.getHtml()
+        with XSSHtml() as parser:
+            return parser.clean(value or "")