Move rotating and cropping solutions to a new file

somacdivad · somacdivad · commit 35c2cfc8c556 · 2020-01-10T10:52:44.000-06:00
diff --git a/ch13-interact-with-pdf-files/4-rotating-and-cropping-pdf-pages.py b/ch13-interact-with-pdf-files/4-rotating-and-cropping-pdf-pages.py
@@ -0,0 +1,41 @@
+# 13.2 - Extract Pages From a PDF
+# Solutions to review exercises
+
+import os
+import copy
+from pyPDF2 import PdfFileReader, PdfFileWriter
+
+
+# Exercise 1
+path = "C:/python-basics-exercises/ch13-interact-with-pdf-files/\
+        practice_files"
+
+input_file_path = os.path.join(path, "Walrus.pdf")
+input_file = PdfFileReader(input_file_path)
+output_PDF = PdfFileWriter()
+
+input_file.decrypt("IamtheWalrus")  # decrypt the input file
+
+
+# Exercise 2
+for page_num in range(0, input_file.getNumPages()):
+    # rotate pages (call everything page_left for now; will make a copy)
+    page_left = input_file.getPage(page_num)
+    page_left.rotateCounterClockwise(90)
+
+    page_right = copy.copy(page_left)  # split each page in half
+    upper_right = page_left.mediaBox.upperRight  # get original page corner
+
+    # crop and add left-side page
+    page_left.mediaBox.upperRight = (upper_right[0] / 2, upper_right[1])
+    output_PDF.addPage(page_left)
+    # crop and add right-side page
+    page_right.mediaBox.upperLeft = (upper_right[0] / 2, upper_right[1])
+    output_PDF.addPage(page_right)
+
+
+# Exercise 3
+# save new pages to an output file
+output_file_path = os.path.join(path, "Output/Updated Walrus.pdf")
+with open(output_file_path, "wb") as output_file:
+    output_PDF.write(output_file)
