【python】为markdown自动生成目录


markdown很优雅,但层级一多浏览起来就不够优雅了。
我们需要可跳转的目录,这样就可以随时按home键回到目录,再跳转到文章的任意部分。
结合[name](#hid)

name

可以做出可跳转的目录列表,但一个个做就有点麻烦。
这个脚本能生效的前提是标题书写符合规范,即若干个#加上若干个空格,当然,也可以修改文件头部正则匹配的pattern。
目录效果可以参照这篇文章,当然,自定义也很简单。
在GitHub上可获取最新版本。

#-*-coding:utf-8-*-
import re,sys
d={"#":1,"##":2,"###":3,"####":4,"#####":5,"######":6}
pattern='#+\s'
def usage():
	print "usage:"
	print "python script.py srcFilename.md"
	print "then you will get a res.md with contents "
	print "under the same path as srcFile\nenjoy!"
def ganMenu(filename):
  headId=0
  targetname="res.md"
  with open(targetname,'w+') as f2:
      with open(filename,'r') as f:
          for i in f.readlines():
            if not re.match(pattern,i.strip(' \t\n')): 
              continue
            i=i.strip(' \t\n') 
            head=i.split(' ')[0]
            f2.write('|'+'-----'*(len(head)-1)+'@['+i[len(head):].strip(' \t\n')+'](#id'+str(headId)+')   \n')
            headId+=1
      headId=0     
      with open(filename,'r') as f  :
        for i in f.readlines():
            if not re.match(pattern,i.strip(' \t\n')):   
              f2.write(i)
            else:
              i=i.strip(' \t\n')
              head=i.split(' ')[0]
              if head in d.keys():
                menu=''.join(['',i[len(head):].strip(' \t\n'),'   \n'])
                f2.write(menu)
                headId+=1

if __name__ == '__main__':
	try:
		ganMenu(sys.argv[1])   
	except:
		usage()