【第29回PYTHON講座】Zipファイル：ファイル形式の違いでファイルを削除

こんにちは！

今回は、PYTHONを使って、Zipファイル内で圧縮されているファイルの中でファイル形式が違うファイルを削除するプログラムを作ってみたいと思います。

プログラムの内容
ライブラリについて『zipfile』、『shutil』
プログラム
まとめ

プログラムの内容

ZIPファイル内に複数のJPGファイルに圧縮されています。しかし、その中で、TXTファイルが紛れているので、TXTファイルを削除します。また、ZIPのファイル名の変更はしないようにし、削除処理がされたかをコメント出力します。

今回は、下記のzipファイルを使用します。rabbit.zipというファイルの中にjpgファイルが格納されていますが、TXTファイルが複数紛れています。JPGファイルを残し、TXTファイルを削除します。

zipファイル内

ライブラリについて『zipfile』、『shutil』

使用するライブラリは、『zipfile』、『shutil』の2つになります。

『zipfile』は、zipファイルの操作をする為に使用します。

『shutil』は、ファイルの置き換えをする時に使用します。

zipfileのライブラリがインストールされていなければ、下記のコマンドでインストールをして下さい。

#zipfileをインストール
pip install zipfile

プログラム

import zipfile
from shutil import move


ofile = 'C:/Labo/trial/rabbit.zip'
bfile = 'C:/Labo/trial/bfr.zip'
fextension = 'jpg'
tsign = 0 
dfile = list()


zip1 = zipfile.ZipFile (ofile, 'r')


for item in zip1.infolist():
    if item.filename.endswith(fextension) == 0:
        tsign =1


if tsign == 1:
    zip2 = zipfile.ZipFile (bfile, 'w')

    for item in zip1.infolist():
        buffer = zip1.read(item.filename)
        if item.filename.endswith(fextension) == 1:
            zip2.writestr(item, buffer)
        else :
            dfile.append(item.filename)

    zip2.close()


zip1.close()


if tsign == 1:
    move(bfile, ofile)
    print('deleted files are ')
    print(dfile)
else :
    print(ofile + '内で削除するファイルはありませんでした')

１、２行目：ライブラリを宣言します。

5、６行目：ofileはオリジナルのファイルです。bfileは一時ファイルです。小さいファイルを取り除いたzipファイルをbfileとして作成します。プログラムの後半で、ofileに置き換える(move)ことによって、bfileは無くなります。

７行目：ZIPファイル内で残したいファイル形式jpgを指定します。

８行目：tsignは処理（ファイル削除）がされたかどうかを確認するフラグになります。０の場合は処理されず、１の場合は処理されたことになります。

９行目：削除されたファイル名を格納するリストdfileを作成します。

１２行目：zipファイルを読み込む関数です。rはread(読む）になります。

１５から１７行目：infolistでzip内の個々のファイルをitemに入れていきます。

itemにfilenameを付けることにより、拡張子を含めたファイル名を取得することが出来ます。

endswithは文字列の後半の文字検索の関数になります（前半の文字検索はstartswtich）。JPGファイル以外が存在するかをNOT判定でチェックし、存在する場合はtsignを1に設定します。

２０から３３行目：削除するファイルがある場合の処理。bfileのzipファイルを書き換えモードwで作成します。

先ほどと同様にファイル名の文字検索をし、JPGファイルのみをbufferに格納し、writestrで、bfileに追記していきます。追加しないファイル（削除するファイル）はdfileに格納します。

処理が終わればclose()でzipファイルを閉じてください。