refactors to use find to fix #17

BITPlan · Mar 14, 2021 · 367a631 · 367a631
1 parent 4c68377
commit 367a631
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/frontend/wikicms.py b/frontend/wikicms.py
@@ -126,9 +126,9 @@ def doFilter(self,html,filterKeys):
  # https://stackoverflow.com/questions/5598524/can-i-remove-script-tags-with-beautifulsoup
  soup = BeautifulSoup(html,'lxml')
  if "parser-output" in filterKeys:
- resultset=soup.select('div',{"class": "mw-parser-output"})
- if len(resultset)==1:
- soup=resultset[0].children.__next__()
+ parserdiv=soup.find('div',{"class": "mw-parser-output"})
+ if parserdiv:
+ soup=parserdiv.children.__next__()
  pass
  # https://stackoverflow.com/questions/5041008/how-to-find-elements-by-class
  if "editsection" in filterKeys: