[회사원] 업무 코딩기록 - (PDrwF)File Slicing(1EA)

import os

import pdfrw

# 분할 대상 PDF 파일명

target_pdf_file = 'test.pdf'

# 분할 대상 PDF가 있는 폴더 (실제 경로)

input_folder = r'C:\Users\user\Desktop\VSCode(python)\(CODE)PDF,OCR\pdf_raw'

# 분할 결과 PDF를 저장할 폴더

output_folder = r'C:\Users\user\Desktop\VSCode(python)\(CODE)PDF,OCR\pdf_result'

# 분할 대상 PDF 가져오기

input_pdf_path = os.path.join(input_folder, target_pdf_file)

# 파일이 존재하는지 확인

if os.path.isfile(input_pdf_path):

    input_pdf = pdfrw.PdfReader(input_pdf_path)

    # PDF 분할

    for page_num, page in enumerate(input_pdf.pages):

        output_pdf = pdfrw.PdfWriter()

        output_pdf.addpage(page)

        pdf_result = f'{os.path.splitext(target_pdf_file)[0]}_분할_page{page_num+1}.pdf'

        # 'pdf_result' 폴더 생성하고 저장하기

        if not os.path.isdir(output_folder):

            os.mkdir(output_folder)

        output_pdf.write(os.path.join(output_folder, pdf_result))

else:

    print(f"{target_pdf_file} 파일이 {input_folder} 폴더에 존재하지 않습니다.")