pdf_image_vd_pillow.py

#!/usr/bin/env python3

import os
import subprocess
import glob
from PIL import Image

# Function to process PDF files
def process_pdf(pdf_file):
    pdfinfo_process = subprocess.run(["pdfinfo", pdf_file], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
    return pdfinfo_process.returncode == 0

# Function to process JPEG files
def process_jpeg(jpeg_file):
    try:
        Image.open(jpeg_file).verify()
    except Exception:
        os.remove(jpeg_file)
        print(f"Deleted corrupted or invalid JPEG: {jpeg_file}")
        return False
    return True

# Function to recursively scan for files (PDF and JPEG) and process them
def scan_and_process(directory):
    count = 0
    total_invalid = 0
    deleted_count = 0
    
    print(f"Scanning directory: {directory}\n")
    
    for file_path in glob.iglob(os.path.join(directory, "**", "*.*"), recursive=True):
        _, file_extension = os.path.splitext(file_path)
        if os.path.isfile(file_path) and (file_extension.lower() == ".pdf" or file_extension.lower() == ".jpeg"):
            count += 1
            if file_extension.lower() == ".pdf":
                if process_pdf(file_path):
                    print(f"Validated PDF: {file_path}")
                else:
                    total_invalid += 1
                    deleted_count += 1
            elif file_extension.lower() == ".jpeg":
                if process_jpeg(file_path):
                    print(f"Validated JPEG: {file_path}")
                else:
                    total_invalid += 1
                    deleted_count += 1

            if count % 1000 == 0:
                print(f"Processed: {count} files | Invalid files found: {total_invalid} | Files deleted: {deleted_count}")

    print("\nScan complete.")
    print(f"Processed: {count} files | Invalid files found: {total_invalid} | Files deleted: {deleted_count}")

# Main function
def main():
    drive_menu = [
        "C:",
        "D:",
        "E:",
        "Custom Directory",
        "Quit"
    ]
    
    print("Select a drive or directory to scan:")
    for index, option in enumerate(drive_menu, start=1):
        print(f"{index}) {option}")
    
    choice = int(input())
    
    if choice == 5:
        print("Quitting.")
        return
    
    if choice == 4:
        custom_path = input("Enter the path to the custom directory (e.g., C:/path/to/directory): ")
        directory = os.path.normpath(custom_path)
    else:
        drive = drive_menu[choice - 1]
        directory = os.path.join("/mnt", drive)
    
    scan_and_process(directory)
    print("\nScript execution finished.")

if __name__ == "__main__":
    main()