Python:ディレクトリ内のPDFに別のPDFを挿入したい。

Question

挿入前フォルダ内にある各PDFファイルの3ページ目に、「A].PDFを挿入して
ファイル名を挿入前と同じファイル名で保存したい。
連続処理を行いたいのですが、エラーとなってしまい悩んでおります。

ご教授いただければ幸いです。

import glob
import PyPDF2

files = glob.glob(r'C:\PDF挿入\挿入前*.pdf')
merger = PyPDF2.PdfFileMerger(strict=False)

for i in files:
    merger.append(i)
    merger.merge(2, 'C:\PDF挿入\A.pdf')
    
    merger.write(r'C:\PDF挿入\挿入後****.pdf') #←挿入前と同じPDFファイル名で保存させたい。
    merger.close()

エラー内容は下記となります。
addpageの引数がエラーとなっています。
ファイルが一つしかない場合は、この引数にファイルパスを入力すればよいのですが、
ディレクトリ内の複数あるPDFにはエラーとなってしまうようです。

保存先に関しては、C:\PDF挿入\挿入後フォルダに挿入前と同じファイル名で保存したいのですが、その方法が分からないため※※※※で表記してあります。

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-3-72ca3ec7b122> in <module>
      3     merger.merge(2, 'C:\PDF挿入\A.pdf')
      4 
----> 5     merger.write(r'C:\PDF挿入\挿入後\000.pdf')
      6     merger.close()
      7 

~\anaconda3\lib\site-packages\PyPDF2\merger.py in write(self, fileobj)
    218         # The commented out line below was replaced with the two lines below it to allow PdfFileMerger to work with PyPdf 1.13
    219         for page in self.pages:
--> 220             self.output.addPage(page.pagedata)
    221             page.out_pagedata = self.output.getReference(self.output._pages.getObject()["/Kids"][-1].getObject())
    222             #idnum = self.output._objects.index(self.output._pages.getObject()["/Kids"][-1].getObject()) + 1

AttributeError: 'NoneType' object has no attribute 'addPage'

下記のコードで元のPDFを挿入後、元のファイル名で保存することに成功しましたが、成功するのは最初のPDFのみで、
ディレクトリ内にある2番目以降のPDFは上手く処理できませんでした。

import glob
import PyPDF2

files = glob.glob(r'C:\PDF挿入\挿入前*.pdf')
merger = PyPDF2.PdfFileMerger(strict=False)

for i in files:
    merger.append(i)
    merger.merge(2, 'C:\PDF挿入\A.pdf')

    file_name = r'C:\PDF挿入\挿入後' + "/" + os.path.basename(i)
    
    merger.write(file_name)
    merger.close()

エラーは下記です。解決方法が分からず悩んでおります。

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-3-6deca51f3d5a> in <module>
     12 
     13 
---> 14     merger.write(file_name)
     15     merger.close()
     16 

~\anaconda3\lib\site-packages\PyPDF2\merger.py in write(self, fileobj)
    218         # The commented out line below was replaced with the two lines below it to allow PdfFileMerger to work with PyPdf 1.13
    219         for page in self.pages:
--> 220             self.output.addPage(page.pagedata)
    221             page.out_pagedata = self.output.getReference(self.output._pages.getObject()["/Kids"][-1].getObject())
    222             #idnum = self.output._objects.index(self.output._pages.getObject()["/Kids"][-1].getObject()) + 1

AttributeError: 'NoneType' object has no attribute 'addPage'

Comments

[AkiFuku]

実際のソースでも
merger.write(r'C:\PDF挿入\挿入後\****.pdf')
というコードで保存しようとしているのでしょうか？そうでない場合、実際のソースで書いているままにした方が良いかと思います。
また、どういったエラーが表示されるかも併記しておくと回答が得られやすいかと思います。

Answer 1

ファイルが一つの場合はうまくいって、merger.write(r'C:\PDF挿入\挿入後****.pdf')だと上手くいかないというのは、このファイル名の指定の方法が良くないのではと思います。
os.path.basename(i)
でパスの中からファイル名のみが取り出せますので、これをもとに以下のように保存先のパスを作成し、保存を実行すれば良さそうな気がします。

newname = r'C:\PDF挿入\挿入後' + "/" + os.path.basename(i)
merger.write(newname)

参考：「python パス　ファイル名のみ」などで検索するとこういった記事が見つかるかと思います。
https://note.nkmk.me/python-os-basename-dirname-split-splitext/

---

【追記】
merger = PyPDF2.PdfFileMerger(strict=False)
をfor文の前ではなく、for文の中で宣言してみてください。
for文の中でこれをClose()しているため、二度目ではClose()済みのmergerで実行しようとしてエラーになっているのだと思われます、

Comments

[monopoly_man]

解決出来ました！初心者のくだらないご質問にお付き合い頂き感謝しております。
ありがとうございます。