skorotkiewicz/rDrama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fgdfg

Browse source
This commit is contained in:
Aevann1 2021-10-08 15:28:57 +02:00
parent f166c165e8
commit c76b2b4e63
1 changed files with 23 additions and 49 deletions
Download patch file Download diff file Expand all files Collapse all files

48
files/helpers/sanitize.py
View file

@ -75,45 +75,20 @@ no_images = ['b',
'span', 'span',
] ]
_allowed_attributes = { _allowed_attributes = {'*': ['href', 'style', 'src', 'class', 'title', 'rel', 'data-bs-original-name', 'direction']}
'*': ['href', 'style', 'src', 'class', 'title', 'rel', 'data-bs-original-name', 'direction']
}
_allowed_protocols = [
'http',
'https'
]
_allowed_styles =[ def sanitize(sanitized):
'color',
'font-weight',
'transform',
'-webkit-transform',
]
def sanitize(sanitized, noimages=False):
sanitized = sanitized.replace("\ufeff", "").replace("m.youtube.com", "youtube.com") sanitized = sanitized.replace("\ufeff", "").replace("m.youtube.com", "youtube.com")
for i in re.finditer('https://i.imgur.com/(([^_]*?)\.(jpg|png|jpeg))', sanitized): for i in re.finditer('https://i.imgur.com/(([^_]*?)\.(jpg|png|jpeg))', sanitized):
sanitized = sanitized.replace(i.group(1), i.group(2) + "_d." + i.group(3) + "?maxwidth=9999") sanitized = sanitized.replace(i.group(1), i.group(2) + "_d." + i.group(3) + "?maxwidth=9999")
if noimages:
sanitized = bleach.Cleaner(tags=no_images,
attributes=_allowed_attributes,
protocols=_allowed_protocols,
styles=_allowed_styles,
filters=[partial(LinkifyFilter,
skip_tags=["pre"],
parse_email=False,
)
]
).clean(sanitized)
else:
sanitized = bleach.Cleaner(tags=_allowed_tags, sanitized = bleach.Cleaner(tags=_allowed_tags,
attributes=_allowed_attributes, attributes=_allowed_attributes,
protocols=_allowed_protocols, protocols=['http', 'https'],
styles=_allowed_styles, styles=['color','font-weight','transform','-webkit-transform'],
filters=[partial(LinkifyFilter, filters=[partial(LinkifyFilter,
skip_tags=["pre"], skip_tags=["pre"],
parse_email=False, parse_email=False,
@ -125,7 +100,7 @@ def sanitize(sanitized, noimages=False):
for tag in soup.find_all("img"): for tag in soup.find_all("img"):
if tag.get("src", "") and "profile-pic-20" not in tag.get("class", ""): if tag.get("src") and "profile-pic-20" not in tag.get("class", ""):
tag["rel"] = "nofollow noopener noreferrer" tag["rel"] = "nofollow noopener noreferrer"
tag["class"] = "in-comment-image" tag["class"] = "in-comment-image"
@ -134,10 +109,9 @@ def sanitize(sanitized, noimages=False):
tag["src"] = "" tag["src"] = ""
link = soup.new_tag("a") link = soup.new_tag("a")
link["href"] = tag["src"] link["href"] = tag["data-src"]
link["rel"] = "nofollow noopener noreferrer" link["rel"] = "nofollow noopener noreferrer"
link["target"] = "_blank" link["target"] = "_blank"
link["onclick"] = f"expandDesktopImage('{tag['data-src']}');" link["onclick"] = f"expandDesktopImage('{tag['data-src']}');"
link["data-bs-toggle"] = "modal" link["data-bs-toggle"] = "modal"
link["data-bs-target"] = "#expandImageModal" link["data-bs-target"] = "#expandImageModal"
@ -145,7 +119,7 @@ def sanitize(sanitized, noimages=False):
tag.wrap(link) tag.wrap(link)
for tag in soup.find_all("a"): for tag in soup.find_all("a"):
if tag["href"]:
tag["target"] = "_blank" tag["target"] = "_blank"
if site not in tag["href"]: tag["rel"] = "nofollow noopener noreferrer" if site not in tag["href"]: tag["rel"] = "nofollow noopener noreferrer"
@ -153,8 +127,8 @@ def sanitize(sanitized, noimages=False):
try: tag.string = tag["href"] try: tag.string = tag["href"]
except: tag.string = "" except: tag.string = ""
sanitized = str(soup)
sanitized = str(soup)
start = '<s>' start = '<s>'
end = '</s>' end = '</s>'
@ -168,14 +142,14 @@ def sanitize(sanitized, noimages=False):
for i in re.finditer("[^a]>\s*(:!?\w+:\s*)+<\/", sanitized): for i in re.finditer("[^a]>\s*(:!?\w+:\s*)+<\/", sanitized):
old = i.group(0) old = i.group(0)
if 'marseylong' in old: new = old.lower().replace(">", " style='margin-bottom:0 !important'>") if 'marseylong1' in old or 'marseylong2' in old: new = old.lower().replace(">", " class='mb-0'>")
else: new = old.lower() else: new = old.lower()
for i in re.finditer('(?<!"):([^ ]{1,30}?):', new): for i in re.finditer('(?<!"):([^ ]{1,30}?):', new):
emoji = i.group(1).lower() emoji = i.group(1).lower()
if emoji.startswith("!"): if emoji.startswith("!"):
emoji = emoji[1:] emoji = emoji[1:]
if path.isfile(f'./files/assets/images/emojis/{emoji}.webp'): if path.isfile(f'./files/assets/images/emojis/{emoji}.webp'):
new = re.sub(f'(?<!"):!{emoji}:', f'<img loading="lazy" data-bs-toggle="tooltip" alt=":!{emoji}:" title=":!{emoji}:" delay="0" class="mirrored bigemoji" src="https://{site}/assets/images/emojis/{emoji}.webp" >', new) new = re.sub(f'(?<!"):!{emoji}:', f'<img loading="lazy" data-bs-toggle="tooltip" alt=":!{emoji}:" title=":!{emoji}:" delay="0" class="bigemoji mirrored" src="https://{site}/assets/images/emojis/{emoji}.webp" >', new)
if emoji in session["favorite_emojis"]: session["favorite_emojis"][emoji] += 1 if emoji in session["favorite_emojis"]: session["favorite_emojis"][emoji] += 1
else: session["favorite_emojis"][emoji] = 1 else: session["favorite_emojis"][emoji] = 1
@ -227,7 +201,7 @@ def sanitize(sanitized, noimages=False):
for i in re.finditer('<a href="(https://open.spotify.com/embed/.*?)"', sanitized): for i in re.finditer('<a href="(https://open.spotify.com/embed/.*?)"', sanitized):
url = i.group(1) url = i.group(1)
replacing = f'<a href="{url}" rel="nofollow noopener noreferrer" target="_blank">{url}</a>' replacing = f'<a href="{url}" rel="nofollow noopener noreferrer" target="_blank">{url}</a>'
htmlsource = f'<iframe data-src="{url}" frameBorder="0" allowtransparency="true" allow="encrypted-media"></iframe>' htmlsource = f'<iframe data-src="{url}" class="spotify" frameBorder="0" allowtransparency="true" allow="encrypted-media"></iframe>'
sanitized = sanitized.replace(replacing, htmlsource) sanitized = sanitized.replace(replacing, htmlsource)
for rd in ["https://reddit.com/", "https://new.reddit.com/", "https://www.reddit.com/", "https://redd.it/"]: for rd in ["https://reddit.com/", "https://new.reddit.com/", "https://www.reddit.com/", "https://redd.it/"]: