Python으로 특정 경로에서 파일 찾기: 경로 리스트 출력하기

 

안녕하세요, Python 프로그래밍에 관심이 많은 여러분! 오늘은 Python을 사용하여 특정 경로에서 원하는 파일을 찾아 그 경로를 리스트로 출력하는 방법에 대해 알아보겠습니다. 파일 시스템을 탐색하고, 원하는 파일을 빠르게 찾는 것은 프로그래밍에서 자주 필요한 작업 중 하나입니다. 이 글에서는 Python의 os 모듈을 활용하여 간단하고 효율적으로 파일을 찾는 방법을 소개하겠습니다.

 

파일을 찾는 작업은 데이터 분석, 자동화 스크립트, 시스템 관리 등 다양한 분야에서 필요합니다. Python의 os 모듈은 파일 시스템과 상호작용할 수 있는 강력한 도구를 제공하며, 특히 os.walk 함수를 사용하면 디렉토리와 하위 디렉토리를 쉽게 탐색할 수 있습니다. 오늘은 이 함수를 활용하여 특정 파일명을 가진 파일들의 경로를 리스트로 수집하고 출력하는 방법을 배워보겠습니다.

 

728x90

1. 필요한 모듈 임포트

먼저, 파일 시스템 작업을 위해 Python의 os 모듈을 임포트합니다.

 

import os

2. 파일 검색 함수 정의

다음으로, 특정 디렉토리에서 파일을 찾는 함수를 정의합니다. 이 함수는 사용자로부터 디렉토리 경로와 파일명을 입력받아, 해당 파일이 존재하는 모든 경로를 리스트로 반환합니다.

 

def find_file(directory, filename):
    # 디렉토리가 존재하는지 확인
    if not os.path.isdir(directory):
        return f"디렉토리가 존재하지 않습니다: {directory}"
    
    paths = []
    # 디렉토리와 하위 디렉토리를 순회하며 파일 검색
    for root, dirs, files in os.walk(directory):
        if filename in files:
            paths.append(os.path.join(root, filename))
    
    if paths:
        return paths
    else:
        return f"파일을 찾을 수 없습니다: {filename}"

• 입력값: directory (검색할 디렉토리 경로), filename (찾고자 하는 파일명).
• 동작:
  • os.path.isdir(directory)로 입력된 경로가 유효한 디렉토리인지 확인합니다. 없으면 오류 메시지를 반환합니다.
  • os.walk(directory)를 사용해 지정된 디렉토리와 그 하위 디렉토리를 탐색합니다.
  • 탐색 중 filename과 일치하는 파일을 발견하면, os.path.join(root, filename)으로 전체 경로를 생성해 paths 리스트에 추가합니다.
• 반환값:
  • 파일이 발견되면: 발견된 경로들의 리스트 (paths).
  • 파일이 없거나 디렉토리가 유효하지 않으면: 오류 메시지 문자열.

3. 사용자 입력 받기

이제 사용자로부터 디렉토리 경로와 파일명을 입력받습니다. 이를 위해 input() 함수를 사용합니다.

 

directory = input("디렉토리 경로를 입력하세요: ")
filename = input("파일명을 입력하세요: ")

4. 파일 검색 및 결과 출력

마지막으로, 정의한 find_file 함수를 호출하여 파일을 검색하고, 결과를 출력합니다.

 

result = find_file(directory, filename)

if isinstance(result, list):
    print("파일이 발견된 경로:")
    for path in result:
        print(path)
else:
    print(result)

• 결과가 리스트일 경우: "파일이 발견된 경로:"라는 제목 아래 경로를 하나씩 출력합니다.
• 결과가 문자열일 경우: 오류 메시지(예: "파일을 찾을 수 없습니다: report.txt")를 출력합니다.

실행 예시

 

경우 1: 파일이 존재할 때

 

디렉토리 경로를 입력하세요: /home/user
파일명을 입력하세요: test.txt

파일이 발견된 경로:
/home/user/test.txt
/home/user/subfolder/test.txt

경우 2: 파일이 없을 때

 

 

디렉토리 경로를 입력하세요: /home/user
파일명을 입력하세요: missing.txt

파일을 찾을 수 없습니다: missing.txt

 

경우 3: 디렉토리가 잘못된 경우

 

디렉토리 경로를 입력하세요: /wrong/path
파일명을 입력하세요: test.txt

디렉토리가 존재하지 않습니다: /wrong/path

결론

Python의 os 모듈을 사용하면 파일 시스템을 쉽게 탐색하고, 원하는 파일을 빠르게 찾을 수 있습니다. 이 글에서 소개한 방법을 통해, 여러분도 자신의 프로젝트나 작업에서 파일을 효율적으로 관리하고 검색할 수 있을 것입니다. 파일 검색은 프로그래밍의 기본이지만, 그 중요성은 매우 큽니다. 앞으로도 Python의 다양한 기능을 활용하여 더 많은 작업을 자동화하고 효율성을 높여보세요!