Renaming and Filing Photos and Videos in the HEIC/HEIF era

#!/bin/zsh

for FILE in "$@" 
do
    if [[ ! -f $FILE ]] then
        continue
    fi
    EXTENSION="${FILE##*.}"

    # Ensure we only handle the kinds of media we want
    if [[ "DNG dng JPG JPEG jpeg jpg PNG png MOV mov HEIC heic" != *$EXTENSION* ]] then
        continue
    fi

    # Normalize extensions

    if [[ "DNG" == *$EXTENSION* ]] then
      EXTENSION="dng"
    fi
    if [[ "JPG JPEG jpeg" == *$EXTENSION* ]] then
      EXTENSION="jpg"
    fi
    if [[ "PNG" == *$EXTENSION* ]] then
      EXTENSION="png"
    fi
    if [[ "HEIC" == *$EXTENSION* ]] then
      EXTENSION="heic"
    fi
    if [[ "MOV" == *$EXTENSION* ]] then
      EXTENSION="mov"
    fi

    # Now grab EXIF/IPTC data that MacOS has already figured out for us
    METADATA=$(mdls "$FILE")
    # The space prevents matching derived properties
    ISO_DATE=$(echo $METADATA | grep "kMDItemContentModificationDate " | cut -d= -f 2 | sed 's/[^0-9]//g' | cut -c1-14)
    if [[ ${#ISO_DATE} -eq 14 ]] then 
      if [[ "$FILE" == "$ISO_DATE.$EXTENSION" ]] then
        continue
      fi
      if [[ ! -f "$ISO_DATE.$EXTENSION" ]] then
        mv "$FILE" "$ISO_DATE.$EXTENSION"
      else
        for SUFFIX in {a..z}; do
          SUFFIXED_DATE="$ISO_DATE$SUFFIX"
          if [[ ! -f "$SUFFIXED_DATE.$EXTENSION" ]] then
            mv "$FILE" "$SUFFIXED_DATE.$EXTENSION"
            break
          fi
        done
      fi
    fi
done

#!/bin/env python3

from PIL import Image, ImageFilter, UnidentifiedImageError
from PIL.ExifTags import TAGS
from pi_heif import HeifImagePlugin
from os import listdir, rename, stat, chdir
from stat import S_ISDIR
from os.path import splitext, exists
from pprint import pprint
from ffmpeg import probe
from sys import argv, exit
from time import strftime, gmtime

# build a list of alphabetical suffixes, starting with a blank
SUFFIXES = ['']
SUFFIXES.extend(list(map(chr,range(ord("a"), ord("z")+1))))

PHOTO_EXTENSIONS = [".jpg",".jpeg",".heic", ".png", ".cr2", ".dng", ".gif"]
VIDEO_EXTENSIONS = [".mp4", ".m4v", ".mov"]

def parse_exif(image: Image) -> dict:
    exif = image.info.get("exif")
    if not exif:
        return None
    tags={}
    for k, v in image.getexif().items():
        tag = TAGS.get(k)
        tags[tag] = v
    return tags


def safe_rename(filename: str, date: str, marker: str="-") -> str:
    ext = splitext(filename)[1].lower()
    for s in SUFFIXES:
        new_filename = f"{date}{s}{ext}"
        if not exists(new_filename):
            print(f"{filename} -{marker}-> {new_filename}")
            rename(filename, new_filename)
            return new_filename
            break
    print(f"{filename} -!-> {filename}")
    return filename


def scan_files(path: str) -> int:
    if exists(path) and S_ISDIR(stat(path).st_mode):
        chdir(path)
    else:
        print(f"invalid path {path}")
        return -1

    for filename in listdir():
        (name, ext) = splitext(filename)
        ext = ext.lower()
        # photos
        if ext in PHOTO_EXTENSIONS:
            try:
                image = Image.open(filename, "r")
                tags = parse_exif(image)
                image.close()
            except UnidentifiedImageError:
                tags = None
            # use EXIF data
            if tags and 'DateTime' in tags:
                # We get this as a string, so we can use it right away
                date = tags['DateTime'].replace(" ","").replace(":","")
                if(not filename.startswith(date)):
                    safe_rename(filename, date)
            # use modification date instead (Apple Photos sets it correctly on export)
            else:
                date = strftime("%Y%m%d%H%M%S", gmtime(stat(filename).st_mtime))
                if(not filename.startswith(date)):
                    safe_rename(filename, date, marker="?")
        # video            
        elif ext in VIDEO_EXTENSIONS:
            streams = probe(filename)["streams"]
            for s in streams:
                if 'creation_time' in s['tags']:
                    date = s['tags']['creation_time'].replace("T",'').replace("-","").replace(":","")[:14]
                    if(not filename.startswith(date)):
                        safe_rename(filename, date)
                        break
        else:
            print(f"skipping {filename}")
    return 0

if __name__ == "__main__":
    if len(argv) == 2:
        exit(scan_files(argv[1]))
    else:
        print(f"Usage: {__file__} <path>")