Tests and fixes whitespace bugs in sanitizer

This commit is contained in:
Mouse Reeve 2020-05-09 18:30:24 -07:00
parent ab7acc6db6
commit db898e362b
1 changed files with 5 additions and 3 deletions

View File

@ -45,6 +45,8 @@ class InputHtmlParser(HTMLParser):
def get_output(self): def get_output(self):
''' convert the output from a list of tuples to a string ''' ''' convert the output from a list of tuples to a string '''
if self.tag_stack:
self.allow_html = False
if not self.allow_html: if not self.allow_html:
return ''.join(v for (k, v) in self.output if k == 'data') return ''.join(v for (k, v) in self.output if k == 'data')
return ''.join(v for (k, v) in self.output) return ''.join(v for (k, v) in self.output)