自動檢測文件擴展名

Question

正如你可以看到這是非常非常笨重。我將如何去自動檢測文件擴展名？

txt_files = glob('*txt') 
png_files = glob('*.png') 
jpg_files = glob('*.jpg') 
zip_files = glob('*.zip') 
gif_files = glob('*.gif') 
iso_files = glob('*iso') 
epub_files = glob('*epub') 
ico_files = glob('*.ico') 

#Unimportant bit to question 

for n in txt_files: 
    move_files(n, 'txt') 
for n in png_files: 
    move_files(n, 'png') 
for n in jpg_files: 
    move_files(n, 'jpg') 
for n in zip_files: 
    move_files(n, 'zip') 
for n in gif_files: 
    move_files(n, 'gif') 
for n in iso_files: 
    move_files(n, 'iso') 
for n in epub_files: 
    move_files(n, 'epub') 
for n in ico_files: 
    move_files(n, 'ico') 

Answer 1

爲「自動檢測新的擴展名」，只是經過所有文件，並看看他們的擴展：

for f in os.listdir(): 
    name, ext = os.path.splitext(f) 
    ext = ext.lstrip('.') 
    if ext: 
     os.mkdirs(ext) 
     move_file(filename, ext) 

請注意，如果你的一些「擴展」的實際上不是這樣的擴展將無法正常工作。我注意到您現有的代碼搜索諸如*txt和*iso之類的東西，這當然會匹配諸如this_is_not_txt和spam.aniso等之類的東西。如果這是一個必需的功能，而不是一個錯誤，那麼你必須爲你的「擴展」所指的內容制定一些規則，然後才能在代碼中實現它...

另請注意，沒有擴展名的文件因爲那if ext:），並且有一個空的擴展名（只是.）的文件也將是（因爲if檢查發生在lstrip之後）。如果需要，你當然可以改變其中的任何一種（儘管你需要提出一個子目錄名稱，它不可能與任何可能的實際擴展模糊不清......）。

最後，這不會處理「雙擴展」 - 例如，.tar.gz文件將與.gz進入同一目錄。同樣，如果需要，這很容易改變。

Answer 2

for ext in "txt png jpg zip gif iso epub ico".split(): 
    for file in glob("*.%s"%ext): #if there are no files of this type it just skips it 
     move_files(file,ext) 

可能？它不是很清楚你的問題是什麼

從某種意義上說，這是「自動檢測」以後如果有適合的模式中不存在的文件將跳過move_files一步