pdf_vd.py

#!/usr/bin/env python3

import os
import subprocess
import glob

# 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 recursively scan for PDF files and process them
def scan_and_process(directory):
    count = 0
    total_invalid = 0
    deleted_count = 0
    
    for pdf_file in glob.iglob(os.path.join(directory, "**", "*.pdf"), recursive=True):
        count += 1
        if process_pdf(pdf_file):
            os.remove(pdf_file)
            print(f"Deleted corrupted or invalid PDF: {pdf_file}")
            total_invalid += 1
            deleted_count += 1

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

    print("\nScan complete.")
    print(f"Processed: {count} files | Invalid PDFs 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)
    
    output_dir = f"/mnt/k/PDF_Validation_Output_{subprocess.check_output(['date', '+%Y-%m-%d_%H-%M-%S']).decode().strip()}"
    os.makedirs(output_dir, exist_ok=True)
    
    print(f"\nScanning directory: {directory}")
    print(f"Logging details to: {output_dir}/Validation_Log.txt")
    
    # Redirect stdout to the log file
    sys.stdout = open(f"{output_dir}/Validation_Log.txt", "w")
    
    scan_and_process(directory)
    
    sys.stdout.close()
    sys.stdout = sys.__stdout__  # Reset stdout
    
    print(f"\nOutput saved to: {output_dir}")

if __name__ == "__main__":
    import sys
    main()