65 lines
1.9 KiB
Python
65 lines
1.9 KiB
Python
import os
|
||
|
||
def file_name(file_dir):
|
||
temp = []
|
||
for root, dirs, files in os.walk(file_dir):
|
||
print(dirs) #当前路径下所有子目录
|
||
temp = dirs #存储需要的子目录
|
||
break
|
||
|
||
# 删除不需要的子目录
|
||
temp.remove('_templates')
|
||
temp.remove('_build')
|
||
temp.remove('_static')
|
||
chinese_name = ['模型训练', '介绍', '图像识别流程', '其他', 'FAQ系列', '数据准备', '安装', '模型库', '高级教程', '算法介绍', '推理部署', '快速开始']
|
||
# 写根目录下的rst文件
|
||
with open('./index.rst', 'w') as f:
|
||
f.write('欢迎使用PaddleClas图像分类库！\n')
|
||
f.write('================================\n\n')
|
||
f.write('.. toctree::\n')
|
||
f.write(' :maxdepth: 1\n\n')
|
||
for dir in temp:
|
||
f.write(' ' + dir + '/index\n')
|
||
f.close()
|
||
|
||
# 写各个子目录下的rst文件
|
||
for dir in temp:
|
||
for root, dirs, files in os.walk(dir):
|
||
print(root) #当前目录路径
|
||
if 'index.rst' in files:
|
||
files.remove('index.rst')
|
||
print(files) #当前路径下所有非目录子文件
|
||
curDir = os.path.join(file_dir, dir)
|
||
filename = curDir + '/index.rst'
|
||
idx = temp.index(dir)
|
||
ch_name = chinese_name[idx]
|
||
with open(filename, 'w') as f:
|
||
f.write(ch_name+'\n')
|
||
f.write('================================\n\n')
|
||
f.write('.. toctree::\n')
|
||
f.write(' :maxdepth: 2\n\n')
|
||
|
||
for f1 in files:
|
||
f.write(' ' + f1 + '\n')
|
||
|
||
f.close()
|
||
|
||
|
||
def readfile(filename):
|
||
file = open(filename)
|
||
i = 0
|
||
while 1:
|
||
line = file.readline()
|
||
print(i)
|
||
print(line)
|
||
i += 1
|
||
if not line:
|
||
break
|
||
pass # do something
|
||
file.close()
|
||
|
||
|
||
file_name('./')
|
||
#filename = './index.rst'
|
||
#readfile(filename)
|