[python] PDF 파일에서 특정 유형의 요소 수정하기

30 Nov 2023

python

PDF는 널리 사용되는 문서 형식 중 하나이며 Python을 사용하여 PDF 파일을 수정하는 것은 매우 유용합니다. 특히 특정 유형의 요소를 수정하는 기능은 많은 작업에 도움이 됩니다.

PyPDF2 라이브러리 설치

PDF 파일을 수정하기 위해 PyPDF2 라이브러리를 사용할 것입니다. 먼저 해당 라이브러리를 설치해야 합니다. 아래의 명령어를 사용하여 PyPDF2를 설치합니다.

pip install PyPDF2

특정 유형의 요소 수정하기

PDF 파일에서 특정 유형의 요소를 수정하기 위해서는 다음 단계를 따라야 합니다.

PdfFileReader를 사용하여 PDF 파일을 엽니다.
페이지 수를 확인하고 원하는 페이지를 선택합니다.
원하는 요소를 수정합니다.
변경된 내용을 새로운 PDF 파일에 저장합니다.

이를 구체적으로 설명하기 위해 예제 코드를 제공하겠습니다.

from PyPDF2 import PdfFileReader, PdfFileWriter

def modify_pdf_element(filename, page_number, element_text):
    # 원본 PDF 파일 열기
    with open(filename, 'rb') as file:
        pdf_reader = PdfFileReader(file)
        
        # 새로운 PDF 파일 생성을 위한 객체 생성
        pdf_writer = PdfFileWriter()
        
        # 원하는 페이지 선택
        page = pdf_reader.getPage(page_number)
        
        # 원하는 요소 수정
        page.extract_text = element_text
        
        # 변경된 페이지를 새로운 PDF에 추가
        pdf_writer.addPage(page)
        
        # 새로운 PDF 파일 저장
        with open('modified.pdf', 'wb') as output_file:
            pdf_writer.write(output_file)

# 예제 사용법
modify_pdf_element('original.pdf', 2, '새로운 요소 내용')

위의 코드 예제에서 modify_pdf_element 함수는 filename에 있는 원본 PDF 파일을 열고, page_number에 해당하는 페이지를 선택합니다. 그리고 element_text로 지정된 내용으로 해당 요소를 수정합니다. 최종적으로 변경된 페이지를 modified.pdf라는 이름의 새로운 PDF 파일에 저장합니다.

이제 위의 예제 코드를 기반으로 필요한 요소를 수정하는 기능을 구현할 수 있습니다.