Get uncompressed size of a .gz file in python

2019-01-27 17:46发布

Using gzip, tell() returns the offset in the uncompressed file.
In order to show a progress bar, I want to know the original (uncompressed) size of the file.
Is there an easy way to find out?

标签: python gzip
10条回答
Summer. ? 凉城
2楼-- · 2019-01-27 18:16

GzipFile.size stores the uncompressed size, but it's only incremented when you read the file, so you should prefer len(fd.read()) instead of the non-public GzipFile.size.

查看更多
虎瘦雄心在
3楼-- · 2019-01-27 18:17

I am not sure about performance, but this could be achieved without knowing gzip magic by using:

with gzip.open(filepath, 'rb') as file_obj:
    file_size = file_obj.seek(0, io.SEEK_END)

This should also work for other (compressed) stream readers like bz2 or the plain open.

EDIT: as suggested in the comments, 2 in second line was replaced by io.SEEK_END, which is definitely more readable and probably more future-proof.

EDIT: Works only in Python 3.

查看更多
Root(大扎)
4楼-- · 2019-01-27 18:21

Looking at the source for the gzip module, I see that the underlying file object for GzipFile seems to be fileobj. So:

mygzipfile = gzip.GzipFile()
...
mygzipfile.fileobj.tell()

?

Maybe it would be good to do some sanity checking before doing that, like checking that the attribute exists with hasattr.

Not exactly a public API, but...

查看更多
看我几分像从前
5楼-- · 2019-01-27 18:26

The last 4 bytes of the .gz hold the original size of the file

查看更多
手持菜刀,她持情操
6楼-- · 2019-01-27 18:28

Unix way: use "gunzip -l file.gz" via subprocess.call / os.popen, capture and parse its output.

查看更多
Animai°情兽
7楼-- · 2019-01-27 18:29

The gzip format specifies a field called ISIZE that:

This contains the size of the original (uncompressed) input data modulo 2^32.

In gzip.py, which I assume is what you're using for gzip support, there is a method called _read_eof defined as such:

def _read_eof(self):
    # We've read to the end of the file, so we have to rewind in order
    # to reread the 8 bytes containing the CRC and the file size.
    # We check the that the computed CRC and size of the
    # uncompressed data matches the stored values.  Note that the size
    # stored is the true file size mod 2**32.
    self.fileobj.seek(-8, 1)
    crc32 = read32(self.fileobj)
    isize = U32(read32(self.fileobj))   # may exceed 2GB
    if U32(crc32) != U32(self.crc):
        raise IOError, "CRC check failed"
    elif isize != LOWU32(self.size):
        raise IOError, "Incorrect length of data produced"

There you can see that the ISIZE field is being read, but only to to compare it to self.size for error detection. This then should mean that GzipFile.size stores the actual uncompressed size. However, I think it's not exposed publicly, so you might have to hack it in to expose it. Not so sure, sorry.

I just looked all of this up right now, and I haven't tried it so I could be wrong. I hope this is of some use to you. Sorry if I misunderstood your question.

查看更多
登录 后发表回答