Сценарій Python для вирівнювання видобутого дерева
Сценарій, написаний внизу, витягує zip-файл і переміщує файли, що містяться в найвищому каталозі, з нього в поточний робочий каталог. Цей швидкий скрипт розроблений відповідно до цього конкретного питання, коли є один єдиний верхній каталог, який містить усі файли, хоча з кількома правками може бути зроблений придатний для більш загальних випадків.
#!/usr/bin/env python3
import sys
import os
from zipfile import PyZipFile
for zip_file in sys.argv[1:]:
pzf = PyZipFile(zip_file)
namelist=pzf.namelist()
top_dir = namelist[0]
pzf.extractall(members=namelist[1:])
for item in namelist[1:]:
rename_args = [item,os.path.basename(item)]
print(rename_args)
os.rename(*rename_args)
os.rmdir(top_dir)
Тестовий запуск
Ось приклад того, як повинен працювати сценарій. Все, що дістається до поточного робочого каталогу, але вихідний файл взагалі може бути в різній директорії. Тест виконується в zip-архіві мого особистого сховища github.
$ ls
flatten_zip.py* master.zip
$ ./flatten_zip.py master.zip
['utc-time-indicator-master/.gitignore', '.gitignore']
['utc-time-indicator-master/LICENSE', 'LICENSE']
['utc-time-indicator-master/utc-time-indicator', 'utc-time-indicator']
['utc-time-indicator-master/utc_indicator.png', 'utc_indicator.png']
$ ls
flatten_zip.py* LICENSE master.zip utc_indicator.png utc-time-indicator
Перевірте, чи вихідний файл знаходиться в іншому місці
$ mkdir test_unzip
$ cd test_unzip
$ ../flatten_zip.py ../master.zip
['utc-time-indicator-master/.gitignore', '.gitignore']
['utc-time-indicator-master/LICENSE', 'LICENSE']
['utc-time-indicator-master/utc-time-indicator', 'utc-time-indicator']
['utc-time-indicator-master/utc_indicator.png', 'utc_indicator.png']
$ ls
LICENSE utc_indicator.png utc-time-indicator
unzip archive.zip && mv Release/* .