pathlibを使って特定ディレクトリ以下の全ファイルを取得する

はじめに

テキストファイルを分かち書きするのにファイルを全て結合させて処理していたが、サイズが大きすぎてメモリが悲鳴を上げたのでファイルを１つずつ開いて処理することにした。
そのため、まずは全ファイルを取得して１つずつ開くコードを書いた。

コード

Path().rglob('*') を使う
対象のファイルの拡張子が決まっているなら Path().rglob("*.txt") のように書けばよいのだが、私が扱いたいファイルは拡張子がないので is_file() を使ってファイルのみを返すようなメソッドを書く。

from pathlib import Path


def main():
    for path in get_file('./text'):
        with path.open("r", encoding='utf-8') as file:
            file.read()

def get_file(dirctory):
    for f in Path(dirctory).rglob('*'):
        if f.is_file():
            yield f

コード(修正前)

良くない書き方

from pathlib import Path


def main():
    for path in get_file('./text'):
        with path.open("r", encoding='utf-8') as file:
            file.read()

def get_file(dirctory):
    d = Path(dirctory).iterdir()
    for f in d:
        if f.is_file():
            yield f
        elif f.is_dir():
            yield from get_file(f)

if __name__ == '__main__':
    main()