浏览代码

remove undesired <html> and <body> tags from bsoup

Tastalian 10 年之前
父节点
当前提交
1e1f8999b9
共有 1 个文件被更改,包括 2 次插入0 次删除
  1. 2 0
      extract_toc/extract_toc.py

+ 2 - 0
extract_toc/extract_toc.py

@@ -41,6 +41,8 @@ def extract_toc(content):
         toc.extract()
         content._content = soup.decode()
         content.toc = toc.decode()
+        if content.toc.startswith('<html>'):
+            content.toc = content.toc[12:-14]
 
 
 def register():