pypixpro–code

#!/usr/bin/env python3

import os
import sys
import shutil
import subprocess
from pathlib import Path
import hashlib
from PIL import Image, ExifTags, UnidentifiedImageError

try:
    import pyheif  # For HEIC files
except ImportError as e:
    subprocess.run(['osascript', '-e', f'display alert "Error: pyheif not found. Install it using pip."'])
    sys.exit(1)

# Constants
STANDARD_IMAGE_EXTENSIONS = {".jpg", ".jpeg", ".gif", ".bmp", ".tiff", ".tif", ".webp", ".psd",
                             ".svg", ".ico", ".jfif", ".pjpeg", ".pjp", ".avif", ".apng"}
PNG_EXTENSION = ".png"
HEIC_EXTENSIONS = {".heic", ".heif"}
DNG_RAW_EXTENSIONS = {".dng", ".raw", ".nef", ".cr2", ".cr3", ".arw", ".orf", ".rw2", ".raf", ".srw", ".kdc"}
MISC_EXTENSIONS = {".doc", ".docx", ".txt", ".py", ".zsh", ".sh", ".mov", ".mp4", ".pdf", ".xlsx", ".pptx"}

# Folder Names
PORTRAIT_FOLDER_NAME = "Portrait"
LANDSCAPE_FOLDER_NAME = "Landscape"
SCREENSHOTS_FOLDER_NAME = "Screenshots"
MISC_FOLDER_NAME = "Misc"

def get_input_folder():
    """Get the folder path from command-line arguments or show a prompt."""
    if len(sys.argv) > 1:
        input_folder_path = sys.argv[1]
        return Path(input_folder_path)
    else:
        prompt_message = (
            "⚠️ Please back up your files before processing them.\n\n"
            "Drag and drop a folder onto the droplet to begin."
        )
        subprocess.run(['osascript', '-e', f'display alert "{prompt_message}"'])
        sys.exit(1)

def get_prefix(folder_name):
    """Prompt for a prefix for Portrait and Landscape folders."""
    try:
        result = subprocess.run(
            ['osascript', '-e', f'display dialog "Enter prefix for {folder_name} photos:" default answer ""'],
            capture_output=True, text=True, check=True
        )
        output = result.stdout.strip().split("text returned:")[-1].strip()
        return output or f"{folder_name} Photo"
    except subprocess.CalledProcessError:
        return f"{folder_name} Photo"

def generate_checksums(root_folder):
    """Generate file checksums using hashlib."""
    checksums = {}
    for path in root_folder.rglob('*'):
        if path.is_file():
            try:
                hasher = hashlib.sha256()
                with path.open('rb') as f:
                    while chunk := f.read(8192):
                        hasher.update(chunk)
                checksum = hasher.hexdigest()
                checksums.setdefault(checksum, []).append(path)
            except Exception:
                pass  # Continue on error
    return checksums

def correct_orientation(image):
    """Correct image orientation using EXIF metadata."""
    try:
        exif = image._getexif()
        if exif is not None:
            orientation = exif.get(274)  # EXIF tag 274 is Orientation
            if orientation == 3:
                image = image.rotate(180, expand=True)
            elif orientation == 6:
                image = image.rotate(270, expand=True)
            elif orientation == 8:
                image = image.rotate(90, expand=True)
        return image
    except Exception as e:
        print(f"Error correcting orientation: {e}")
        return image  # Return original image on error

def rename_files(target_folder, prefix):
    """Rename files with the given prefix."""
    files = sorted(target_folder.glob('*'))
    for idx, file in enumerate(files, 1):
        new_name = f"{prefix} {str(idx).zfill(3)}{file.suffix}"
        try:
            file.rename(target_folder / new_name)
        except Exception:
            pass  # Continue on error

def process_heic_image(path):
    """Process HEIC image to determine its dimensions."""
    try:
        heif_file = pyheif.read(path)
        width, height = heif_file.size
        print(f"Processed HEIC: {path.name} - {width}x{height}")  # Debugging dimensions
        return width, height
    except Exception as e:
        print(f"Error processing HEIC {path}: {e}")
        return None, None

def sort_images(root_folder, portrait_prefix, landscape_prefix):
    """Sort images into folders and apply prefixes."""
    portrait_folder = root_folder / PORTRAIT_FOLDER_NAME
    landscape_folder = root_folder / LANDSCAPE_FOLDER_NAME

    portrait_folder.mkdir(exist_ok=True)
    landscape_folder.mkdir(exist_ok=True)

    screenshots_folder = None
    misc_folder = None

    for path in root_folder.rglob('*'):
        if path.is_file() and path.parent not in {portrait_folder, landscape_folder}:
            suffix = path.suffix.lower()
            try:
                if suffix in HEIC_EXTENSIONS:
                    width, height = process_heic_image(path)
                    if width and height:
                        target = portrait_folder if height > width else landscape_folder
                        shutil.move(str(path), target / path.name)
                elif suffix in STANDARD_IMAGE_EXTENSIONS:
                    with Image.open(path) as img:
                        img = correct_orientation(img)  # Correct orientation for JPEGs
                        width, height = img.size
                        print(f"Processed JPEG: {path.name} - {width}x{height}")  # Debugging dimensions
                    target = portrait_folder if height > width else landscape_folder
                    shutil.move(str(path), target / path.name)
                elif suffix == PNG_EXTENSION:
                    if not screenshots_folder:
                        screenshots_folder = root_folder / SCREENSHOTS_FOLDER_NAME
                        screenshots_folder.mkdir(exist_ok=True)
                    shutil.move(str(path), screenshots_folder / path.name)
                elif suffix in DNG_RAW_EXTENSIONS:
                    shutil.move(str(path), landscape_folder / path.name)
                elif suffix in MISC_EXTENSIONS:
                    if not misc_folder:
                        misc_folder = root_folder / MISC_FOLDER_NAME
                        misc_folder.mkdir(exist_ok=True)
                    shutil.move(str(path), misc_folder / path.name)
            except Exception as e:
                print(f"Error processing {path}: {e}")

    rename_files(portrait_folder, portrait_prefix)
    rename_files(landscape_folder, landscape_prefix)

def main():
    """Main entry point."""
    input_folder = get_input_folder()

    checksums = generate_checksums(input_folder)
    portrait_prefix = get_prefix("Portrait")
    landscape_prefix = get_prefix("Landscape")

    sort_images(input_folder, portrait_prefix, landscape_prefix)

    subprocess.run(['osascript', '-e', 'display alert "Process completed successfully!"'])

if __name__ == "__main__":
    main()