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} 디렉토리가 생성되었습니다.') |
반응형