toc_script.py

import pathlib
from os import popen

# 配置项
Project_Name = "PyQt_practice"  # 项目名称，将显示在TOC文件中
Working_Directory = "./"  # 本脚本的工作目录
TOC_File_Output = "./toc.md"  # 输出的目录文件路径
Git_Ignore_File = ".gitignore"  # gitignore文件路径，用以排除不需要列目录的文件

Headline = f"""# TOC of {Project_Name}\n
This TOC file is auto-generated by *toc_scripy.py* @[muzing](https://muzing.top/about/).\n
"""

# 通过修改以下静态函数，即可将本脚本移植到其他需要生成目录的项目中使用


def dir_is_valid(dir_p: pathlib.PosixPath) -> bool:
    """
    验证目录有效性：只有以2位数字开头的文件夹才是需要的
    """
    return dir_p.name[0:2].isdigit()


def file_is_valid(file_p: pathlib.PosixPath) -> bool:
    """
    验证文件有效性：根据本项目命名规律，所有扩展名为.py或.md文件才是需要目录的
    """
    return file_p.suffix in (".py", ".md")


def dir_name_to_markdown(dir_name: str) -> str:
    """
    将目录名称写入 Markdown 文件时的规则
    """
    return f"## {dir_name[:2]} [{dir_name[3:]}](./{dir_name})\n\n"


def file_name_to_markdown(list_counter: int, file_name: pathlib.PosixPath) -> str:
    """
    将文件名称写入 Markdown 文件时的规则
    """
    return f"{list_counter}. [{file_name.name[3:]}](./{file_name.as_posix()})\n\n"


class TocMaker:
    """
    生成项目toc目录、统计代码行数并写入markdown文件
    """

    def __init__(self, working_dir: str = "./"):
        self.working_dir = pathlib.Path(working_dir)
        self.tree_dict = None
        self.cloc_result = None
        self._tree()

    def _tree(self) -> None:
        """
        列出工作目录下所有符合要求的目录和文件
        """
        tree_result = dict()
        file_name_list = list()

        # 列出工作目录下所有项目并验证有效性
        valid_dir_list = [
            x for x in self.working_dir.iterdir() if x.is_dir() and dir_is_valid(x)
        ]
        valid_dir_list.sort()  # 按照名称排序
        # TODO 处理多层嵌套目录

        # 验证有效性
        for valid_dir in valid_dir_list:
            iterated_files = list(valid_dir.iterdir())
            iterated_files.sort()
            for file in iterated_files:
                if file_is_valid(file):
                    file_name_list.append(file)
            tree_result.update({valid_dir: file_name_list})
            file_name_list = list()  # 重新置空

        self.tree_dict = tree_result

    def cloc(self, gitignore_file: str = ".gitignore") -> None:
        """
        使用cloc统计项目代码行数
        """
        ignored_dir = ""
        gitignore_file_p = pathlib.Path(gitignore_file)
        with gitignore_file_p.open("r", encoding="UTF-8") as f:
            for dir_name in f.readlines():
                dir_name = dir_name.replace("/", "")
                dir_name = dir_name.replace("\n", ",")
                ignored_dir += dir_name

        # 调用cloc，并排除gitignore中的目录，需要提前将cloc添加到系统环境变量
        cmd = f"cloc --exclude-dir {ignored_dir} {str(self.working_dir)}"

        with popen(cmd) as p:
            cmd_result = p.read()
            # 如果cmd执行正常退出则p.close()返回None，失败则返回状态码
            if p.close():
                print("cloc调用失败，请检查")
            else:
                # 根据cloc返回结果，连续两个换行符后面的内容是需要的信息
                self.cloc_result = cmd_result.split("\n\n", 1)[1]
                print(self.cloc_result)
        # TODO 将 self.cloc_result 写入 Markdown 文件

    def write_into_md(self, toc_file: str = "./toc.md") -> None:
        """
        把目录和文件名写入toc.md文件
        """
        write_lines = [Headline]
        file_counter = 0
        toc_file_p = pathlib.Path(toc_file)

        if self.tree_dict:
            for dir_name in self.tree_dict:
                list_counter = 0
                write_lines.append(dir_name_to_markdown(dir_name.name))
                for file_name in self.tree_dict[dir_name]:
                    write_lines.append(file_name_to_markdown(list_counter, file_name))
                    file_counter += 1
                    list_counter += 1
            counter_info = f"共{len(self.tree_dict)}个目录，{file_counter}个文件。\n"  # 计数器
            write_lines += counter_info
            with toc_file_p.open("wt", encoding="UTF-8") as f:
                f.writelines(write_lines)
            print(f"TOC生成成功，{counter_info}")
        else:
            print("tree列表为空，请检查")
            # TODO 调用失败时抛出一个异常


if __name__ == "__main__":
    toc_maker = TocMaker(Working_Directory)
    toc_maker.write_into_md(TOC_File_Output)
    toc_maker.cloc(Git_Ignore_File)  # 需要在系统中安装cloc并且把命令 'cloc' 添加到环境变量，否则注释掉本行