Get all files in folder in python (recursive optional)

get_all_files.py

from typing import Iterator
from pathlib import Path


def get_all_files(root: Path | str, recursive=False) -> Iterator[Path]:
    for item in Path(root).iterdir():
        if item.is_file():
            yield item
        elif recursive and item.is_dir():
            yield from get_all_files(item, recursive=True)

usage.py

def print_files_in_folder(root: str):
    for file in get_all_files(root):
        print(file)

def get_videos_recursive(root: str) -> Iterator[Path]:
    return (v for v in get_all_files(root, recursive=True) if v.suffix.lower() == '.mp4')