* fix bug

* 一口老血。。。
YaoZeyuan · Dec 14, 2015 · d73b5bf · d73b5bf
1 parent 1bca50a
commit d73b5bf
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/tools/create_html.py b/src/tools/create_html.py
@@ -19,25 +19,27 @@ def __init__(self, image_container):
     def fix_image(self, content):
         content = Match.fix_html(content)
         for img in re.findall(r'<img[^>]*', content):
+            # fix img
             if img[-1] == '/':
                 img = img[:-1]
+            img += '>'
+
             src = re.search(r'(?<=src=").*?(?=")', img)
             if not src:
-                new_image = img + '></img>'
+                new_image = img + '</img>'
                 content = content.replace(img, new_image)
                 continue
             else:
                 src = src.group(0)
                 if src.replace(' ', '') == '':
-                    new_image = img + '></img>'
+                    new_image = img + '</img>'
                     content = content.replace(img, new_image)
                     continue
             src_download = CreateHtml.fix_image_src(src)
             if src_download:
                 filename = self.image_container.add(src_download)
             else:
                 filename = ''
-            img += '>'
             new_image = img.replace('"{}"'.format(src), '"../images/{}"'.format(filename))
             new_image = new_image.replace('//zhstatic.zhihu.com/assets/zhihu/ztext/whitedot.jpg',
                                           '../images/{}'.format(filename))