Find all files in a directory

find_files.py

from pathlib import Path


def check_types(suffix, path):
    assert isinstance(suffix, str), 'Suffix must be a string'
    assert isinstance(path, str), 'Path must be a string'


def find_files(suffix: str, path: str, dirs=False) -> list:
    check_types(suffix, path)

    path_obj = Path(path)
    path_iter = path_obj.rglob(f'*{suffix}')

    if dirs:
        file_list = [str(f) for f in path_iter]
    else:
        file_list = [str(f) for f in path_iter if f.is_file()]

    return file_list

path = './pictures'
suffix = '.png'
print(find_files(suffix, path))