forked from BilalY/Rasagar
92 lines
2.8 KiB
Python
92 lines
2.8 KiB
Python
|
# MakeYamlToc
|
||
|
|
import re
|
||
|
|
import os.path
|
||
|
|
from os import path
|
||
|
|
|
||
|
|
|
||
|
|
#
|
||
|
|
# Reads through the file line by line and turns the MD input into YAML output:
|
||
|
|
#
|
||
|
|
# * [display](href) => - name: display
|
||
|
|
# href: href
|
||
|
|
# <indent>* [display](href) => items:
|
||
|
|
# - name: display
|
||
|
|
# href: href
|
||
|
|
#
|
||
|
|
def ConvertTOC(infile) :
|
||
|
|
# storage
|
||
|
|
yamlstream = ""
|
||
|
|
plevel = 0
|
||
|
|
level = 0
|
||
|
|
indent = ""
|
||
|
|
tab = ""
|
||
|
|
newtab = " "
|
||
|
|
|
||
|
|
# action
|
||
|
|
f = open(infile, "r")
|
||
|
|
for line in f :
|
||
|
|
r = re.search("^(\s*?)\* \[(.*?)\]\((.*?)\).*?$", line)
|
||
|
|
if r != None :
|
||
|
|
# get matches
|
||
|
|
indent = r.group(1)
|
||
|
|
display = r.group(2)
|
||
|
|
href = r.group(3)
|
||
|
|
|
||
|
|
if indent != "" and tab == "" :
|
||
|
|
# set tab character (first time only)
|
||
|
|
tab = indent
|
||
|
|
level = 1
|
||
|
|
elif indent == "" :
|
||
|
|
# no indent (use two spaces)
|
||
|
|
level = 0
|
||
|
|
else :
|
||
|
|
# compare current match to tab to calculate level
|
||
|
|
m = re.findall(tab, indent)
|
||
|
|
level = len(m)
|
||
|
|
|
||
|
|
# analysis
|
||
|
|
indent = newtab * level
|
||
|
|
if level > plevel :
|
||
|
|
pindent = newtab * plevel
|
||
|
|
yamlstream += "\n"+pindent+" items:\n"+indent+"- name: "+display+"\n"+indent+" href: "+href
|
||
|
|
else :
|
||
|
|
yamlstream += "\n"+indent+"- name: "+display+"\n"+indent+" href: "+href
|
||
|
|
|
||
|
|
# prime the loop
|
||
|
|
plevel = level
|
||
|
|
|
||
|
|
# send back the results (minus the leading line break)
|
||
|
|
f.close()
|
||
|
|
return yamlstream.lstrip("\n")
|
||
|
|
|
||
|
|
|
||
|
|
#
|
||
|
|
# Main / launch point
|
||
|
|
#
|
||
|
|
tocstream = ""
|
||
|
|
mdfile = "TableOfContents.md"
|
||
|
|
if path.exists(mdfile) :
|
||
|
|
tocstream = ConvertTOC("TableOfContents.md")
|
||
|
|
else :
|
||
|
|
mdfile = raw_input("Enter the location of the TableOfContents.md file you want to convert: ")
|
||
|
|
if mdfile == "" :
|
||
|
|
print("Can't continue. You need to specify the location of the MD toc file (eg., `TableOfContents.md`)")
|
||
|
|
else :
|
||
|
|
if path.exists(mdfile) and path.isfile(mdfile) :
|
||
|
|
print("Processing "+mdfile+"...")
|
||
|
|
tocstream = ConvertTOC(path.abspath(mdfile))
|
||
|
|
elif not(path.exists(mdfile)) :
|
||
|
|
print("Can't continue. "+mdfile+" doesn't exist.")
|
||
|
|
else :
|
||
|
|
print("Can't continue. "+mdfile+" isn't a file.")
|
||
|
|
|
||
|
|
if tocstream == "" :
|
||
|
|
print("Failed to find convert the "+mdfile+" file to YAML.")
|
||
|
|
else :
|
||
|
|
print("Finished creating toc.yml file from "+mdfile+".")
|
||
|
|
basedir = path.dirname(path.abspath(mdfile))
|
||
|
|
outfile = path.join(basedir, "toc.yml")
|
||
|
|
o = open(outfile, "w")
|
||
|
|
o.write(tocstream)
|
||
|
|
o.close()
|