PDF-merger.py

"""
PDF Merger Utility (Production Ready)
Description: Merges multiple PDFs from different directories into one file.
Author: Rizwan Ahmed (RZ-Logic)
"""

import pypdf
import sys
import shlex
import warnings
from pathlib import Path

# Suppress specific PyPDF warnings
warnings.filterwarnings("ignore", category=UserWarning, module="pypdf")

def pdf_combiner(pdf_list, output_filename='super.pdf'):
    merger = pypdf.PdfWriter()
    try:
        for pdf in pdf_list:
            print(f"Adding: {pdf.name}")
            merger.append(pdf)
            
        output_path = Path.cwd() / output_filename
        with open(output_path, 'wb') as out:
            merger.write(out)
            
        print(f"\n✅ Created: {output_path}")
    except Exception as e:
        print(f"\n❌ Error: {e}")
    finally:
        merger.close()

def show_examples():
    example_text = """
💡 TIP: You can drag and drop files or folders directly into this terminal!

USAGE EXAMPLES:
1. Single File:  "C:\\Docs\\Invoice.pdf"
2. Entire Folder: "C:\\Users\\HomePC\\Downloads"
3. Bulk Paste:   "File1.pdf" "File2.pdf" "C:\\Invoices"
4. Finish:       Type 'done' to merge everything.
5. Cancel:       Type 'quit' or 'exit' to close without merging.
    """
    print(example_text)

def main():
    print("--- 📄 Professional PDF Merger ---")
    show_examples()
    
    pdf_files = []
    
    while True:
        user_input = input("Enter path(s), 'done', or 'quit': ").strip()
        
        # Check for exit commands first
        if user_input.lower() in ['quit', 'exit']:
            print("\n👋 Exiting program. No files were merged.")
            return # Exits the main function immediately

        if user_input.lower() == 'done':
            break
        
        if not user_input:
            continue
            
        try:
            paths = shlex.split(user_input)
        except ValueError:
            print("❌ Formatting error. Check your quotes!")
            continue

        for p in paths:
            path = Path(p.strip('"\'')).resolve()
            
            if path.is_file():
                if path.suffix.lower() == '.pdf':
                    pdf_files.append(path)
                    print(f"  + Added File: {path.name}")
                else:
                    print(f"  x Ignored: {path.name} (Not a PDF)")
            elif path.is_dir():
                found = sorted(path.glob("*.pdf"))
                if found:
                    pdf_files.extend(found)
                    print(f"  + Added {len(found)} PDFs from folder: {path.name}")
                else:
                    print(f"  ⚠️ No PDFs found in: {path.name}")
            else:
                print(f"  ❌ Path not found: {p}")

    if pdf_files:
        print(f"\n⏳ Merging {len(pdf_files)} total files...")
        pdf_combiner(pdf_files)
    else:
        print("\n⚠️ No files selected. Exiting.")

if __name__ == "__main__":
    try:
        main()
    except KeyboardInterrupt:
        print("\n\n🛑 Process stopped by user.")
        sys.exit(0)