Python으로 폴더를 쉽게 만들고 삭제하는 방법 | os 모듈과 pathlib 모듈 활용

1. Python으로 폴더를 생성하는 기본 방법

Python으로 폴더를 생성하는 방법：os 모듈

import os

# 생성할 폴더의 경로 지정
path = 'example_folder'

# 폴더 생성
os.mkdir(path)

다중 계층 폴더를 생성하는：os.makedirs()

import os

# 다중 계층 폴더 지정
path = 'example_folder/subfolder'

# 상위 디렉터리가 없어도 폴더 생성
os.makedirs(path)

에러 처리：exist_ok=True 사용

import os

path = 'example_folder/subfolder'

# 이미 폴더가 존재하는 경우에도 에러 무시
os.makedirs(path, exist_ok=True)

2. Pathlib 모듈을 사용한 폴더 생성

Pathlib 모듈 사용법

from pathlib import Path

# 만들고자 하는 폴더의 경로 지정
path = Path('example_folder/subfolder')

# 폴더 생성
path.mkdir(parents=True, exist_ok=True)

Pathlib 모듈의 장점

3. 폴더 생성에 대한 오류 처리

권한 오류 및 무효한 경로 처리

import os

path = 'example_folder/subfolder'

try:
    os.makedirs(path, exist_ok=True)
    print(f'폴더 "{path}" 가 생성되었습니다.')
except PermissionError:
    print('폴더를 생성할 권한이 없습니다.')
except FileNotFoundError:
    print('지정된 경로가 유효하지 않습니다.')
except Exception as e:
    print(f'예기치 않은 오류가 발생했습니다: {e}')

보다 고급 오류 처리

4. 폴더 삭제 방법

os.rmdir()를 사용한 폴더 삭제

import os

# 삭제할 폴더의 경로를 지정
path = 'example_folder/subfolder'

# 폴더를 삭제
os.rmdir(path)

shutil.rmtree()를 사용한 폴더의 재귀적 삭제

import shutil

# 삭제할 폴더의 경로를 지정
path = 'example_folder/subfolder'

# 폴더와 그 안의 내용을 삭제
shutil.rmtree(path)

에러 핸들링

import shutil

path = 'example_folder/subfolder'

try:
    shutil.rmtree(path)
    print(f'폴더 "{path}" 가 삭제되었습니다.')
except PermissionError:
    print('폴더를 삭제할 권한이 없습니다.')
except FileNotFoundError:
    print('지정된 폴더를 찾을 수 없습니다.')
except Exception as e:
    print(f'예기치 않은 오류가 발생했습니다: {e}')

5. 실제 적용 사례

연월별 폴더 생성

import os
from datetime import datetime

# 현재 연도와 월을 가져오기
current_year = datetime.now().year
current_month = datetime.now().month

# 폴더 경로 설정
folder_path = f'data/{current_year}/{current_month}'

# 폴더 생성
os.makedirs(folder_path, exist_ok=True)
print(f'폴더 "{folder_path}" 가 생성되었습니다.')

폴더 일괄 삭제

import shutil
import os
from datetime import datetime, timedelta

# 30일 전 날짜를 가져오기
threshold_date = datetime.now() - timedelta(days=30)

# 삭제할 폴더의 기준 경로
base_path = 'data/'

# 폴더를 확인하고 오래된 폴더를 삭제
for folder_name in os.listdir(base_path):
    folder_path = os.path.join(base_path, folder_name)
    if os.path.isdir(folder_path):
        folder_date = datetime.strptime(folder_name, '%Y-%m-%d')
        if folder_date < threshold_date:
            shutil.rmtree(folder_path)
            print(f'오래된 폴더 "{folder_path}" 가 삭제되었습니다.')

6. 요약