How to Detect Unused Images in a Project with Python

import os
import re
from pathlib import Path

def find_unused_images(project_path):
    image_exts = {'.png', '.jpg', '.jpeg', '.gif', '.svg', '.webp'}
    used_images = set()
    all_images = set()
    
    # Find all image files
    for root, _, files in os.walk(project_path):
        for file in files:
            if Path(file).suffix.lower() in image_exts:
                all_images.add(os.path.join(root, file))
    
    # Find references in HTML/CSS/JS files
    for root, _, files in os.walk(project_path):
        for file in files:
            if file.endswith(('.html', '.css', '.js')):
                filepath = os.path.join(root, file)
                try:
                    with open(filepath, 'r', encoding='utf-8', errors='ignore') as f:
                        content = f.read()
                        # Find image references in src, url(), etc.
                        refs = re.findall(r'[\'"]?([\w./-]+\.(?:png|jpg|jpeg|gif|svg|webp))[\'"]?', content, re.IGNORECASE)
                        for ref in refs:
                            # Resolve relative paths
                            abs_path = os.path.normpath(os.path.join(os.path.dirname(filepath), ref))
                            if os.path.exists(abs_path):
                                used_images.add(abs_path)
                except (IOError, UnicodeDecodeError):
                    continue
    
    unused = all_images - used_images
    return sorted(unused)

if __name__ == "__main__":
    import tempfile
    import shutil
    
    # Create test project structure
    test_dir = tempfile.mkdtemp()
    os.makedirs(os.path.join(test_dir, 'images'))
    os.makedirs(os.path.join(test_dir, 'css'))
    
    # Create some test files
    Path(os.path.join(test_dir, 'images', 'logo.png')).touch()
    Path(os.path.join(test_dir, 'images', 'banner.jpg')).touch()
    Path(os.path.join(test_dir, 'images', 'old_bg.gif')).touch()
    Path(os.path.join(test_dir, 'images', 'icon.svg')).touch()
    
    with open(os.path.join(test_dir, 'index.html'), 'w') as f:
        f.write('<img src="images/logo.png" alt="Logo">')
    
    with open(os.path.join(test_dir, 'css', 'style.css'), 'w') as f:
        f.write('background: url("../images/banner.jpg");')
    
    unused = find_unused_images(test_dir)
    print("Unused images found:")
    for img in unused:
        print(f"  {os.path.relpath(img, test_dir)}")
    
    shutil.rmtree(test_dir)