Python zipファイルの中にあるフォルダのリネームがしたい

Question

実現したいこと

Pythonを使って、zipファイルの中にあるフォルダに対して解凍前、もしくは解凍直後にそのフォルダの名前を変えたい

前提

Aというzipファイルの中にBというフォルダがあり、そのBの中に画像ファイルがあるとします。
Bの名前はわからない、という状態のときにzipファイルの解凍前後にBを指定して名前の変更を行う方法はあるのか、というところが気になっています。
zipfileモジュールを利用して解凍まではできているのですが、名前の不定なフォルダに対してどうすれば名前の変更を行えるのかがわからないです。

A.zip
|ーB
ー|ー001～～.jpg
ー|ー002～～.jpg
ー|ー003～～.jpg

該当のソースコード

zip = A.zip

with zipfile.ZipFile(zip) as z:
    for info in z.infolist():
        info.filename = info.filename.encode('cp437').decode('cp932')
        z.extract(info, 'testdir')

現在ファイルの解凍に使っているコードです。

試したこと

pathlibもos.renameも使えないかとやってみたのですが、名前を変えるにはファイルパスの指定が必要であるため、わたしの知識ではフォルダの名前が不定なものに使うのは少し厳しかったです。

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

VScode上でPythonを動かしています。
Bの名前の変更はできなくても解凍した後にBの中身を別のフォルダに取り出す方法などがあればぜひとも教えていただきたいです。

Answer 1

リネームしたいフォルダが「B」といった１個だけなら以下の流れでよいかと思います。

• そのまま解凍
• 解凍先からフォルダを探す
• 見つかったらフォルダ名をリネームして終了

import zipfile
import os
import glob

ext_dir = 'testdir'
new_name = 'hoge'

with zipfile.ZipFile('A.zip') as z:
    for info in z.infolist():
        info.filename = info.filename.encode('cp437').decode('cp932')
        z.extract(info, ext_dir)

for path in glob.glob(f'{ext_dir}/*'):
    if os.path.isdir(path):
        os.rename(path, os.path.join(ext_dir, new_name))
        break

Comments

[shinyspark]

無事に思い通りの動きをさせることができました、とても勉強になります。
ありがとうございました。

Answer 2

BAついていますし、他の方のとあまり変らない方法ですが、質問にも出てくるinfo.filenameは展開するときのパスなので、これを加工すれば任意の場所に展開できます。

with zipfile.ZipFile(zip) as z:
    for info in z.infolist():
        # パスを取得
        fname = info.filename 
        # pathlibを使ってパスを分解
        parts = list(pathlib.Path(fname).parts)
        # 先頭を任意の名前に変更
        parts[0] = 'new_dir'
        # 変更後の名前を生成して info.filenameを上書き
        info.filename = '/'.join(parts)

        z.extract(info, 'testdir')

ここではpathlib.Path.parts を使いましたが、info.filenameは文字列なので、好きな方法で加工すればいいでしょう。

Comments

[shinyspark]

なるほど・・・そういう仕組みなんですね。
とても勉強になります、ありがとうございました。

Answer 3

os.path.basenameを使えばいいのではないでしょうか。
階層が2以上のファイルがある場合は文字列処理(str.split)で最上位のディレクトリを除去するなどですかね。

下記で、testdir以下にファイルが直接展開されると思います。

import zipfile
import os.path

zipname = 'A.zip'

with zipfile.ZipFile(zipname) as z:
    for info in z.infolist():
        if not info.is_dir():
            info.filename = os.path.basename(info.filename.encode('cp437').decode('cp932'))
            # info.filename = info.filename.encode('cp437').decode('cp932').split('/', 1)[-1]
            z.extract(info, 'testdir')

B(不明)を hoge に変えて展開したいとかなら、z.extract(info, 'testdir/hoge') にすればいいです。

Comments

[shinyspark]

osモジュールは本当にいろいろとあるんですね・・・
とても勉強になりました、ありがとうございました。