Good PDF Merge Script

gistfile1.txt

#! /usr/local/bin/python3.12

from pypdf import PdfWriter
from pathlib import Path
import sys
import os

def get_files(directory):
    pdfs = []

    for fname in os.listdir(directory):
        if fname.lower().endswith('.pdf'):
            pdfs.append(os.path.join(directory, fname))

    return pdfs

def merge_pdfs(directory, pdfs):
    merger = PdfWriter()
    [ merger.append(str(pdf)) for pdf in pdfs ]
    merger.write((dir / f"{directory}-merged.pdf").as_posix())
    merger.close()

if __name__ == '__main__':
    if len(sys.argv) == 2:
        dir = sys.argv[1]
        dir = Path(dir)
        pdfs = get_files(dir)
        merge_pdfs(dir, pdfs)
    else:
        print("You must supply the directory, fool")