|
| 1 | +import PyPDF2 |
| 2 | + |
| 3 | +def split_pdf(input_pdf, output_path, page_ranges): |
| 4 | + with open(input_pdf, 'rb') as file: |
| 5 | + pdf_reader = PyPDF2.PdfFileReader(file) |
| 6 | + for i, page_range in enumerate(page_ranges): |
| 7 | + start_page, end_page = page_range |
| 8 | + pdf_writer = PyPDF2.PdfFileWriter() |
| 9 | + for page_num in range(start_page - 1, end_page): |
| 10 | + pdf_writer.addPage(pdf_reader.getPage(page_num)) |
| 11 | + output_pdf = f"{output_path}/split_{i + 1}.pdf" |
| 12 | + with open(output_pdf, 'wb') as output_file: |
| 13 | + pdf_writer.write(output_file) |
| 14 | + |
| 15 | +def merge_pdfs(input_pdfs, output_pdf): |
| 16 | + pdf_writer = PyPDF2.PdfFileWriter() |
| 17 | + for pdf in input_pdfs: |
| 18 | + with open(pdf, 'rb') as file: |
| 19 | + pdf_reader = PyPDF2.PdfFileReader(file) |
| 20 | + for page_num in range(pdf_reader.numPages): |
| 21 | + pdf_writer.addPage(pdf_reader.getPage(page_num)) |
| 22 | + with open(output_pdf, 'wb') as output_file: |
| 23 | + pdf_writer.write(output_file) |
| 24 | + |
| 25 | +# Example usage for splitting |
| 26 | +input_pdf = 'input.pdf' |
| 27 | +output_path = 'output' |
| 28 | +page_ranges = [(1, 3), (4, 6)] # List of tuples representing page ranges |
| 29 | +split_pdf(input_pdf, output_path, page_ranges) |
| 30 | + |
| 31 | +# Example usage for merging |
| 32 | +input_pdfs = ['input1.pdf', 'input2.pdf'] |
| 33 | +output_pdf = 'merged_output.pdf' |
| 34 | +merge_pdfs(input_pdfs, output_pdf) |
0 commit comments