# coding:utf-8
"""
作者：雪山凌狐
网址：http://fox-9.com
修订时间：2022-09-14

写书好伴侣，删除生成的_book文件夹中的所有多余的md文件
"""

# 这里打印出有颜色的内容必须\n换行才能看到颜色，暂时没找到办法不换行同一行多种颜色的，以后再想想

import os
from color_print import printBlue, printGreen

# gitbook生成的文件夹名，默认是_book
folder_name = "_book"


# 遍历文件夹
def walkFile_and_delete_md(folder):
    for root, dirs, files in os.walk(folder):
        # root 表示当前正在访问的文件夹路径
        # dirs 表示该文件夹下的子目录名list
        # files 表示该文件夹下的文件list
        # 遍历文件
        for f in files:
            if os.path.splitext(f)[1].lower() == ".md":
                filename = os.path.join(root, f)
                os.remove(filename)
                printGreen(f"info: Deleted file [{filename}]\n")


def main():
    walkFile_and_delete_md(folder_name)
    printBlue("\ninfo: Delete all extra md files done!\n\n")


if __name__ == '__main__':
    main()