상세 컨텐츠

본문 제목

python 만으로 날짜 네이밍 폴더 자동화 하기

머신러닝/함수, 라이브러리

by byoelcardi 2023. 8. 10. 19:04

본문

데이터 크롤링할 때 날짜 별로 데이터를 저장하려면 폴더의 이름이 날짜인 경우가 필요할 때가 있습니다. 이런 경우에 필요한 코드를 소개드리려고 합니다.

import os
import stat
import datetime

# 폴더를 생성할 경로 설정
path = "/path"

def make_folder():
    # 오늘의 날짜 계산
    today = datetime.date.today()
    folder_name = today.strftime("%Y-%m-%d")

    # 폴더 생성
    folder_path = os.path.join(path, folder_name)

    # 폴더가 없다면 생성하고 권한 부여
    if not os.path.exists(folder_path):
        os.makedirs(folder_path)
        os.chmod(folder_path, stat.S_IRWXU | stat.S_IRWXG | stat.S_IRWXO)

        print("폴더가 성공적으로 생성되었습니다:", folder_path)
    else:
        print("이미 폴더가 존재합니다:", folder_path)

if __name__ == "__main__":
    make_folder()
  • 라이브러리: 코드의 첫 부분에서 필요한 라이브러리들을 임포트 합니다. os, stat, datetime 모듈이 사용됩니다.
  • 경로 설정: path 변수에는 폴더를 생성할 경로를 지정합니다. 새로운 폴더를 만들 위치를 의미합니다.
  • make_folder 함수 정의: make_folder 함수는 오늘의 날짜를 기반으로 폴더 이름을 생성하고, 해당 경로에 폴더를 생성하며 권한을 설정합니다.
  • 오늘의 날짜 계산: datetime.date.today()를 사용하여 현재 날짜를 가져옵니다.
  • 폴더 이름 생성: today.strftime("%Y-%m-%d")를 통해 오늘의 날짜를 "YYYY-MM-DD" 형식의 문자열로 변환합니다.
  • 폴더 생성: os.path.join() 함수를 사용하여 path와 folder_name을 결합하여 새 폴더의 전체 경로를 생성합니다.
  • 폴더 확인 및 생성: os.path.exists() 함수로 폴더의 존재 여부를 확인하고, 폴더가 존재하지 않을 경우 os.makedirs() 함수로 폴더를 생성합니다.
  • 권한 설정: os.chmod() 함수를 사용하여 생성된 폴더에 권한을 부여합니다. stat.S_IRWXU, stat.S_IRWXG, stat.S_IRWXO는 각각 사용자, 그룹, 기타 사용자에 대한 읽기/쓰기/실행 권한을 나타냅니다.
  • 출력: 폴더 생성 여부에 따라 적절한 메시지를 출력합니다.
  • 메인 부분: if __name__ == "__main__":을 사용하여 이 코드 블록이 직접 실행되는 경우에만 make_folder() 함수를 호출하여 폴더 생성 및 권한 설정 작업을 수행합니다.

 

 

 

 

 

 

 

관련글 더보기