add pdf splitter tutorial

x4nth055 · x4nth055 · commit abab573613cc · 2021-12-12T11:53:34.000+01:00
diff --git a/README.md b/README.md
@@ -167,6 +167,7 @@ This is a repository of all the tutorials of [The Python Code](https://www.thepy
     - [How to Merge PDF Files in Python](https://www.thepythoncode.com/article/merge-pdf-files-in-python). ([code](handling-pdf-files/pdf-merger))
     - [How to Sign PDF Files in Python](https://www.thepythoncode.com/article/sign-pdf-files-in-python). ([code](handling-pdf-files/pdf-signer))
     - [How to Extract PDF Metadata in Python](https://www.thepythoncode.com/article/extract-pdf-metadata-in-python). ([code](handling-pdf-files/extract-pdf-metadata))
+    - [How to Split PDF Files in Python](https://www.thepythoncode.com/article/split-pdf-files-in-python). ([code](handling-pdf-files/split-pdf))
     
 - ### [Python for Multimedia](https://www.thepythoncode.com/topic/python-for-multimedia)
     - [How to Make a Screen Recorder in Python](https://www.thepythoncode.com/article/make-screen-recorder-python). ([code](general/screen-recorder))
diff --git a/handling-pdf-files/split-pdf/README.md b/handling-pdf-files/split-pdf/README.md
@@ -0,0 +1 @@
+# [How to Split PDF Files in Python](https://www.thepythoncode.com/article/split-pdf-files-in-python)
diff --git a/handling-pdf-files/split-pdf/bert-paper.pdf b/handling-pdf-files/split-pdf/bert-paper.pdf
diff --git a/handling-pdf-files/split-pdf/requiremenst.txt b/handling-pdf-files/split-pdf/requiremenst.txt
@@ -0,0 +1 @@
+pikepdf
diff --git a/handling-pdf-files/split-pdf/split_pdf.py b/handling-pdf-files/split-pdf/split_pdf.py
@@ -0,0 +1,42 @@
+import os
+from pikepdf import Pdf
+
+# a dictionary mapping PDF file to original PDF's page range
+file2pages = {
+    0: [0, 9], # 1st splitted PDF file will contain the pages from 0 to 9 (9 is not included)
+    1: [9, 11], # 2nd splitted PDF file will contain the pages from 9 (9 is included) to 11
+    2: [11, 100], # 3rd splitted PDF file will contain the pages from 11 until the end or until the 100th page (if exists)
+}
+
+# the target PDF document to split
+filename = "bert-paper.pdf"
+# load the PDF file
+pdf = Pdf.open(filename)
+# make the new splitted PDF files
+new_pdf_files = [ Pdf.new() for i in file2pages ]
+# the current pdf file index
+new_pdf_index = 0
+# iterate over all PDF pages
+for n, page in enumerate(pdf.pages):
+    if n in list(range(*file2pages[new_pdf_index])):
+        # add the `n` page to the `new_pdf_index` file
+        new_pdf_files[new_pdf_index].pages.append(page)
+        print(f"[*] Assigning Page {n} to the file {new_pdf_index}")
+    else:
+        # make a unique filename based on original file name plus the index
+        name, ext = os.path.splitext(filename)
+        output_filename = f"{name}-{new_pdf_index}.pdf"
+        # save the PDF file
+        new_pdf_files[new_pdf_index].save(output_filename)
+        print(f"[+] File: {output_filename} saved.")
+        # go to the next file
+        new_pdf_index += 1
+        # add the `n` page to the `new_pdf_index` file
+        new_pdf_files[new_pdf_index].pages.append(page)
+        print(f"[*] Assigning Page {n} to the file {new_pdf_index}")
+
+# save the last PDF file
+name, ext = os.path.splitext(filename)
+output_filename = f"{name}-{new_pdf_index}.pdf"
+new_pdf_files[new_pdf_index].save(output_filename)
+print(f"[+] File: {output_filename} saved.")

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1 @@`
	`1`	`+# [How to Split PDF Files in Python](https://www.thepythoncode.com/article/split-pdf-files-in-python)`