Add: custom template filters

archivist/sites/furaffinity/templatetags/custom_filters.py

@@ -0,0 +1,47 @@
+# custom_filters.py
+from django import template
+
+import nh3
+
+from bs4 import BeautifulSoup
+
+register = template.Library()
+
+@register.filter
+def is_image(file_url):
+    image_extensions = ['.png', '.jpg', '.jpeg', '.gif', '.webp']
+    return file_url.lower().endswith(tuple(image_extensions))
+ 
+
+@register.filter
+def is_flash(file_url):
+    flash_extensions = ['.swf']
+    return file_url.lower().endswith(tuple(flash_extensions))
+
+
+@register.filter
+def is_pdf(file_url):
+    pdf_extensions = ['.pdf']
+    return file_url.lower().endswith(tuple(pdf_extensions))
+
+@register.filter
+def descriptionHtml2Text(description):
+    if description.startswith("<div"):
+        cleanHtml = nh3.clean(description)
+        soup = BeautifulSoup(cleanHtml, "html.parser")
+        descriptionText = soup.get_text()
+    else:
+        descriptionText = description
+    
+    return descriptionText
+
+
+@register.filter
+def stylizeDescription(description):
+    if description.startswith("<div"):
+        stylizedDescription = nh3.clean(description)
+    
+    else:
+        stylizedDescription = description
+    
+    return stylizedDescription