Главная » Языки программирования

Reading zip files in python

На чтение 9 мин
Содержание
  1. How to read text files in a zipped folder in Python
  2. 3 Answers 3
  3. Reading zip files in python
  4. Создание и закрытие файла
  5. Запись файлов в архив
  6. Получение информации о файлах в архиве
  7. Извлечение файлов из архива
  8. Считывание файла
  9. Открытие файла

How to read text files in a zipped folder in Python

I have a compressed data file (all in a folder, then zipped). I want to read each file without unzipping. I tried several methods but nothing works for entering the folder in the zip file. How should I achieve that? Without folder in the zip file:

with zipfile.ZipFile('data.zip') as z: for filename in z.namelist(): data = filename.readlines() 
with zipfile.ZipFile('data.zip') as z: for filename in z.namelist(): if filename.endswith('/'): # Here is what I was stucked

3 Answers 3

namelist() returns a list of all items in an archive recursively.

You can check whether an item is a directory by calling os.path.isdir():

import os import zipfile with zipfile.ZipFile('archive.zip') as z: for filename in z.namelist(): if not os.path.isdir(filename): # read the file with z.open(filename) as f: for line in f: print line

I believe os.path.isdir is meant to check if a path is a directory in the local file system. Here, it is used on a path from inside a zip file — is that really guarantueed to work?

I got Alec’s code to work. I made some minor edits: (note, this won’t work with password-protected zipfiles)

import os import sys import zipfile z = zipfile.ZipFile(sys.argv[1]) # Flexibility with regard to zipfile for filename in z.namelist(): if not os.path.isdir(filename): # read the file for line in z.open(filename): print line z.close() # Close the file after opening it del z # Cleanup (in case there's further work after this)

I got RichS’ code to work. I made some minor edits:

import os import sys import zipfile archive = sys.argv[1] # assuming launched with `python my_script.py archive.zip` with zipfile.ZipFile(archive) as z: for filename in z.namelist(): if not os.path.isdir(filename): # read the file for line in z.open(filename): print(line.decode('utf-8'))

As you can see the edits are minor. I’ve switched to Python 3, the ZipFile class has a capital F, and the output is converted from b-strings to unicode strings. Only decode if you are trying to unzip a text file.

PS I’m not dissing RichS at all. I just thought it would be hilarious. Both useful and a mild shitpost. PPS You can get file from an archive with a password: ZipFile.open(name, mode=’r’, pwd=None, *, force_zip64=False) or ZipFile.read(name, pwd=None) . If you use .read then there’s no context manager so you would simply do

 # read the file print(z.read(filename).decode('utf-8'))

Источник

Reading zip files in python

Zip представляет наиболее популярный формат архивации и сжатия файлов. И язык Python имеет встроенный модуль для работы с ними — zipfile . С помощью этого модуля можно создавать, считывать, записывать zip-файлы, получать их содержимое и добавлять в них файлы. Также поддерживается шифрование, но не поддерживается дешифрование.

Читайте также:  Opencv use gpu python

Для представления zip-файла в этом модуле определен класс ZipFile . Он имеет следующий конструктор:

ZipFile(file, mode='r', compression=ZIP_STORED, allowZip64=True, compresslevel=None, *, strict_timestamps=True, metadata_encoding=None)
  • file : путь к zip-файлу
  • mode : режим открытия файла. Может принимать следующие значения:
    • r : применяется для чтения существующего файла
    • w : применяется для записи нового файла
    • a : применяется для добавления в файл
    • ZIP_STORED : архивация без сжатия (значение по умолчанию)
    • ZIP_DEFLATED : стандартный тип сжатия при архивации в zip
    • ZIP_BZIP2 : сжатие с помощью способа BZIP2
    • ZIP_LZMA : сжатие с помощью способа LZMA

    Для работы с файлами этот класс предоставляет ряд методов:

    • close() : закрывает zip-файл
    • getinfo() : возвращает информацию об одном файле из архива в виде объекта ZipInfo
    • namelist() : возвращает список файлов архива
    • infolist() : возвращает информацию обо всех файлах из архива в виде списока объектов ZipInfo
    • open() : предоставляет доступ к одному из файлов в архиве
    • read() : считывает файл из архива в набор байтов
    • extract() : извлекает из архива один файл
    • extractall() : извлекает все элементы из архива
    • setpassword() : устанавливает пароль для zip-файла
    • printdir() : выводит на консоль содержимое архива

    Создание и закрытие файла

    Для создания архивного файла в конструктор ZipFile передается режим «w» или «a»:

    from zipfile import ZipFile myzip = ZipFile("metanit.zip", "w")

    После выполнения кода в текущей папке будет создаваться пустой архивный файл «metanit.zip».

    После окончания работы с архивом для его закрытия применяется метод close() :

    from zipfile import ZipFile myzip = ZipFile("metanit.zip", "w") myzip.close()

    Но так как ZipFile также представляет менеджер контекста, то он поддерживает выражение with , которое определяет контекст и автоматически закрывает файл по завершению контекста:

    from zipfile import ZipFile with ZipFile("metanit.zip", "w") as myzip: pass

    Запись файлов в архив

    Для записи файлов в архив применяется файл write() :

    write(filename, arcname=None, compress_type=None, compresslevel=None)

    Первый параметр представляет файл, который записиывается в архив. Второй параметр — arcname устанавливает произвольное имя для файла внутри архива (по умолчанию это само имя файла). Третий параметр — compress_type представляет тип сжатия, а параметр compresslevel — уровень сжатия.

    Например, запишем в архив «metanit.zip» файл «hello.txt» (который, как предполагается, находится в той же папке, где и текущий скрипт python):

    from zipfile import ZipFile with ZipFile("metanit.zip", "w") as myzip: myzip.write("hello.txt")

    Стоит учитывать, что при открытии файла в режиме «w» при всех последующих записях текущее содержимое будет затираться, то есть фактически архивный файл будет создаваться заново. Если нам необходимо добавить, то необходимо определять zip-файл в режиме «a»:

    from zipfile import ZipFile with ZipFile("metanit.zip", "a") as myzip: myzip.write("hello2.txt") myzip.write("forest.jpg")

    Стоит отметить, что по умолчанию сжатие не применяется. Но при необходимости можно применить какой-нибудь способ сжатия и уровень сжатия»

    from zipfile import ZipFile, ZIP_DEFLATED with ZipFile("metanit.zip", "w", compression=ZIP_DEFLATED, compresslevel=3) as myzip: myzip.write("hello.txt")

    Необходимо учитывать, что если мы попробуем добавить в архив файлы с уже имеющимися именами, то консоль выведет предупреждение. Чтобы избежать наличия файлов с дублирующимися именами можно через второй папаметр метода write явным образом определить для них уникальное имя внутри архива:

    from zipfile import ZipFile with ZipFile("metanit.zip", "a") as myzip: myzip.write("hello.txt", "hello1.txt") myzip.write("hello.txt", "hello2.txt") myzip.write("hello.txt", "hello3.txt")

    Получение информации о файлах в архиве

    Метод infolist() возвращает информацию о файлах в архиве с виде списка, где каждый отдельный файл представлен объектом ZipInfo:

    from zipfile import ZipFile with ZipFile("metanit.zip", "a") as myzip: print(myzip.infolist())

    Класс ZipInfo предоставляет ряд атрибутов для хранения информации о файле. Основные из них:

    • filename : название файла
    • date_time : дата и время последнего изменения файла в виде кортежа в формате (год, месяц, день, час, минута, секунда)
    • compress_type : тип сжатия
    • compress_size : размер после сжатия
    • file_size : оригинальный размер файла до сжатия

    Получим эти данные по каждому отдельному файлу в архиве:

    from zipfile import ZipFile with ZipFile("metanit.zip", "r") as myzip: for item in myzip.infolist(): print(f"File Name: Date: Size: ")

    Примерный консольный вывод:

    File Name: hello.txt Date: (2022, 11, 23, 20, 21, 34) Size: 18 File Name: forest.jpg Date: (2022, 11, 19, 20, 46, 52) Size: 103956 File Name: hello1.txt Date: (2022, 11, 23, 20, 21, 34) Size: 18 File Name: hello2.txt Date: (2022, 11, 23, 20, 21, 34) Size: 18 File Name: hello3.txt Date: (2022, 11, 23, 20, 21, 34) Size: 18

    С помощью метода is_dir() можно проверить, является ли элемент в архиве папкой:

    from zipfile import ZipFile with ZipFile("metanit.zip", "r") as myzip: for item in myzip.infolist(): if(item.is_dir()): print(f"Папка: ") else: print(f"Файл: ")

    Если надо получить только список имен входящих в архив файлов, то применяется метод namelist() :

    from zipfile import ZipFile with ZipFile("metanit.zip", "r") as myzip: for item in myzip.namelist(): print(item)

    Консольный вывод в моем случае:

    hello.txt forest.jpg hello1.txt hello2.txt hello3.txt

    С помощью метода getinfo() можно получить данные по одному из архивированных файлов, передав в метод его имя в архиве. Результат метода — объект ZipInfo:

    from zipfile import ZipFile with ZipFile("metanit.zip", "r") as myzip: try: hello_file = myzip.getinfo("hello.txt") print(hello_file.file_size) except KeyError: print("Указанный файл отсутствует")

    Если в архиве не окажется элемента с указанным именем, то метод сгенерирует ошибку KeyError.

    Извлечение файлов из архива

    Для извлечения всех файлов из архива применяется метод extractall() :

    extractall(path=None, members=None, pwd=None)

    Первый параметр метода устанавливает каталог для извлечения архива (по умолчанию извлечение идет в текущий каталог). Параметр members представляет список строк — список названий файлов, которые надо извлечт из архива. И третий параметр — pwd представляет пароль, в случае если архив закрыт паролем.

    Например, извлечем все файлы из архива:

    from zipfile import ZipFile with ZipFile("metanit.zip", "r") as myzip: myzip.extractall()

    Извлечение в определенную папку:

    myzip.extractall(path="metanit")
    # извлекаем файлы "hello.txt", "forest.jpg" в папку "metanit2" myzip.extractall(path="metanit2", members=["hello.txt", "forest.jpg"])

    Для извлечения одного файла применяется метод extract() , в который в качестве обязательного параметра передается имя извлекаемого файла:

    Считывание файла

    Метод read() позволяет считать содержимое файла из архива в набор байтов:

    from zipfile import ZipFile with ZipFile("metanit.zip", "r") as myzip: content = myzip.read("hello5.txt") print(content)

    Открытие файла

    Метод open() позволяет открывать отдельные файлы из архива без непосредственного их извлечения:

    open(name, mode='r', pwd=None, *, force_zip64=False)

    В качестве первого обязательного параметра передается имя файла внутри архива. Второй параметр — mode устанавливает режим открытия. Параметр pwd задает пароль, если файл защищен паролем. И параметр force_zip64 при значении True позволяет открывать файлы больше 4 Гб.

    Этот файл может быть полезен для манипулирования файлом, например, для считывания его содержимого или, наоборот, для записи в него. Например, откроем файл и считаем его содержимое:

    from zipfile import ZipFile with ZipFile("metanit.zip", "a") as myzip: # записываем в архив новый файл "hello5.txt" with myzip.open("hello5.txt", "w") as hello_file: encoded_str = bytes("Python. ", "UTF-8") hello_file.write(encoded_str)

    Источник

Оцените статью
© 2023 Программирование