Скрипт 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/* .