Не получается сделать обход по всем папкам в поисках файла

import os
 
 
def find(fileName):
 
    if fileName not in os.listdir():
        for i in os.listdir():
            if os.path.isdir(i):
                os.chdir(f'{os.getcwd()}\\{i}')
                find(fileName)
    else:
        print(f'{fileName} находится в папке {os.getcwd()}')
 
 
 
find('string.py')

import os
from typing import Optional
 
def find_file(path: str, file: str) -> Optional[str]:
    for root, dirs, files in os.walk(path):
        if file in files:
            return f"{root}\\{file}"
    return None
 
 
print(find_file(".", "settings.txt"))  # .\settings.txt
print(find_file(".", "key.ico"))  # .\media\icon\key.ico

import os
 
 
def find_file(path, file):
    for obj in os.listdir(path):
        full_path = f"{path}\\{obj}"
        if os.path.isdir(full_path):
            if find_file(full_path, file):
                break
        elif obj == file:
            break
    else:
        return None
    return f"{full_path}\\{file}"
 
print(find_file("D:\\", "example.txt"))

os.path.join(path, obj)

import os
 
''' Найти в данной в файловой структуре все директории,
    в которых есть хотя бы один файл с расширением
    ".py".
 
    Вывести упорядоченный список таких директорий. '''
 
root = os.getcwd()      # текущая директория
answer = []             # результаты поиска
 
''' ищем все директории, которые содержат хотя бы один
    файл с расширением .py '''
for current, directories, files in os.walk('.'):
    for f in files:
        if f.endswith('.py'):
            d = current.replace('.', root)
            if d not in answer:
                answer.append(d)
 
''' сохраняем результаты поиска в файл '''
with open('output.txt', 'w') as fp:
    fp.write('\n'.join(sorted(answer)))