본문 바로가기

My Work/Work Automation(업무 자동화)

이미지 파일 크기 확인 코드(파이썬)

import os
from PIL import Image

# 사용자로부터 폴더 위치를 입력 받는 함수
def Fn_GetDirectoryPath():
    str_directoryPath = input("현재 폴더 위치를 입력하세요: ")
    return str_directoryPath

# 이미지 크기를 확인하는 함수
def Fn_CheckImageSize(str_imagePath):
    obj_image = Image.open(str_imagePath)
    int_width, int_height = obj_image.size
    return int_width == 1200 and int_height == 1200

# 모든 하위 폴더를 순회하면서 이미지 크기 확인
def Fn_TraverseAndCheck(str_directoryPath):
    # 이미지 파일 확장자 목록
    lst_imageExtensions = ['.jpg', '.jpeg', '.png', '.gif', '.bmp']
    
    for str_root, lst_dirs, lst_files in os.walk(str_directoryPath):
        bool_noIssue = True
        for str_fileName in lst_files:
            if any(str_fileName.endswith(ext) for ext in lst_imageExtensions):
                str_imagePath = os.path.join(str_root, str_fileName)
                if not Fn_CheckImageSize(str_imagePath):
                    print(f"이미지 크기가 1200x1200이 아닌 파일 위치: {str_imagePath}")
                    bool_noIssue = False
        if bool_noIssue:
            print(f"{str_root} 폴더는 이상이 없습니다.")

# 현재 폴더 위치를 입력 받음
str_directoryPath = Fn_GetDirectoryPath()

# 모든 하위 폴더를 순회하면서 이미지 크기 확인
Fn_TraverseAndCheck(str_directoryPath)

'My Work > Work Automation(업무 자동화)' 카테고리의 다른 글

Chat GPTs 사용후기  (0) 2023.11.21
JSON을 CSV로 바꾸는 코드  (0) 2023.11.07