获取目录下的文件名称

需求

期望整理出电脑中所有的pdf文档列表，把自己看过的书，罗列出来，整理一下自己的财产。

代码

import os
import os.path as op
import platform

fileNames=[]
print(platform.system())

if(platform.system()=='Windows'):
    sep='\r\n'
elif(platform.system()=='Linux'):
    sep='\n'
else:
    sep='\n'



def main():
    #os.path.pa
    print("start")
    get_filename_pdf()
    write_file()



def  get_filename_pdf():
    fileItems=os.listdir('.')
    print(os.path.abspath("."))
    for item in fileItems:
       #print(os.path.abspath(item))       
       if op.isfile(item) and item.endswith(".pdf"):
           # print()
            fileNames.insert(0,(item+sep).encode())
       elif op.isdir(item):
            os.chdir(item)
            get_filename_pdf()
            os.chdir("..")


def write_file(): 
    with  open("file.txt",'wb') as file:
        print(fileNames)
        file.writelines(fileNames)  


if __name__=="__main__":
    main()

拓展

可以通过配置支持拓展。也可以做其他的功能，文件整理。

总结

python的包用的不熟