Python/날짜와 시간으로 사진 이름 변경 - 1. 파일 목록 가져오기

이 코드는 특정 디렉토리에서 특정 확장자를 가진 파일 목록을 가져오는 함수다.

매개변수

1. path (str): 파일 목록을 가져올 디렉토리의 경로.
2. extensions (list[str]): 선택하고자 하는 파일의 확장자 목록.

기능

1. 파일 목록 가져오기: os.listdir(path)를 사용하여 지정된 경로에서 파일 목록을 가져온다.
2. 확장자 선택 및 정렬: 가져온 파일 목록 중에서 주어진 확장자(extensions)에 해당하는 파일을 선택하고, 선택된 파일들을 정렬.
3. 확장자 미사용 목록 출력: 선택된 확장자 외의 다른 확장자들을 모아서 미사용 목록을 출력.

설명

• os.path.splitext(file)를 사용하여 파일의 확장자를 추출.
• extension.lower() in extensions를 통해 대소문자를 무시하고 주어진 확장자 목록에 있는 파일 선택.
• unused_extensions는 선택된 확장자 외의 확장자를 추적.
• 최종적으로 선택된 파일들은 알파벳순으로 정렬되어 반환.

이 함수는 특정 디렉토리에서 특정 확장자를 가진 파일들을 선택하고, 선택된 파일들을 정렬하여 반환하는데 사용. 함수의 출력에는 선택된 파일 목록과 사용되지 않은 확장자 목록이 포함.

728x90

def get_files(path: str, extensions: list[str]) -> list[str]:
    '''
    \n 특정 디렉토리에서 특정 확장자를 가진 파일들을 선택하고, 선택된 파일들을 정렬하여 반환하는데 사용.
    \n 함수의 출력에는 선택된 파일 목록과 사용되지 않은 확장자 목록이 포함.
    '''
    files = os.listdir(path)
    
    selections = []
    unused_extensions = []
    for file in files:
        _, extension = os.path.splitext(file)
        if extension.lower() in extensions:
            selections.append(file)
        else:
            if not extension in unused_extensions:
                unused_extensions.append(extension)
    selections.sort()
    
    file_count = len(selections)
    print(f'{file_count} files in {path}')
    print(f'Extensions unused: {unused_extensions}')
    
    return selections