获取目录下的文件名称
需求
期望整理出电脑中所有的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的包用的不熟