-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
rename file and merge cli mode (praneetk2704/JPG-to-PDF#2)
- Loading branch information
Showing
2 changed files
with
135 additions
and
37 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,73 @@ | ||
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:]) |