Python/Python 예제 코드

디렉토리 존재 여부 확인 후, 새로 디렉토리 생성하는 Python 예제

webnautes 2024. 4. 23. 21:35
반응형

디렉토리가 존재하는지 여부를 확인한 후, 새로 디렉토리를 생성하는 Python 예제입니다.



2024. 4. 23  최초작성




처음 실행하면 디렉토리를 생성합니다.

 

./img 디렉토리가 생성되었습니다.



이제부터 실행하면 디렉토리를 삭제하고 새로 생성할지 물어봅니다. n를 선택하면 디렉토리를 그대로 유지합니다. 

 

./img 디렉토리가 존재합니다. 삭제하고 새로 생성하시겠습니까? (y/n): n



다시 한번 실행하여 이번엔 y를 선택하면 디렉토리를 삭제한 후, 디렉토리를 새로 생성합니다. 

 

./img 디렉토리가 존재합니다. 삭제하고 새로 생성하시겠습니까? (y/n): y
./img 디렉토리가 삭제되었습니다.
./img 디렉토리가 생성되었습니다.




전체 코드입니다. 

 

import os
import shutil


save_dir = './img'


response = 'y'

# 디렉토리가 존재하는지 체크
if os.path.exists(save_dir):

    # 디렉토리를 지울지 물어봄
    response = input(f'{save_dir} 디렉토리가 존재합니다. 삭제하고 새로 생성하시겠습니까? (y/n): ')

    if response.lower() == 'y':

        # 디렉토리 삭제
        shutil.rmtree(save_dir)
        print(f'{save_dir} 디렉토리가 삭제되었습니다.')

# 디렉토리 새로 생성함
if response == 'y':
    os.makedirs(save_dir)
    print(f'{save_dir} 디렉토리가 생성되었습니다.')




반응형