for文での複数PDFファイルの復号化

Question

実現したいこと

複数PDFファイルの復号化を自動化したい。for文を使い、以下のようなコードを作成し、実現しようとしています。

import PyPDF2

encrypted_files =[] #暗号化されたファイルのリスト
decrypted_files =[] #復号化されたファイルのリスト
for i in encrypted_files:
    with open(i, mode='rb') as f:        
        reader = PyPDF2.PdfFileReader(f)
        command=f"/usr/local/bin/qpdf  --decrypt {i} {i+'-DECRYPTED.pdf'};" #復号化の為にqpdfを使用
        os.system(command)            
        with open(i, mode='rb') as fp:
            reader = PyPDF2.PdfFileReader(fp)
            decrypted_files.append(i+'-DECRYPTED.pdf')
            encrypted_files.remove(i)

発生している問題

エラーメッセージは出ないのですが、encrypted_filesに残ってしまうファイルがいくつか出てきます。しかし、２度、３度同じコードを手動で回すと全て復号化できます。手動で繰り返し回すことなく可能なものは全て復号化するためにはどうすれば良いでしょうか？

PDFファイルの復号化に限らず、for文などの繰り返し処理で似たような経験や解決策をお知りの方がいらっしゃいましたらよろしくお願いいたします。

試したこと

sleepを入れて以下のようにコードを変えてみましたが、解決しませんでした。

import PyPDF2

for i in encrypted_files:
    with open(i, mode='rb') as f:        
        reader = PyPDF2.PdfFileReader(f)
        try: 
            reader.decrypt('')
            sleep(3)
            decrypted_files.append(i)
            encrypted_files.remove(i)
        except NotImplementedError:
            command=f"/usr/local/bin/qpdf  --decrypt {i} {i+'-DECRYPTED.pdf'};"
            os.system(command)            
            with open(i, mode='rb') as fp:
                reader = PyPDF2.PdfFileReader(fp)
                decrypted_files.append(i+'-DECRYPTED.pdf')
                encrypted_files.remove(i)
        sleep(3)

補足情報

本問題の本質と関連なさそうと思っていますが、qpdfについてはこちらが補足情報になるかと思います。

環境

• Python 3.7.3
• macOS 10.14.6

Answer 1

hoge = [1,2,3,4,5,6,7,8,9]

for i in hoge:
    print(i)
    hoge.remove(i)
print(hoge)

# 1
# 3
# 5
# 7
# 9
# [2, 4, 6, 8]

質問にあるコードではこんな感じのことが起こっています。
1回目のループでは1がiに入り、1が削除されます。この時、hoge の中は [2,3,4,5,6,7,8,9] になります。次のループでは2つ目の要素がiに入るのでiには3が入り、2が削除されず、3が削除されます。
こんなことがループの最後まで続くのでリストの中にいくつかの要素が残った状態になってしまいます。

対策としてはfor文に使うリストは hoge.copy()としてコピーしたリストを使ってあげるといいと思います。

8.3. for 文の注釈も読んでおくといいです。注釈の中ではスライスを使ってコピーを作っていますがcopy()でも同等の結果が得られます。

Comments

[otonashi]

ありがとうございます。実現したいことが実現できました。

Answer 2

下記コードでどうでしょうか？

import PyPDF2

encrypted_files =[] #暗号化されたファイルのリスト
decrypted_files =[] #復号化されたファイルのリスト
for i in encrypted_files:
    with open(i, mode='rb') as f:        
        reader = PyPDF2.PdfFileReader(f)
        command=f"/usr/local/bin/qpdf  --decrypt {i} {i+'-DECRYPTED.pdf'};" #復号化の為にqpdfを使用
        os.system(command)            
        
        decrypted_files.append(i+'-DECRYPTED.pdf')

１．encrypted_filesから要素を削除する必要はないと思います。※
２．2回目の「with open(i, mode='rb') as fp:
reader = PyPDF2.PdfFileReader(fp)」は不要だと思います。

※必要であれば復号化処理が終わったあとにdecrypted_filesとencrypted_filesを比較して削除すれば良いです

Comments

[otonashi]

ありがとうございます。確かに不要な部分があったので、削除してコードをスリムにできました。