[Python] PDF 파일을 특정 페이지를 기준으로 두 개로 나누기

PDF 파일을 특정 페이지를 기준으로 하여 두 개의 PDF 파일로 나누는 스크립트입니다.

 

p.1 ~ p.(특정 페이지 - 1)까지를 하나의 파일로,

 

p.(특정 페이지) ~ p.(마지막 페이지)를 다른 하나의 파일로 생성하며,

 

파일명에 페이지 범위를 표기합니다.

 

제가 테스트해봤는데 일단 잘 작동하네요.

 

추후에 더 간명하게 다시 작성해보겠습니다 ^^

 

 

from pdfrw import PdfReader, PdfWriter


def split_pdf(p, l, s):
    parts = [(1, s), (s, l)]
    for part in parts:
        start_page = part[0]
        last_page = part[1] - 1
        outdata = PdfWriter(f'pages_{start_page}_{last_page}.pdf')
        for pagenum in range(*part):
            outdata.addpage(p[pagenum-1])
        outdata.write()


pdf_file = r'파일명을 입력하세요'
pages = PdfReader(pdf_file).pages
length = len(pages) + 1
splitter = 나눌 페이지를 입력하세요
split_pdf(pages, length, splitter)