반응형

디렉토리가 존재하는지 여부를 확인한 후, 새로 디렉토리를 생성하는 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} 디렉토리가 생성되었습니다.')




반응형

문제 발생시 지나치지 마시고 댓글 남겨주시면 가능한 빨리 답장드립니다.

도움이 되셨다면 토스아이디로 후원해주세요.
https://toss.me/momo2024


제가 쓴 책도 한번 검토해보세요 ^^

+ Recent posts