def split_pdf_with_page_numbers()

in seed/util/preprocess.py [0:0]

• 13 lines of code
• 2 McCabe index (conditional complexity)

def split_pdf_with_page_numbers(input_path, output_path, pages):
    pdf_document = fitz.open(input_path)
    output_pdf = fitz.open()
    
    for page_num in pages:
        page = pdf_document.load_page(page_num )
        # 원본 페이지 번호 추가
        page_text = "Original Page: {}".format(page_num)
        rect = page.rect
        point = fitz.Point(rect.width - 100, rect.height - 20)
        page.insert_text(point, page_text, fontsize=12, color=(0, 0, 0))
        output_pdf.insert_pdf(pdf_document, from_page=page_num, to_page=page_num)
    
    output_pdf.save(output_path)
    output_pdf.close()
    pdf_document.close()