파이썬 폴더 정리 자동화하기 (os.mkdir/os.path.join)

💁🏻‍♀️ 폴더 내에 다양한 파일들을 정리해보자!

1)특정 제목으로 시작하는 파일 정리하기 - Screen Shot xxx
2)특정 확장자명을 가진 파일 정리하기 - 이미지 파일 (png, jpg, jpeg등)

OS 모듈 

 파이썬에 기본 내장되어있는 모듈로 os은 운영체제 종속 기능에 대한 간단한 명령을 모아놓은 기본모듈이다. 자세한 문서는 python docs

폴더 생성 - os.mkdir(경로)

import os

#상대 경로
os.mkdir('file')
os.mkdir('folder/file')

#절대 경로
os.mkdir('user/documents/folder')

• os.mkdir은 파이썬에서 폴더를 생성할 때 사용하는 함수
• 지정한 경로에 같은 이름의 폴더가 존재하는 경우 에러가 발생합니다.

#오류없이 실행되기
path = '/Users/hollyyoon/Documents/folder'
os.makedirs(path, exist_ok=True)

#폴더가 없을때만 만들기
if not os.path.exists('/Users/hollyyoon/Documents/folder'):
	mkdirs('/Users/hollyyoon/Documents/folder')

• 위의 코드처럼 makedirs를 사용하고 exist_ok=True 옵션을 사용할 경우 포더가 존재하더라도 오류없이 프로그램이 실행됩니다.
• 혹은 조건식을 활용하여, 같은 이름의 폴더가 없을 때만 만들도록 명령할 수 있습니다.

파일 리스트 가져오기 - os.listdir(경로)

import os
file_list = os.listdir("/Users/hollyyoon/Desktop")

• 결과는 리스트 형태로 출력됩니다.

파일 경로 - os.path()

1) os.path.split / splitext

#path를 디렉토리와 파일로 분리
import os
os.path.split("/Users/hollyyoon/Desktop/images.png")

#실행 결과 "/Users/hollyyoon/Desktop", "images.png"

#파일 이름과 확장자명 분리
import os
os.path.splitext("/Users/hollyyoon/Desktop/images.png")

#실행 결과 "images", ".png"

• splitext함수는 튜플(데이터가 변하지 않는 자료형) 형태로 반환합니다.

2) os.path.join

#각각의 경로를 하나로 이어줌
import os
os.path.join("/Users/hollyyoon","Desktop","images.png")

#실행 결과 "/Users/hollyyoon/Desktop/images.png"

 

파일 정리 자동화하기

1. 폴더 지정하기 (정리할 폴더/정리한 결과가 들어갈 새 폴더)
2. 정리할 폴더에 있는 파일 확인하기 (확장자명 확인)
3. 타겟 파일들을 정리한 결과가 들어갈 폴더로 경로 변경해주기

1)특정 제목으로 시작하는 파일 정리하기 - Screen xxx

import os

before = '/Users/hollyyoon/Desktop'
after = '/Users/hollyyoon/Desktop/screenshot'
if not os.path.exists('/Users/hollyyoon/Desktop/screenshot'):
	os.mkdir('/Users/hollyyoon/Desktop/screenshot')

file_list = os.listdir(before)
for file in file_list :
    if file.startswith('Screen'):
        before_file = os.path.join(before,file)
        after_file = os.path.join(after,file)
        os.rename(before_file, after_file)

 

2)특정 확장자명을 가진 파일 정리하기 - 이미지 파일 (png, jpg, jpeg등)

import os

before = '/Users/hollyyoon/Downloads'
after = '/Users/hollyyoon/Downloads/img'
ext_list = ['.jpg', '.png', '.jpeg', '.gif', ',HEIC']
if not os.path.exists('/Users/hollyyoon/Downloads/img'):
	os.mkdir('/Users/hollyyoon/Downloads/img')

file_list = os.listdir(before)
for file in file_list :
    name, ext = os.path.splitext(file)
    if ext in ext_list:
        before_file = os.path.join(before,file)
        after_file = os.path.join(after,file)
        os.rename(before_file, after_file)