Python, найти указанный файл в папке

Question

Что не так в этом коде?Я хочу найти указанный файл в папке.Спасибо за помощь.

import os, fnmatch

def find(root, mask): //Error
    files_list = os.listdir(os.path.abspath(root))
    for filename in fnmatch.filter(files_list, mask):
        yield filename

def test():
    res = find ('D:\\Sample\\', 'hallo.txt') 

test()

ошибка:

Error: Traceback (most recent call last): def find(root, mask): NoneType

Answer 1

Ваша функция - генератор , он будет выдавать имена файлов один за другим. Вы можете позвонить list на res, если хотите получить список всех совпадений:

list_of_matches = list(res)

Также ваш аргумент по умолчанию root=os.dir не имеет смысла. Может быть, использовать '.'?

Answer 2

Отсутствует отступ для тестовой функции

def test ():
    res = find ('D:\\Sample\\', 'hallo.txt')