Возврат только определенное имя файла без расширения

Question

Так из предыдущего поста, который я спросил я нашел следующий код для работы:

b = os.walk('C:\\Users\\USER\\Desktop\\FOLDER') 
f = [] 
for(dirpath, dirnames, filenames) in b: 
    f.extend(os.path.splitext(name)[0] for name in filenames) 
    break 

Однако, сейчас я хотел бы этот список для того чтобы только определенный тип файла, например как только файлы, которые являются .JPEG или .GIF (все еще без расширения).

я уже пытался использовать fnmatch, но он не работает ..

После комментариев я изменил код:

b = os.walk('C:\\Users\\USER\\Desktop\\FOLDER') 
os.listdir('C:\\Users\\USER\\Desktop\\FOLDER') 
f = [] 
for(dirpath, dirnames, filenames) in b: 
    f.extend(os.path.splitext(name)[0] for name in filenames if name.lower().endswith((".jpeg", ".gif"))) 

def save_to_file(text): 
    for name in f: 
     with open('C:\\Users\\USER\\Desktop\\FOLDER\\test.txt', mode='wt', encoding='utf-8') as myfile: 
      myfile.write('\n'.join(text)) 
      myfile.write('\n') 

save_to_file(f) 

Answer 1

Вы можете использовать str.endswith прохождения кортежа расширений:

f.extend((os.path.splitext(name)[0] for name in filenames if name.lower().endswith((".jpeg", ".gif"))) 

Я не знаю, почему у вас есть перерыв в вашем цикле for, так как это будет означать, что вы переходите через один каталог, который вы можете делать с помощью os.listdir

Answer 2

Если я вас правильно понял, вы хотите, чтобы проверить каждый файл name и возвращает только определенный тип файла без расширения? Поскольку вы работаете на Окна, я думаю, вы могли бы сделать:

for filename in files: 
    try: 
     ext = filename.split('.') 
     if ext[1].lower() == 'gif' or ext[1].lower() == 'jpeg': 
      print ext[0] 
    except IndexError: 
     continue 

Answer 3

def find_files_by_filter(folder_path, filter): 

    """ 
    :rtype : list 
    """ 
    if not isdir(folder_path): 
     log('Folder path: '+ str(folder_path) + ' not found') 
     return [] 

    re_filter = re.compile(filter) 
    all_files = listdir(folder_path) 
    result = [file for file in all_files if re_filter.search(file)] 
    return result 

следующая:

file_name = [] 
for res in result: 
    file_name.append(res.split('.')[0] 
print(file_name)