import os
path = 'D:/pyframet'
def cleartext():
file_list = os.listdir(os.getcwd())
print(file_list)
for i in file_list:
kind1 = i.split('.')[-1]
if kind1 == 'txt':
os.remove(i)
file_list = os.listdir(os.getcwd())
print(file_list)
def classification():
for dirPath, dirName, fileName in os.walk(path):
for i in fileName:
info = dirPath + I
kind2 = info.split('.')[-1]
if kind2 in ['pcb', 'dra']:
with open(r'pcb.txt', 'a') as f:
f.write(info + '\n')
if kind2 in ['pdf']:
with open(r'pdf.txt', 'a') as f:
f.write(info + '\n')
if kind2 in ['brd']:
with open(r'brd.txt', 'a') as f:
f.write(info + '\n')
if kind2 in ['xlsx', 'xls']:
with open(r'excel.txt', 'a') as f:
f.write(info)
if kind2 in ['py']:
with open(r'py.txt', 'a') as f:
f.write(info + '\n')
cleartext()
classification()