폴더 안에 있는 파일 모두 읽기

폴더 안에서 특정 확장자를 가진 파일들을 모두 찾고 읽는 방법을 정리해 보았다.

import os
import sys

root_dir = sys.argv[1]

print('walk_dir = ' + root_dir)
print('walk_dir (absolute) = ' + os.path.abspath(root_dir))
for folder, subs, files in os.walk(root_dir):
    with open(os.path.join(folder, 'python-outfile.txt'), 'w') as dest:
            for filename in files:
                for extension in extensions:
                    if filename.endswith(extension):
                        with open(os.path.join(folder, filename), 'r') as src:
                            ...파일 관련 작업...

실행할때는 입력 파라미터로 폴더 이름을 전달한다.