mirror of
https://github.com/halfrost/LeetCode-Go.git
synced 2025-07-06 17:44:10 +08:00
56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
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
|
||
|