Update get_file_list_and_write_in_markdown.py

This commit is contained in:
guanjihuan 2022-09-08 09:15:05 +08:00
parent 736413a518
commit ccc8250e36

View File

@ -8,16 +8,16 @@ def main():
import os import os
directory = 'E:/literature' directory = 'E:/literature'
write_file_list_in_markdown(directory) write_file_list_in_markdown(directory)
# write_file_list_in_markdown(directory, banned_type=['.md'], divided_line=1, show_second_number=1, show_third_number=1) # write_file_list_in_markdown(directory, banned_type=['.md'], hide_file_type=1, divided_line=1, show_second_number=1, show_third_number=1)
# write_file_list_in_markdown(directory, filename='a', reverse_positive_or_negative=1, starting_from_h1=None, banned_type=[], divided_line=None, show_second_number=None, show_third_number=None) # write_file_list_in_markdown(directory, filename='a', reverse_positive_or_negative=1, starting_from_h1=None, banned_type=[], hide_file_type=None, divided_line=None, show_second_number=None, show_third_number=None)
# import guan # import guan
# guan.write_file_list_in_markdown(directory) # guan.write_file_list_in_markdown(directory)
# guan.write_file_list_in_markdown(directory, banned_type=['.md'], divided_line=1, show_second_number=1, show_third_number=1) # guan.write_file_list_in_markdown(directory, banned_type=['.md'], hide_file_type=1, divided_line=1, show_second_number=1, show_third_number=1)
# guan.write_file_list_in_markdown(directory, filename='a', reverse_positive_or_negative=1, starting_from_h1=None, banned_type=[], divided_line=None, show_second_number=None, show_third_number=None) # guan.write_file_list_in_markdown(directory, filename='a', reverse_positive_or_negative=1, starting_from_h1=None, banned_type=[], hide_file_type=None, divided_line=None, show_second_number=None, show_third_number=None)
def write_file_list_in_markdown(directory, filename='a', reverse_positive_or_negative=1, starting_from_h1=None, banned_type=[], divided_line=None, show_second_number=None, show_third_number=None): def write_file_list_in_markdown(directory, filename='a', reverse_positive_or_negative=1, starting_from_h1=None, banned_type=[], hide_file_type=None, divided_line=None, show_second_number=None, show_third_number=None):
import os import os
f = open(filename+'.md', 'w', encoding="utf-8") f = open(filename+'.md', 'w', encoding="utf-8")
filenames1 = os.listdir(directory) filenames1 = os.listdir(directory)
@ -26,22 +26,28 @@ def write_file_list_in_markdown(directory, filename='a', reverse_positive_or_neg
filename1_with_path = os.path.join(directory,filename1) filename1_with_path = os.path.join(directory,filename1)
if os.path.isfile(filename1_with_path): # 文件 if os.path.isfile(filename1_with_path): # 文件
if os.path.splitext(filename1)[1] not in banned_type: if os.path.splitext(filename1)[1] not in banned_type:
f.write('+ '+str(os.path.splitext(filename1)[0])+'\n') if hide_file_type == None:
else: # 文件夹 f.write('+ '+str(filename1)+'\n')
else:
f.write('+ '+str(os.path.splitext(filename1)[0])+'\n')
else: # 文件夹
u0 += 1 u0 += 1
if divided_line != None and u0 != 1: if divided_line != None and u0 != 1:
f.write('\n--------\n\n') f.write('\n--------\n\n')
if starting_from_h1 == None: if starting_from_h1 == None:
f.write('#') f.write('#')
f.write('# '+str(filename1)+'\n') f.write('# '+str(filename1)+'\n')
filenames2 = os.listdir(filename1_with_path) filenames2 = os.listdir(filename1_with_path)
i0 = 0 i0 = 0
for filename2 in filenames2[::reverse_positive_or_negative]: for filename2 in filenames2[::reverse_positive_or_negative]:
filename2_with_path = os.path.join(directory, filename1, filename2) filename2_with_path = os.path.join(directory, filename1, filename2)
if os.path.isfile(filename2_with_path): # 文件 if os.path.isfile(filename2_with_path): # 文件
f.write('+ '+str(os.path.splitext(filename2)[0])+'\n') if hide_file_type == None:
else: # 文件夹 f.write('+ '+str(filename2)+'\n')
else:
f.write('+ '+str(os.path.splitext(filename2)[0])+'\n')
else: # 文件夹
i0 += 1 i0 += 1
if starting_from_h1 == None: if starting_from_h1 == None:
f.write('#') f.write('#')
@ -54,8 +60,11 @@ def write_file_list_in_markdown(directory, filename='a', reverse_positive_or_neg
filenames3 = os.listdir(filename2_with_path) filenames3 = os.listdir(filename2_with_path)
for filename3 in filenames3[::reverse_positive_or_negative]: for filename3 in filenames3[::reverse_positive_or_negative]:
filename3_with_path = os.path.join(directory, filename1, filename2, filename3) filename3_with_path = os.path.join(directory, filename1, filename2, filename3)
if os.path.isfile(filename3_with_path): # 文件 if os.path.isfile(filename3_with_path): # 文件
f.write('+ '+str(os.path.splitext(filename3)[0])+'\n') if hide_file_type == None:
f.write('+ '+str(filename3)+'\n')
else:
f.write('+ '+str(os.path.splitext(filename3)[0])+'\n')
else: # 文件夹 else: # 文件夹
j0 += 1 j0 += 1
if starting_from_h1 == None: if starting_from_h1 == None:
@ -68,9 +77,12 @@ def write_file_list_in_markdown(directory, filename='a', reverse_positive_or_neg
filenames4 = os.listdir(filename3_with_path) filenames4 = os.listdir(filename3_with_path)
for filename4 in filenames4[::reverse_positive_or_negative]: for filename4 in filenames4[::reverse_positive_or_negative]:
filename4_with_path = os.path.join(directory, filename1, filename2, filename3, filename4) filename4_with_path = os.path.join(directory, filename1, filename2, filename3, filename4)
if os.path.isfile(filename4_with_path): # 文件 if os.path.isfile(filename4_with_path): # 文件
f.write('+ '+str(os.path.splitext(filename4)[0])+'\n') if hide_file_type == None:
else: # 文件夹 f.write('+ '+str(filename4)+'\n')
else:
f.write('+ '+str(os.path.splitext(filename4)[0])+'\n')
else: # 文件夹
if starting_from_h1 == None: if starting_from_h1 == None:
f.write('#') f.write('#')
f.write('#### '+str(filename4)+'\n') f.write('#### '+str(filename4)+'\n')
@ -78,8 +90,11 @@ def write_file_list_in_markdown(directory, filename='a', reverse_positive_or_neg
filenames5 = os.listdir(filename4_with_path) filenames5 = os.listdir(filename4_with_path)
for filename5 in filenames5[::reverse_positive_or_negative]: for filename5 in filenames5[::reverse_positive_or_negative]:
filename5_with_path = os.path.join(directory, filename1, filename2, filename3, filename4, filename5) filename5_with_path = os.path.join(directory, filename1, filename2, filename3, filename4, filename5)
if os.path.isfile(filename5_with_path): # 文件 if os.path.isfile(filename5_with_path): # 文件
f.write('+ '+str(os.path.splitext(filename5)[0])+'\n') if hide_file_type == None:
f.write('+ '+str(filename5)+'\n')
else:
f.write('+ '+str(os.path.splitext(filename5)[0])+'\n')
else: # 文件夹 else: # 文件夹
if starting_from_h1 == None: if starting_from_h1 == None:
f.write('#') f.write('#')