【Python】PDFファイルを分割するツール

2024-08-10 2024-08-10

tasu

1 はじめに
2 コード
3 使い方

はじめに

PythonでPDFを分割するツールをご紹介します。

コード

入力ディレクトリ存在する全てのPDFファイルを取得し、出力ディレクトリに１ページずつ分割したPDFファイルを作成します。

import os
import datetime
import glob
import pypdf

INPUT_DIR = "pdf-split-input"
OUTPUT_DIR = "pdf-split-output"


def main():
    split()


def split():
    # 入力ディレクトリ作成
    input_dir = INPUT_DIR
    if not os.path.isdir(input_dir):
        os.makedirs(input_dir)

    # 出力ディレクトリ作成
    output_dir = OUTPUT_DIR
    if not os.path.isdir(output_dir):
        os.makedirs(output_dir)

    # 入力ファイル一覧取得
    input_files = glob.glob(os.path.join(input_dir, "*"))
    input_files = list(filter(lambda x: os.path.isfile(x), input_files))
    input_files = sorted(input_files)

    # 入力PDFを１ファイルずつ分割
    for input_file in input_files:

        # 入力PDFオープン
        reader = pypdf.PdfReader(input_file)
        print(f"input_file:{input_file}")

        # ページ分割
        for i_page in range(len(reader.pages)):
            # 出力ファイル名作成
            filename = os.path.basename(input_file)
            name, ext = os.path.splitext(filename)
            output_file = os.path.join(output_dir, f"{name}_{i_page}{ext}")

            # １ページずつPDFファイル作成
            writer = pypdf.PdfWriter()
            writer.add_page(reader.pages[i_page])
            writer.write(output_file)
            writer.close()
            print(f"output_file:{output_file}")

        # 入力PDFクローズ
        reader.close()


if __name__ == "__main__":
    main()