grokking-algorithms/chapter7/dfs_dirtrav.py
2024-01-12 12:32:53 +00:00

26 lines
573 B
Python

import logging
from collections import deque
from pathlib import Path
logging.basicConfig(level=logging.DEBUG)
logger = logging.getLogger(__name__)
EXT = ".rb"
scripts_path = Path.home() / "scripts"
def files_with_extension(directory):
for item in directory.glob("*"):
# if item is a file and has extension EXT then print
if item.is_file():
if item.suffix == EXT:
print(item)
# otherwise pass directory to recursive call
else:
files_with_extension(item)
files_with_extension(scripts_path)