import os

from os.path import join

from shutil import move

toc_string = """\n

---

bookToc: false

---

\n"""

pre_string = """

## 代码

```go

"""

end_string = "\n```"

codeContent = ""

current_working_dir = os.getcwd()

# print(f"current_working_dir: {current_working_dir}")

# dir_names = os.listdir(current_working_dir)

# 重命名目录。 os.rename(dir_name, new_dir_name)

# 遍历目录

# dir_path是当前遍历到的目录。dir_names是dir_path下的文件夹列表。file_names是是dir_path下的文件列表

# 如果想实现目录白名单，将白名单目录从dir_names中去除即可

for (dir_path, dir_names, file_names) in os.walk(current_working_dir):

# print(f"当前遍历到的目录: {dir_path}")

os.chdir(dir_path)

files = dir_path.split("/")

new_file_name = files[len(files) - 1] + '.md'

print(f"当前所在的文件夹: {os.getcwd()}")

for file_name in file_names:

if(file_name.endswith('.go') and not file_name.endswith('_test.go')):

print(f"当前所在文件: {file_name}")

with open(file_name, "r") as myfile:

codeContent = myfile.read()

break

# print(codeContent)

for file_name in file_names:

if(file_name.endswith('.md')):

print(f"当前所在文件: {file_name}")

with open(file_name, "a") as myfile:

myfile.write(pre_string)

myfile.write(codeContent)

myfile.write(end_string)

os.rename(file_name, new_file_name)

move(join(dir_path, new_file_name), current_working_dir)

break