Class ZipFileEx

pex/common.py:320–497

A ZipFile that works around several issues in the stdlib. See: + https://bugs.python.org/issue15795 + https://github.com/pex-tool/pex/issues/298

class ZipFileEx(ZipFile):
    """A ZipFile that works around several issues in the stdlib.

    See:
    + https://bugs.python.org/issue15795
    + https://github.com/pex-tool/pex/issues/298
    """

    class ZipEntry(namedtuple("ZipEntry", ["info", "data"])):
        pass

    @classmethod
    def zip_info_from_file(
        cls,
        filename,  # type: Text
        arcname=None,  # type: Optional[Text]
        date_time=None,  # type: Optional[time.struct_time]
        file_mode=None,  # type: Optional[int]
        compress=True,  # type: bool
    ):
        # type: (...) -> Tuple[ZipInfo, bool]
        """Construct a ZipInfo for a file on the filesystem.

        Usually a similar `from_file` method is provided by `ZipInfo`, but it is not implemented in
         Python 2.7 or Python 3.5; so we re-implement it here adding the possibility to control the
         `ZipInfo` date_time separately from the underlying file mtime.
         See https://github.com/python/cpython/blob/9d3b53c47fab9ebf1f40d6f21b7d1ad391c14cd7/Lib/zipfile/__init__.py#L607-L642
        """
        st = os.stat(filename)
        is_dir = stat.S_ISDIR(st.st_mode)
        if arcname is None:
            arcname = filename
        arcname = os.path.normpath(os.path.splitdrive(arcname)[1])
        while arcname[0] in (os.sep, os.altsep):
            arcname = arcname[1:]
        if is_dir:
            arcname += "/"
        if date_time is None:
            date_time = time.localtime(st.st_mtime)
        zip_info = zipfile.ZipInfo(filename=arcname, date_time=date_time[:6])

        # Unix attributes
        zip_info.external_attr = ((file_mode or st.st_mode) & 0xFFFF) << 16

        if is_dir:
            zip_info.file_size = 0
            zip_info.external_attr |= 0x10  # MS-DOS directory flag
            zip_info.compress_type = zipfile.ZIP_STORED
        else:
            zip_info.file_size = st.st_size
            zip_info.compress_type = zipfile.ZIP_DEFLATED if compress else zipfile.ZIP_STORED
        return zip_info, is_dir

    def write_deterministic(
        self,
        filename,  # type: Text
        arcname=None,  # type: Optional[Text]
        digest=None,  # type: Optional[Digest]