14 lines
454 B
Python
Raw Permalink Normal View History

import os
from tempfile import TemporaryDirectory
from typing import Generator, TextIO
def get_python_files(dir: TemporaryDirectory) -> Generator[TextIO, None, None]:
for dirpath, _, filenames in os.walk(dir.name):
for filename in filenames:
if filename.endswith(".py"):
2021-05-09 03:05:11 +02:00
try:
yield open(os.sep.join([dirpath, filename]), "r")
except FileNotFoundError:
...