JPG to PDF.py

from fpdf import FPDF
from PIL import Image
import os
import sys
import getopt

pdf = FPDF()
imagelist = []                                                 # Contains the list of all images to be converted to PDF.


# --------------- USER INPUT -------------------- #

folder = ""                                                    # Folder containing all the images.
name = ""  
extension = ""                                                    # Name of the output PDF file.

def main(argv):
    try:
        opts, args = getopt.getopt(argv,"hf:n:e:",["folder=","name=","extension="])
    except getopt.GetoptError:
        print("JPG to PDF -f <folder> -n <name> -e <extension>")
        sys.exit(2)
    for opt, arg in opts:
        if opt == "-h":
            print("JPG to PDF\n -f <folder>\n -n <name>\n -e <extension>\n")
            sys.exit()
        elif opt in ("-f", "--folder"):
            folder = arg
        elif opt in ("-n", "--name"):
            name = arg
        elif opt in ("-e", "--extension"):
            extension = arg
    	
    # ------------- ADD ALL THE IMAGES IN A LIST ------------- #
    for dirpath, dirnames, filenames in os.walk(folder):
    	for filename in [f for f in filenames if f.endswith(extension) or f.lower().endswith(extension)]:
    		full_path = os.path.join(dirpath, filename)
    		imagelist.append(full_path)
    
    #for dirpath, dirnames, filenames in os.walk(folder):
    #	for filename in [f for f in filenames if f.endswith(".jpeg")]:
    #		full_path = os.path.join(dirpath, filename)
    #		imagelist.append(full_path)
    
    imagelist.sort()                                               # Sort the images by name.
    for i in range(0, len(imagelist)):
    	print(imagelist[i])
    
    # --------------- ROTATE ANY LANDSCAPE MODE IMAGE IF PRESENT ----------------- #
    
    for i in range(0, len(imagelist)):
    	im1 = Image.open(imagelist[i])                             # Open the image.
    	width, height = im1.size                                   # Get the width and height of that image.
    	if width > height:
    		im2 = im1.transpose(Image.ROTATE_270)                  # If width > height, rotate the image.
    		os.remove(imagelist[i])                                # Delete the previous image.
    		im2.save(imagelist[i])                                 # Save the rotated image.
    		# im.save
    
    print("\nFound " + str(len(imagelist)) + " image files. Converting to PDF....\n")
    
    
    # -------------- CONVERT TO PDF ------------ #
    
    for image in imagelist:
    	pdf.add_page()
    	pdf.image(image, 0, 0, 210, 297)                           # 210 and 297 are the dimensions of an A4 size sheet.
    
    pdf.output(folder +"/"+ name+".pdf", "F")                                 # Save the PDF.
    
    print("PDF generated successfully!\nAnd saved as "+folder +"/"+ name+".pdf")
if __name__ == "__main__":
   main(sys.argv[1:])