Command-line compatibility added

JPG to PDF.py

@@ -1,6 +1,8 @@
 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.
@@ -9,40 +11,63 @@
 # --------------- USER INPUT -------------------- #
 
 folder = ""                                                    # Folder containing all the images.
-name = ""                                                      # Name of the output PDF file.
-
-
-# ------------- 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(".jpg")]:
-        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, "F")                                 # Save the PDF.
-
-print("PDF generated successfully!")
+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)]:
+    		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:])