ディレクトリ名ファイル名を指定したPDFの分割がしたいです。

Question

前提・実現したいこと

実現したいことは、pythonのtkinterを使ってpdfをページごとに分割する仕組みを作りたいです。
動きとしては「参照ボタン」を押してテキストボックスにpdfのフルパスを取得。←ここまでは完了
そのパスを使って対象となるpdfの分割をしたいです。
最後に保存する先を指定するダイアログを出して分割ファイルを保存したいです。

いろんなサイトからコピーをしてやっているのですが、上手くいきません。
コードの中で完結する、ディレクトリ名、ファイル名を指定した分割はできました。

まだ勉強時間が不足しているため、１つ１つの言葉やエラーの意味を理解して使っておりません。
良い案がある方、書き方にはこだわっておりませんので、お手数ですがご教授お願い致します。

発生している問題・エラーメッセージ

Exception in Tkinter callback
Traceback (most recent call last):
  File "C:\Users\user1\Anaconda3\lib\tkinter\__init__.py", line 1705, in __call__
    return self.func(*args)
TypeError: pdf_split() missing 2 required positional arguments: 'p_file' and 'p_folder'

該当のソースコード

##テキストボックス（ここに参照で取得したフルパスを入れる）
split_txt = tkinter.Entry(width=58, justify="left")
split_txt.place(x=85, y=210)

##参照ボタンの動作
def ask_file():
    fTyp = [("", "*")]
    pdf_dir = filedialog.askopenfilename(
        filetypes=fTyp)
    split_txt.insert(0, pdf_dir)

##メインの動作（PDF分割）
def pdf_split(p_file, p_folder):
    p_path = split_txt.get()
    pdf_dir, pdf_file = os.path.split(p_path)

    file = PdfFileReader(open(p_file + '.pdf', 'rb'))
    for page_num in range(file.numPages):
        page = file.getPage(page_num)
        file_output = PdfFileWriter()
        file_output.addPage(page)
        with open(p_folder + '\' + p_file + '_split_' + str(page_num) + '.pdf', 'wb') as f:
            file_output.write(f)
    pdf_split(pdf_file, pdf_dir)

    # メッセージボックス
    messagebox.showinfo("完了", "完了しました。")

##参照ボタン
btn = tkinter.Button(
    root, text='参照', command=ask_file)
btn.place(x=450, y=210)

##実行ボタン（pdf_splitの呼び出し）
btn = tkinter.Button(
    root, text='実行', command=pdf_split, width=10)
btn.place(x=210, y=400)

＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝
##以下は単体で動かせていた時点のソースコードです
　# 指定のフォルダへ指定のPDFファイルを分割する
　from pathlib import Path
　import PyPDF2
　from PyPDF2 import PdfFileReader
　from PyPDF2 import PdfFileWriter

　pdf_file = "pdfファイル名"
　pdf_dir = "./pdf_files"

　# PDFファイルをページごとに分割して保存


　def pdf_split(p_file, p_folder):
    　file = PdfFileReader(open(p_file + '.pdf', 'rb'))
    　for page_num in range(file.numPages):
        　page = file.getPage(page_num)
        　file_output = PdfFileWriter()
        　file_output.addPage(page)
        　with open(p_folder + '\' + p_file + '_split_' + str(page_num) + '.pdf', 'wb') as f:
            　file_output.write(f)


　# 保存
　pdf_split(pdf_file, pdf_dir)
＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝

試したこと

理解をしてやっていることは特にありません。

補足情報（FW/ツールのバージョンなど）

pythonのバージョンは3.7.6でvscodeのバージョンは1.52.0を使用してます。

Comments

[TakaiY]

1) pdf_split の中で pdf_split を再帰的に呼んでいるのは意図の通りですか？
このままでは無限ループか例外発生になりそうですが。

2) エラーは pdf_split を呼ぶところ、たぶんメインの処理側でpdf_splitの引数が足りないのが原因と思われますが、その部分のソースを提示ください。

[seatyun]

ご意見ありがとうございます。私用により返信遅くなり申し訳ありません。

1) は意図通りではありません。こちらを単体で動かすと分割が出来ていたので丸ごとコピペしております。

2) 引数が足りないとのことで、単体で動かせていたソースコードを「該当のソースコード」に追記させて頂きます。再度ご確認頂けたら有難いです。

[TakaiY]

たしかに、単体で動いていた方のコードは動きそうですね。

そうではなく、あなたが提示したエラーが発生したときの、ソースコードを提示してください。その中で、pdf_split関数を呼んでいるはずです。
問題が発生したとして提示されているコードには関数定義しか存在しません(その定義にも問題がありますが)ので、この関数をファイルに書いて実行しても何も起きないほずです。

[seatyun]

pdf_splitを呼び出すためのコードを追加しました。
これ以外でpdf_splitにの動作に関わるコードはありませんので、おっしゃっている提示するべきコードでなければ、私の力不足のため、そもそも入力しているコードに間違いや不足のコードがあると思われます。

Answer 1

こんな感じでしょうか。

まずは、エラーですが、発生しているのはここが原因です。

##実行ボタン（pdf_splitの呼び出し）
btn = tkinter.Button(
    root, text='実行', command=pdf_split, width=10)

このようにボタンに設定されていると、押されたときにpdf_split関数が引数無しで呼ばれます。 引数無しなのは、tkinterの仕様です。

呼ばれるpdf_split関数の方ですが、

def pdf_split(p_file, p_folder):
    p_path = split_txt.get()
    pdf_dir, pdf_file = os.path.split(p_path)  # ★2

    file = PdfFileReader(open(p_file + '.pdf', 'rb'))
    for page_num in range(file.numPages):
        page = file.getPage(page_num)
        file_output = PdfFileWriter()
        file_output.addPage(page)
        with open(p_folder + '\' + p_file + '_split_' + str(page_num) + '.pdf', 'wb') as f:
            file_output.write(f)
    pdf_split(pdf_file, pdf_dir)  # ★1

    # メッセージボックス
    messagebox.showinfo("完了", "完了しました。")

内部で取得したpdf_dir, pdf_file 情報(★2)を関数で使いたいということなのでしょうが、内部で関数の引数に入れて読んでいます(★1).この方法ではうまくいきません。

動くようにするには、内部で使っている変数を、取得したpdf_dir, pdf_fileで置き換えればいいのです。

こんな感じです。

def pdf_split():  # ▲1
    p_path = split_txt.get()
    pdf_dir, pdf_file = os.path.split(p_path)

    file = PdfFileReader(open(pdf_file + '.pdf', 'rb'))  # ▲2
    for page_num in range(file.numPages):
        page = file.getPage(page_num)
        file_output = PdfFileWriter()
        file_output.addPage(page)
        with open(pdf_dir + '\' + pdf_file + '_split_' + str(page_num) + '.pdf', 'wb') as f:  ▲3
            file_output.write(f)

▲4

▲1 関数の引数は削除。 必要なパスとファイル名は処理内で取得しているから。 また、tkinterで呼ぶのにも都合がいい。
▲2 出力ファイル名を、内部で取得したものに変更
▲3 ディレクトリ名とファイル名を内部で取得したものに変更
▲4 pdf_splitの呼び出しや、メッセージは削除。特にpdf_splitの呼び出しはここにあっては動きません。

この関数だけ修正すれば、とりあえず見えている問題は解決できるんではないかと思います。

Comments

[seatyun]

先頭のコードで記載の通り、取得したフルパスをテキストに貼り付けしており、「ファイル名.pdf」という表記になります。
このまま実行するとエラーになり、以下のエラーが出ます。
ーーーーーーーーーーーーーーーー
File "pdf_editor.py", line 109, in pdf_split
file = PdfFileReader(open(pdf_file + '.pdf', 'rb'))
FileNotFoundError: [Errno 2] No such file or directory: 'ファイル名.pdf.pdf'
ーーーーーーーーーーーーーーーー

テキストボックス内の「.pdf」の部分を手動で消せば分割できました。
ファイル名拡張子の表示のチェックが入ってなくても「.pdf」が付いてくるのですが、
何かいい方法はありますでしょうか。

[TakaiY]

パスとかファイル名とか拡張子とかはファイルを扱う上での基本なので、このあたりを見ていただければわかるのではないかと思います。
https://note.nkmk.me/python-os-basename-dirname-split-splitext/

拡張子を表示していなくても、拡張子が取得できるのは普通です。拡張子違いで複数ファイルあるなどが可能なので、ダイアログなどで拡張子を省いて取得することは普通できません。

[seatyun]

以下のように入れることでエラーが出ずに処理できました。
ーーーーーーーーーーーーーーー
pdf_dir, pdf_file = os.path.split(p_path)
pdf_file.rstrip('.pdf')
file = PdfFileReader(open(pdf_file, 'rb'))
ーーーーーーーーーーーーーーー
丁寧にご対応頂き、ありがとうございました。

[TakaiY]

うまくいったのはよかったですね。

'.pdf'限定ということで、rstripされたのだと思いますが、最後の拡張子を除くということであれば
pdf_file_name, pdf_file_suffix = os.path.splitext(pdf_file)
あたりが筋がいいと思います。あ

ちなみに、pytho3系でパスを扱うならos.pathよりpathlibのほうが使いやすいと思っています。

[seatyun]

ご指摘ありがとうございます。
試行錯誤して以下のように変更して動かすことが
出来たのですがどうでしょうか。
ーーーーーーーーーーーーーーー
pdf_dir, pdf_file = os.path.split(p_path)
pdf_file_name, pdf_file_suffix = os.path.splitext(pdf_file)
file = PdfFileReader(open(pdf_file_name + pdf_file_suffix, 'rb'))
ーーーーーーーーーーーーーーー

[TakaiY]

せっかくなので、ちゃんとしたいということであれば、ちょっと違いますね。
ここで、それぞれの変数には
pdf_dir : C:\Users\user1\xxx\xxx のようなディレクトリ
pdf_file : xxxxxx.pdf 拡張子付きのファイル名
pdf_file_name : xxxxxxx ファイル名から拡張子を除いたもの
pdf_file_suffix : .pdf 拡張子
となっています。なので ファイルをopenするときには、pdf_fileだけ使うのがきれいですね。
拡張子を取り除いたのは、出力時に
with open(p_folder + '\' + p_file + '_split_' + str(page_num) + '.pdf', 'wb') as f:
当初こうなっていたところで使うためです。

[seatyun]

pdf_file_suffixは分けるためで、プログラムに組み込む必要はないということでよろしいでしょうか。
以下の感じでどうですか？
ーーーーーーーーーーーーーーー
file = PdfFileReader(open(pdf_file, 'rb'))
for page_num in range(file.numPages):
page = file.getPage(page_num)
file_output = PdfFileWriter()
file_output.addPage(page)
with open(pdf_dir + '\' + pdf_file_name + '_split_' + str(page_num) + '.pdf', 'wb') as f:
file_output.write(f)
ーーーーーーーーーーーーーーー

[TakaiY]

いいと思います。

[seatyun]

ありがとうございます。とても助かりました。