写个小脚本讨媳妇儿开心

晚上媳妇儿整理儿子照片，整理了一部分说太累了；问她还有多少没整理，说还有几十个G…问怎么整理，说把图片根据生日按月份分开；我一听，明显是个小脚本就搞定的事情；虽然没怎么太写过python，不过撸起袖子就是干；一边查一边写，轻松搞定；一条命令，几十个G图片一会儿就分好了；媳妇儿很开心；

会写代码的意义不一定在于非要做多大的项目，生活中有些重复的体力活，写个脚本一劳永逸，其余的时间开开心心发呆也很好；

附代码，规范性差了点，自用轻拍；

# -*- coding: UTF8 -*-

import os
import time
import exifread
import shutil
import sys

# 生日分隔日期
BIRTHDAY = 7

# 原图片目录
ORIGINAL_DIR = 'original'

# 分类结果目录
CLASSIFY_DIR = 'classify'



# 获取图片时间
def getImageTime(imagePath):
    FIELD = 'EXIF DateTimeOriginal'
    fd = open(imagePath,'rb')
    tags = exifread.process_file(fd)
    fd.close()
    # 有拍摄时间优先使用拍摄时间
    if FIELD in tags:
        imageTime = str(tags[FIELD]).replace(':', '-')[:10]
        return imageTime
    else:
        state = os.stat(imagePath)
        imageTime = time.strftime('%Y-%m-%d', time.localtime(state[-2]))
        return imageTime



# 00.简单检查下待处理图片目录
if not os.path.exists(ORIGINAL_DIR):
    print('请将图片文件夹复制到original目录下')
    sys.exit(-1)
        
# 01.创建分类目录
if os.path.exists(CLASSIFY_DIR):
    shutil.rmtree(CLASSIFY_DIR)
os.mkdir(CLASSIFY_DIR)
        
# 02.遍历当前目录下的图片目录
for root, dirs, files in os.walk(ORIGINAL_DIR):
    for image in files:
        # 拼接文件路径
        imagePath = os.path.join(root, image)
        print(imagePath)
        
        # 获取文件时间
        imageTime = getImageTime(imagePath)
        
        year = int(imageTime[:4])
        month = int(imageTime[5:7])
        day = int(imageTime[-2:])

        # 以生日做分隔，小于7号的归到上个月
        if (day < BIRTHDAY):
            month -= 1
            # 1月归档到上个月，需要退年
            if (month == 0):
                year -= 1
                month = 12
        
        # 生成该图片需要存储的目录
        t = (year, month, BIRTHDAY + 1, 0, 0, 0, 0, 0, 0)
        newTime = time.mktime(t)
        newImageDirName = time.strftime('%Y-%m-%d', time.gmtime(newTime))
        newImageDir = os.path.join(CLASSIFY_DIR, newImageDirName)
        
        # 如果目标目录不存在则创建
        if not os.path.exists(newImageDir):
            os.mkdir(newImageDir)
        
        dstImagePath = os.path.join(newImageDir, image)
        
        # 拷贝文件
        shutil.copy2(imagePath, dstImagePath)