[점프 투 파이썬] - 6장 하위 디렉터리 검색하기

특정 디렉터리부터 그 하우 ㅣ모든 파일 중 파이썬(*.py)만 출력해 주는 프로그램

 

 

1. 코드 작성

def search(dirname):
    print(dirname)

search("C:/")

 

 

2. 디텍터리에 있는 파일 검색할 수 있도록 소스 변경

import os

def search(dirname):
    filenames = os.listdir(dirname)
    for filename in filenames:
        full_filename = os.path.join(dirname, filename)
    print(full_filename)

search("C:/")

 

 

3. 확장자가 .py인 파일만 출력하도록 코드 변경

import os

def search(dirname):
    filenames = os.listdir(dirname)
    for filename in filenames:
        full_filename = os.path.join(dirname, filename)
        ext = os.path.splitext(full_filename)[-1]
        if ext == ".py":
            print(full_filename)

search("C:/")

 

 

4. C:/디렉터리 바로 밑에 있는 파일뿐만 아니라 그 하위 디렉터리까지 검색

import os

def search(dirname):
    try:
        filenames = os.listdir(dirname)
        for filename in filenames:
            full_filename = os.path.join(dirname, filename)
            if os.path.isdir(full_filename):
                search(full_filename)
            else:
                ext = os.path.splitext(full_filename)[-1]
                if ext == ".py":
                    print(full_filename)
    except PermissionError:
        pass

search("c:/")

 

 

 

os.walk 사용

import os
for (path, dir, files) in os.walk("C:/"):
    for filename in files:
        ext = os.path.splitext(filename)[-1]
        if ext == '.py':
            print("%s/%s" % (path, filename))