Как исключить из вывода каталоги/файлы с именами из множества?

Question

[nihi1ist]

Пытаюсь добавить исключения в код, который выдает список каталогов и файлов через os.walk:

Код

import os

full_path = '/home/nihi1ist/some_folder'
ignored_names = {'.git', '.gitignore', '.idea', 'venv'}
folder_files_size, subfolders_count, files_count = 0, 0, 0

for root, dirs, files in os.walk(full_path, topdown=True, followlinks=False):
    folder_files_size += os.path.getsize(root)
    root_name = os.path.basename(root)
    if root_name not in ignored_names:
        for element in dirs:
            if element not in ignored_names:
                path = os.path.join(root, element)
                print('Это каталог:', path)
                subfolders_count += 1
                folder_files_size += os.path.getsize(path)
        for element in files:
            if element not in ignored_names:
                path = os.path.join(root, element)
                print('\tЭто файл:', path)
                files_count += 1
                folder_files_size += os.path.getsize(path)

Но не выходит. Подскажите, есть ли возможность таки исключить из выдачи ненужные элементы?

Comments

[Алан Гибизов]

«Исключения» - это термин для немножко другого. Тут лучше термин «фильтр».
Что касается ваших попыток фильтровать - есть у меня предчувствие, что element может иметь тип не str и потому может не находиться в ignored.

[Vindicar]

Проверь, что у тебя попадает в root_name.

[mkone112]

Алан Гибизов, исключения это не термин - это такое слово, которое в том числе может обозначать конкретный термин.

[Алан Гибизов]

mkone112, конечно, вы правы. Но поскольку термин есть, то лучше сразу определиться и не использовать термин в альтернативных вариантах, тем более что там есть свой термин. Ну, чисто для порядку.

[nihi1ist]

Алан Гибизов, да, пропустил этот момент. Но, даже после исправления, вывод не фильтруется:

Код

import os

full_path = '/home/nihi1ist/some_folder'
ignored_names = {'.git', '.gitignore', '.idea', 'venv'}
folder_files_size, subfolders_count, files_count = 0, 0, 0

for root, dirs, files in os.walk(full_path, topdown=True, followlinks=False):
    folder_files_size += os.path.getsize(root)
    root_name = os.path.basename(root)
##    print(root_name)
    if root_name not in ignored_names:
        for element in dirs:
            element_name = os.path.basename(element)
            print(element_name)
            if element_name not in ignored_names:
                path = os.path.join(root, element)
##                print('Это каталог:', path)
                subfolders_count += 1
                folder_files_size += os.path.getsize(path)
        for element in files:
            element_name = os.path.basename(element)
            print(element_name)
            if element_name not in ignored_names:
                path = os.path.join(root, element)
##                print('\tЭто файл:', path)
                files_count += 1
                folder_files_size += os.path.getsize(path)

Vindicar, проверял, там имена каталогов.

[3notH]

nihi1ist, root_name содержит в себе рабочий каталог (в данном случае some_folder). Если создавать исключения, то с указанием всех подкаталогов.
На счет файлов - в массиве указано расширение, а следовательно надо сравнивать только их, например с помощью os.path.splitext(element_name)[1]