gen-summary.py

#!/usr/bin/env python

import os

summary = ''
summary += '# Summary\n'

extend = ' ' * 4
link_format = '* [{title}]({path})\n'

cur_depth = 1

def gen_summary(path):
    global cur_depth

    tmp = ''

    file_list = os.listdir(path)
    for f in file_list:
        file_path = os.path.join(path, f)

        if not os.path.isdir(file_path) and os.path.splitext(f)[1] == '.md':
            tmp += extend * (cur_depth - 1) + link_format.format(title=f, path=file_path)
            continue

        if not os.path.isdir(file_path):
            continue

        cur_depth += 1
        tmp1 = gen_summary(file_path)
        if tmp1:
            tmp += extend * (cur_depth - 2) + link_format.format(title=f, path='')
            tmp += tmp1
        cur_depth -= 1

    return tmp


if __name__ == '__main__':
    import sys
    d = sys.argv[1]

    summary += gen_summary(d)
    print(summary)