rDrama
/
rDrama
9
10
Fork 8
Code Issues Pull Requests Packages Releases Wiki Activity

Upgrade bleach to 5.0.0.

remotes/1693045480750635534/spooky-22
Snakes 2022-05-24 20:27:41 -04:00
parent 826f137a5e
commit 7ead30014c
2 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
files/helpers/sanitize.py
View File

@ -1,5 +1,6 @@
import bleach import bleach
from bs4 import BeautifulSoup from bs4 import BeautifulSoup
from bleach.css_sanitizer import CSSSanitizer
from bleach.linkifier import LinkifyFilter, build_url_re from bleach.linkifier import LinkifyFilter, build_url_re
from functools import partial from functools import partial
from .get import * from .get import *
@ -37,6 +38,8 @@ allowed_tags = ('b','blockquote','br','code','del','em','h1','h2','h3','h4','h5'
'li','ol','p','pre','strong','sub','sup','table','tbody','th','thead','td','tr','ul', 'li','ol','p','pre','strong','sub','sup','table','tbody','th','thead','td','tr','ul',
'marquee','a','span','ruby','rp','rt','spoiler','img','lite-youtube','video','source','audio') 'marquee','a','span','ruby','rp','rt','spoiler','img','lite-youtube','video','source','audio')
allowed_styles = ['color', 'background-color', 'font-weight', 'text-align',]
def allowed_attributes(tag, name, value): def allowed_attributes(tag, name, value):
if name == 'style': return True if name == 'style': return True
@ -298,17 +301,15 @@ def sanitize(sanitized, alert=False, comment=False, edit=False):
sanitized = sanitized.replace('<html><body>','').replace('</body></html>','') sanitized = sanitized.replace('<html><body>','').replace('</body></html>','')
css_sanitizer = CSSSanitizer(allowed_css_properties=allowed_styles)
sanitized = bleach.Cleaner(tags=allowed_tags, sanitized = bleach.Cleaner(tags=allowed_tags,
attributes=allowed_attributes, attributes=allowed_attributes,
protocols=['http', 'https'], protocols=['http', 'https'],
styles=['color', 'background-color', 'font-weight', 'text-align'], css_sanitizer=css_sanitizer,
filters=[partial(LinkifyFilter, skip_tags=["pre"], parse_email=False, callbacks=[callback], url_re=url_re)] filters=[partial(LinkifyFilter, skip_tags=["pre"],
parse_email=False, callbacks=[callback], url_re=url_re)]
).clean(sanitized) ).clean(sanitized)
soup = BeautifulSoup(sanitized, 'lxml') soup = BeautifulSoup(sanitized, 'lxml')
links = soup.find_all("a") links = soup.find_all("a")

3
requirements.txt
View File

@ -1,5 +1,6 @@
beautifulsoup4 beautifulsoup4
bleach==4.1.0 bleach
bleach[css]
Flask Flask
Flask-Caching Flask-Caching
Flask-Compress Flask-Compress