Correct image index

heqin-zhu · heqin-zhu · commit 7fcbeb592b05 · 2019-06-08T16:45:08.000+08:00
diff --git a/README.md b/README.md
@@ -93,6 +93,7 @@ course
     * [代数结构](./代数结构)
     * [光学与原子物理](./光学与原子物理)
     * [images](./images)
+    * [计算机图像学](./计算机图像学)
     * [计算机网络](./计算机网络)
     * [计算机系统详解](./计算机系统详解)
     * [计算机与信息类](./计算机与信息类)
diff --git a/utils/genIndex.py b/utils/genIndex.py
@@ -5,6 +5,16 @@
 import shutil
 from getSize import getSize
 from config import PATH, HTML, WALKDIR, TARDIR, IGNORE, NAME, DOWNLOAD
+
+URL = 'https://github.com/USTC-Resource/USTC-Course/tree/master/'
+ImagePre = ''
+ImagePT = re.compile(r'\!\[(.*?)\]\(([a-zA-Z\d\.].*?)\)')
+
+
+def subFunc(match):
+    name, suf = match.groups()
+    return f'![{name}]({ImagePre+"/"+suf})'
+
 hasPinyin = False
 try:
     from pypinyin import pinyin
@@ -120,7 +130,8 @@ def genIndex(path, dirs, files, htmlTemp=HTML):
         readme=md2html(md))
     filename = os.path.join(tar, NAME)
     with open(filename, 'w') as f:
-        f.write(cont)
+        ImagePre  = URL + path
+        f.write(re.sub(ImagePT,subFunc,cont))
 
 
 def getPath(path):
