照片整理以及各种形式的除旧迎新

ata1.00: failed to enable AA(error_mask=0x1)
I/O error, dev sda, sector 13385600 op 0x0:(READ) flags ox80700 phys_seg 32 prio class 0

import ffmpeg
from datetime import datetime
import os
import exifread
import shutil

source_folder ='/home/jiyou/ycy/2024'
result_folder ='/home/jiyou/ycy/photo/'

def get_creation_time(file_path):
    creation_time = None
    if creation_time is None:
        f = open(file_path, 'rb')
        tags = exifread.process_file(f)
        if 'EXIF DateTimeOriginal' in tags:
            creation_time = tags['EXIF DateTimeOriginal']
        if creation_time is None:
            if 'Image DateTime' in tags:
                creation_time = tags['Image DateTime']
    if creation_time is None:
        creation_time = datetime.fromtimestamp(os.path.getmtime(file_path))
    return creation_time

def get_file_size(file_path):
    probe = ffmpeg.probe(file_path)
    format_info = probe['format']
    file_size = format_info.get('size', None)
    return file_size

def parse_date_with_am_pm(date_string):
    formats = [
        '%Y:%m:%d %H:%M:%S%P',
        '%Y-%m-%dT%H:%M:%S.%fZ',
        '%Y-%m-%d %H:%M:%S',
        '%Y-%m-%d %H:%M:%S.%f',
        '%Y:%m:%d %H:%M:%S',
        '%Y:%m:%d %H:%M:%S下午',
        '%Y:%m:%d %H:%M:%S上午',
    ]
    if date_string == "0000:00:00 00:00:00":
        return None

    if date_string:
        for fmt in formats:
            try:
                return datetime.strptime(date_string, fmt)
            except ValueError:
                continue
        raise Exception(date_string)
    else:
        return None

def traverse_directory(root_dir):
    for root, dirs, files in os.walk(root_dir):
        for file in files:
            file_path = os.path.join(root, file)
            time_string = get_creation_time(file_path)
            date = parse_date_with_am_pm(str(time_string))
            if date is None:
                print("%s拍摄时间有问题"%file_path)
            else:
                if not os.path.exists(result_folder + str(date.year)):
                    os.makedirs(result_folder + str(date.year))
                if not os.path.exists(result_folder + str(date.year) + "/" + str(date.month)):
                    os.makedirs(result_folder + str(date.year) + "/" + str(date.month))
                new_file_path = result_folder + str(date.year) + "/" + str(date.month) + "/" + file
                if os.path.exists(new_file_path):
                    old_stats = os.stat(new_file_path)
                    new_stats = os.stat(file_path)
                    if old_stats.st_size >= new_stats.st_size:
                        os.remove(file_path)
                        print ("%s新的比较小"%file_path)
                    else:
                        os.remove(new_file_path)
                        shutil.move(file_path, new_file_path)
                        print ("%s新的比较大"%file_path)
                else:
                    print ("%s移动"%file_path)
                    shutil.move(file_path, new_file_path)

traverse_directory(source_folder)

import exifread
from datetime import datetime
import os
import shutil

result_folder = "/home/jiyou/ycy/photos/"
movie_folder = "/home/jiyou/ycy/movies/"

def recursive_list_dir(path):
    files = os.listdir(path)
    for file in files:
        file_path = os.path.join(path, file)
        if os.path.isfile(file_path):
            ext = file.split(".")[-1]
            if ext in ['jpeg','JPEG','JPG', 'jpg','jpg', 'png', 'PNG', 'gif', 'GIF']:
                process_img(os.path.abspath(file_path), file)
            elif ext in ['MOV', 'mov', 'mp4', 'MP4']:
                process_mov(os.path.abspath(file_path), file)
            else:
                print ("%s不是图片 %s"%(os.path.abspath(file),ext))
        elif os.path.isdir(file_path):
            recursive_list_dir(file_path)


def process_img(image_path, image_name):
    img = exifread.process_file(open(image_path,'rb'))
    if 'Image DateTime' in img:
        time = img['Image DateTime']
        if str(time) == "0000:00:00 00:00:00":
            print("%s拍摄时间有问题"%image_path)
        else:
            format = "%Y:%m:%d %H:%M:%S"
            try:
                date = datetime.strptime(str(time).replace('下午',''), format)
                if not os.path.exists(result_folder + str(date.year)):
                    os.makedirs(result_folder + str(date.year))
                if not os.path.exists(result_folder + str(date.year) + "/" + str(date.month)):
                    os.makedirs(result_folder + str(date.year) + "/" + str(date.month))
                
                new_file_path = result_folder + str(date.year) + "/" + str(date.month) + "/" + image_name
                if os.path.exists(new_file_path):
                    old_stats = os.stat(new_file_path)
                    new_stats = os.stat(image_path)
                    if old_stats.st_size >= new_stats.st_size:
                        os.remove(image_path)
                    else:
                        os.remove(new_file_path)
                        shutil.move(image_path, new_file_path)
                        print ("%s新的比较大"%image_path)
                else:
                    shutil.move(image_path, new_file_path)
            except Exception as e:
                print (e)
    else:
        print ("%s没有拍摄信息"%image_path)

def process_mov(image_path, image_name):
    new_file_path = movie_folder +  "/" + image_name
    if os.path.exists(new_file_path):
        old_stats = os.stat(new_file_path)
        new_stats = os.stat(image_path)
        if old_stats.st_size == new_stats.st_size:
            os.remove(image_path)
        else:
            print ("%s已经存在，尺寸不一样"%new_file_path)
    else:
        shutil.move(image_path, new_file_path)

recursive_list_dir(".")