Python: Merge all images into one pdf

gistfile1.txt

> python pdf.py -f 1.png 2.png 3.png

> python pdf.py -f 1.png 2.png -n outname

pdf.py

from argparse import ArgumentParser

from PIL import Image

if __name__ == "__main__":
    parser = ArgumentParser()
    parser.add_argument("--files", "-f", type=str, required=True, nargs="+")
    parser.add_argument("--name", "-n", type=str)

    args = parser.parse_args()

    img_list = []
    base_img = Image.open(args.files[0])
    for img_path in args.files[1:]:
        img_list.append(Image.open(img_path))

    if args.name and not args.name.endswith(".pdf"):
        args.name += ".pdf"

    pdf_name = args.name if args.name else "assignment.pdf"

    base_img.save(
        pdf_name, "PDF", resolution=100.0, save_all=True, append_images=img_list
    )