Aroy/Gallery-Archivist
1
0
Fork 0
Code Issues 5 Pull requests Projects Releases Packages 1 Wiki Activity Actions

Add: function to check if string is html

Browse source
This commit is contained in:
Aroy-Art 2023-09-27 19:08:27 +02:00
parent c0f847aefb
commit b631c100b1
Signed by: Aroy
GPG key ID: 583642324A1D2070
1 changed files with 17 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
archivist/sites/furaffinity/templatetags/custom_filters.py
View file

@ -7,6 +7,23 @@ from bs4 import BeautifulSoup
register = template.Library() register = template.Library()
def is_html(string):
'''
Check if string is HTML
'''
soup = BeautifulSoup(string, "html.parser")
# Remove leading and trailing white space
stripped_string = string.strip()
stripped_soup = str(soup).strip()
# If the string remained the same after parsing with BeautifulSoup, it's probably not HTML
if stripped_string == stripped_soup:
return False
# If the string changed when parsed by BeautifulSoup, it's probably HTML
else:
return True
@register.filter @register.filter
def is_image(file_url): def is_image(file_url):
image_extensions = ['.png', '.jpg', '.jpeg', '.gif', '.webp'] image_extensions = ['.png', '.jpg', '.jpeg', '.gif', '.webp']