"""

[0] Fix | Delete

Read and write ZIP files.

[1] Fix | Delete

[2] Fix | Delete

XXX references to utf-8 need further investigation.

[3] Fix | Delete

"""

[4] Fix | Delete

import binascii

[5] Fix | Delete

import importlib.util

[6] Fix | Delete

import io

[7] Fix | Delete

import itertools

[8] Fix | Delete

import os

[9] Fix | Delete

import posixpath

[10] Fix | Delete

import shutil

[11] Fix | Delete

import stat

[12] Fix | Delete

import struct

[13] Fix | Delete

import sys

[14] Fix | Delete

import threading

[15] Fix | Delete

import time

[16] Fix | Delete

import contextlib

[17] Fix | Delete

[18] Fix | Delete

try:

[19] Fix | Delete

import zlib # We may need its compression method

[20] Fix | Delete

crc32 = zlib.crc32

[21] Fix | Delete

except ImportError:

[22] Fix | Delete

zlib = None

[23] Fix | Delete

crc32 = binascii.crc32

[24] Fix | Delete

[25] Fix | Delete

try:

[26] Fix | Delete

import bz2 # We may need its compression method

[27] Fix | Delete

except ImportError:

[28] Fix | Delete

bz2 = None

[29] Fix | Delete

[30] Fix | Delete

try:

[31] Fix | Delete

import lzma # We may need its compression method

[32] Fix | Delete

except ImportError:

[33] Fix | Delete

lzma = None

[34] Fix | Delete

[35] Fix | Delete

__all__ = ["BadZipFile", "BadZipfile", "error",

[36] Fix | Delete

"ZIP_STORED", "ZIP_DEFLATED", "ZIP_BZIP2", "ZIP_LZMA",

[37] Fix | Delete

"is_zipfile", "ZipInfo", "ZipFile", "PyZipFile", "LargeZipFile",

[38] Fix | Delete

"Path"]

[39] Fix | Delete

[40] Fix | Delete

class BadZipFile(Exception):

[41] Fix | Delete

pass

[42] Fix | Delete

[43] Fix | Delete

[44] Fix | Delete

class LargeZipFile(Exception):

[45] Fix | Delete

"""

[46] Fix | Delete

Raised when writing a zipfile, the zipfile requires ZIP64 extensions

[47] Fix | Delete

and those extensions are disabled.

[48] Fix | Delete

"""

[49] Fix | Delete

[50] Fix | Delete

error = BadZipfile = BadZipFile # Pre-3.2 compatibility names

[51] Fix | Delete

[52] Fix | Delete

[53] Fix | Delete

ZIP64_LIMIT = (1 << 31) - 1

[54] Fix | Delete

ZIP_FILECOUNT_LIMIT = (1 << 16) - 1

[55] Fix | Delete

ZIP_MAX_COMMENT = (1 << 16) - 1

[56] Fix | Delete

[57] Fix | Delete

# constants for Zip file compression methods

[58] Fix | Delete

ZIP_STORED = 0

[59] Fix | Delete

ZIP_DEFLATED = 8

[60] Fix | Delete

ZIP_BZIP2 = 12

[61] Fix | Delete

ZIP_LZMA = 14

[62] Fix | Delete

# Other ZIP compression methods not supported

[63] Fix | Delete

[64] Fix | Delete

DEFAULT_VERSION = 20

[65] Fix | Delete

ZIP64_VERSION = 45

[66] Fix | Delete

BZIP2_VERSION = 46

[67] Fix | Delete

LZMA_VERSION = 63

[68] Fix | Delete

# we recognize (but not necessarily support) all features up to that version

[69] Fix | Delete

MAX_EXTRACT_VERSION = 63

[70] Fix | Delete

[71] Fix | Delete

# Below are some formats and associated data for reading/writing headers using

[72] Fix | Delete

# the struct module. The names and structures of headers/records are those used

[73] Fix | Delete

# in the PKWARE description of the ZIP file format:

[74] Fix | Delete

# http://www.pkware.com/documents/casestudies/APPNOTE.TXT

[75] Fix | Delete

# (URL valid as of January 2008)

[76] Fix | Delete

[77] Fix | Delete

# The "end of central directory" structure, magic number, size, and indices

[78] Fix | Delete

# (section V.I in the format document)

[79] Fix | Delete

structEndArchive = b"<4s4H2LH"

[80] Fix | Delete

stringEndArchive = b"PK\005\006"

[81] Fix | Delete

sizeEndCentDir = struct.calcsize(structEndArchive)

[82] Fix | Delete

[83] Fix | Delete

_ECD_SIGNATURE = 0

[84] Fix | Delete

_ECD_DISK_NUMBER = 1

[85] Fix | Delete

_ECD_DISK_START = 2

[86] Fix | Delete

_ECD_ENTRIES_THIS_DISK = 3

[87] Fix | Delete

_ECD_ENTRIES_TOTAL = 4

[88] Fix | Delete

_ECD_SIZE = 5

[89] Fix | Delete

_ECD_OFFSET = 6

[90] Fix | Delete

_ECD_COMMENT_SIZE = 7

[91] Fix | Delete

# These last two indices are not part of the structure as defined in the

[92] Fix | Delete

# spec, but they are used internally by this module as a convenience

[93] Fix | Delete

_ECD_COMMENT = 8

[94] Fix | Delete

_ECD_LOCATION = 9

[95] Fix | Delete

[96] Fix | Delete

# The "central directory" structure, magic number, size, and indices

[97] Fix | Delete

# of entries in the structure (section V.F in the format document)

[98] Fix | Delete

structCentralDir = "<4s4B4HL2L5H2L"

[99] Fix | Delete

stringCentralDir = b"PK\001\002"

[100] Fix | Delete

sizeCentralDir = struct.calcsize(structCentralDir)

[101] Fix | Delete

[102] Fix | Delete

# indexes of entries in the central directory structure

[103] Fix | Delete

_CD_SIGNATURE = 0

[104] Fix | Delete

_CD_CREATE_VERSION = 1

[105] Fix | Delete

_CD_CREATE_SYSTEM = 2

[106] Fix | Delete

_CD_EXTRACT_VERSION = 3

[107] Fix | Delete

_CD_EXTRACT_SYSTEM = 4

[108] Fix | Delete

_CD_FLAG_BITS = 5

[109] Fix | Delete

_CD_COMPRESS_TYPE = 6

[110] Fix | Delete

_CD_TIME = 7

[111] Fix | Delete

_CD_DATE = 8

[112] Fix | Delete

_CD_CRC = 9

[113] Fix | Delete

_CD_COMPRESSED_SIZE = 10

[114] Fix | Delete

_CD_UNCOMPRESSED_SIZE = 11

[115] Fix | Delete

_CD_FILENAME_LENGTH = 12

[116] Fix | Delete

_CD_EXTRA_FIELD_LENGTH = 13

[117] Fix | Delete

_CD_COMMENT_LENGTH = 14

[118] Fix | Delete

_CD_DISK_NUMBER_START = 15

[119] Fix | Delete

_CD_INTERNAL_FILE_ATTRIBUTES = 16

[120] Fix | Delete

_CD_EXTERNAL_FILE_ATTRIBUTES = 17

[121] Fix | Delete

_CD_LOCAL_HEADER_OFFSET = 18

[122] Fix | Delete

[123] Fix | Delete

# The "local file header" structure, magic number, size, and indices

[124] Fix | Delete

# (section V.A in the format document)

[125] Fix | Delete

structFileHeader = "<4s2B4HL2L2H"

[126] Fix | Delete

stringFileHeader = b"PK\003\004"

[127] Fix | Delete

sizeFileHeader = struct.calcsize(structFileHeader)

[128] Fix | Delete

[129] Fix | Delete

_FH_SIGNATURE = 0

[130] Fix | Delete

_FH_EXTRACT_VERSION = 1

[131] Fix | Delete

_FH_EXTRACT_SYSTEM = 2

[132] Fix | Delete

_FH_GENERAL_PURPOSE_FLAG_BITS = 3

[133] Fix | Delete

_FH_COMPRESSION_METHOD = 4

[134] Fix | Delete

_FH_LAST_MOD_TIME = 5

[135] Fix | Delete

_FH_LAST_MOD_DATE = 6

[136] Fix | Delete

_FH_CRC = 7

[137] Fix | Delete

_FH_COMPRESSED_SIZE = 8

[138] Fix | Delete

_FH_UNCOMPRESSED_SIZE = 9

[139] Fix | Delete

_FH_FILENAME_LENGTH = 10

[140] Fix | Delete

_FH_EXTRA_FIELD_LENGTH = 11

[141] Fix | Delete

[142] Fix | Delete

# The "Zip64 end of central directory locator" structure, magic number, and size

[143] Fix | Delete

structEndArchive64Locator = "<4sLQL"

[144] Fix | Delete

stringEndArchive64Locator = b"PK\x06\x07"

[145] Fix | Delete

sizeEndCentDir64Locator = struct.calcsize(structEndArchive64Locator)

[146] Fix | Delete

[147] Fix | Delete

# The "Zip64 end of central directory" record, magic number, size, and indices

[148] Fix | Delete

# (section V.G in the format document)

[149] Fix | Delete

structEndArchive64 = "<4sQ2H2L4Q"

[150] Fix | Delete

stringEndArchive64 = b"PK\x06\x06"

[151] Fix | Delete

sizeEndCentDir64 = struct.calcsize(structEndArchive64)

[152] Fix | Delete

[153] Fix | Delete

_CD64_SIGNATURE = 0

[154] Fix | Delete

_CD64_DIRECTORY_RECSIZE = 1

[155] Fix | Delete

_CD64_CREATE_VERSION = 2

[156] Fix | Delete

_CD64_EXTRACT_VERSION = 3

[157] Fix | Delete

_CD64_DISK_NUMBER = 4

[158] Fix | Delete

_CD64_DISK_NUMBER_START = 5

[159] Fix | Delete

_CD64_NUMBER_ENTRIES_THIS_DISK = 6

[160] Fix | Delete

_CD64_NUMBER_ENTRIES_TOTAL = 7

[161] Fix | Delete

_CD64_DIRECTORY_SIZE = 8

[162] Fix | Delete

_CD64_OFFSET_START_CENTDIR = 9

[163] Fix | Delete

[164] Fix | Delete

_DD_SIGNATURE = 0x08074b50

[165] Fix | Delete

[166] Fix | Delete

_EXTRA_FIELD_STRUCT = struct.Struct('<HH')

[167] Fix | Delete

[168] Fix | Delete

def _strip_extra(extra, xids):

[169] Fix | Delete

# Remove Extra Fields with specified IDs.

[170] Fix | Delete

unpack = _EXTRA_FIELD_STRUCT.unpack

[171] Fix | Delete

modified = False

[172] Fix | Delete

buffer = []

[173] Fix | Delete

start = i = 0

[174] Fix | Delete

while i + 4 <= len(extra):

[175] Fix | Delete

xid, xlen = unpack(extra[i : i + 4])

[176] Fix | Delete

j = i + 4 + xlen

[177] Fix | Delete

if xid in xids:

[178] Fix | Delete

if i != start:

[179] Fix | Delete

buffer.append(extra[start : i])

[180] Fix | Delete

start = j

[181] Fix | Delete

modified = True

[182] Fix | Delete

i = j

[183] Fix | Delete

if not modified:

[184] Fix | Delete

return extra

[185] Fix | Delete

return b''.join(buffer)

[186] Fix | Delete

[187] Fix | Delete

def _check_zipfile(fp):

[188] Fix | Delete

try:

[189] Fix | Delete

if _EndRecData(fp):

[190] Fix | Delete

return True # file has correct magic number

[191] Fix | Delete

except OSError:

[192] Fix | Delete

pass

[193] Fix | Delete

return False

[194] Fix | Delete

[195] Fix | Delete

def is_zipfile(filename):

[196] Fix | Delete

"""Quickly see if a file is a ZIP file by checking the magic number.

[197] Fix | Delete

[198] Fix | Delete

The filename argument may be a file or file-like object too.

[199] Fix | Delete

"""

[200] Fix | Delete

result = False

[201] Fix | Delete

try:

[202] Fix | Delete

if hasattr(filename, "read"):

[203] Fix | Delete

result = _check_zipfile(fp=filename)

[204] Fix | Delete

else:

[205] Fix | Delete

with open(filename, "rb") as fp:

[206] Fix | Delete

result = _check_zipfile(fp)

[207] Fix | Delete

except OSError:

[208] Fix | Delete

pass

[209] Fix | Delete

return result

[210] Fix | Delete

[211] Fix | Delete

def _EndRecData64(fpin, offset, endrec):

[212] Fix | Delete

"""

[213] Fix | Delete

Read the ZIP64 end-of-archive records and use that to update endrec

[214] Fix | Delete

"""

[215] Fix | Delete

try:

[216] Fix | Delete

fpin.seek(offset - sizeEndCentDir64Locator, 2)

[217] Fix | Delete

except OSError:

[218] Fix | Delete

# If the seek fails, the file is not large enough to contain a ZIP64

[219] Fix | Delete

# end-of-archive record, so just return the end record we were given.

[220] Fix | Delete

return endrec

[221] Fix | Delete

[222] Fix | Delete

data = fpin.read(sizeEndCentDir64Locator)

[223] Fix | Delete

if len(data) != sizeEndCentDir64Locator:

[224] Fix | Delete

return endrec

[225] Fix | Delete

sig, diskno, reloff, disks = struct.unpack(structEndArchive64Locator, data)

[226] Fix | Delete

if sig != stringEndArchive64Locator:

[227] Fix | Delete

return endrec

[228] Fix | Delete

[229] Fix | Delete

if diskno != 0 or disks > 1:

[230] Fix | Delete

raise BadZipFile("zipfiles that span multiple disks are not supported")

[231] Fix | Delete

[232] Fix | Delete

# Assume no 'zip64 extensible data'

[233] Fix | Delete

fpin.seek(offset - sizeEndCentDir64Locator - sizeEndCentDir64, 2)

[234] Fix | Delete

data = fpin.read(sizeEndCentDir64)

[235] Fix | Delete

if len(data) != sizeEndCentDir64:

[236] Fix | Delete

return endrec

[237] Fix | Delete

sig, sz, create_version, read_version, disk_num, disk_dir, \

[238] Fix | Delete

dircount, dircount2, dirsize, diroffset = \

[239] Fix | Delete

struct.unpack(structEndArchive64, data)

[240] Fix | Delete

if sig != stringEndArchive64:

[241] Fix | Delete

return endrec

[242] Fix | Delete

[243] Fix | Delete

# Update the original endrec using data from the ZIP64 record

[244] Fix | Delete

endrec[_ECD_SIGNATURE] = sig

[245] Fix | Delete

endrec[_ECD_DISK_NUMBER] = disk_num

[246] Fix | Delete

endrec[_ECD_DISK_START] = disk_dir

[247] Fix | Delete

endrec[_ECD_ENTRIES_THIS_DISK] = dircount

[248] Fix | Delete

endrec[_ECD_ENTRIES_TOTAL] = dircount2

[249] Fix | Delete

endrec[_ECD_SIZE] = dirsize

[250] Fix | Delete

endrec[_ECD_OFFSET] = diroffset

[251] Fix | Delete

return endrec

[252] Fix | Delete

[253] Fix | Delete

[254] Fix | Delete

def _EndRecData(fpin):

[255] Fix | Delete

"""Return data from the "End of Central Directory" record, or None.

[256] Fix | Delete

[257] Fix | Delete

The data is a list of the nine items in the ZIP "End of central dir"

[258] Fix | Delete

record followed by a tenth item, the file seek offset of this record."""

[259] Fix | Delete

[260] Fix | Delete

# Determine file size

[261] Fix | Delete

fpin.seek(0, 2)

[262] Fix | Delete

filesize = fpin.tell()

[263] Fix | Delete

[264] Fix | Delete

# Check to see if this is ZIP file with no archive comment (the

[265] Fix | Delete

# "end of central directory" structure should be the last item in the

[266] Fix | Delete

# file if this is the case).

[267] Fix | Delete

try:

[268] Fix | Delete

fpin.seek(-sizeEndCentDir, 2)

[269] Fix | Delete

except OSError:

[270] Fix | Delete

return None

[271] Fix | Delete

data = fpin.read()

[272] Fix | Delete

if (len(data) == sizeEndCentDir and

[273] Fix | Delete

data[0:4] == stringEndArchive and

[274] Fix | Delete

data[-2:] == b"\000\000"):

[275] Fix | Delete

# the signature is correct and there's no comment, unpack structure

[276] Fix | Delete

endrec = struct.unpack(structEndArchive, data)

[277] Fix | Delete

endrec=list(endrec)

[278] Fix | Delete

[279] Fix | Delete

# Append a blank comment and record start offset

[280] Fix | Delete

endrec.append(b"")

[281] Fix | Delete

endrec.append(filesize - sizeEndCentDir)

[282] Fix | Delete

[283] Fix | Delete

# Try to read the "Zip64 end of central directory" structure

[284] Fix | Delete

return _EndRecData64(fpin, -sizeEndCentDir, endrec)

[285] Fix | Delete

[286] Fix | Delete

# Either this is not a ZIP file, or it is a ZIP file with an archive

[287] Fix | Delete

# comment. Search the end of the file for the "end of central directory"

[288] Fix | Delete

# record signature. The comment is the last item in the ZIP file and may be

[289] Fix | Delete

# up to 64K long. It is assumed that the "end of central directory" magic

[290] Fix | Delete

# number does not appear in the comment.

[291] Fix | Delete

maxCommentStart = max(filesize - (1 << 16) - sizeEndCentDir, 0)

[292] Fix | Delete

fpin.seek(maxCommentStart, 0)

[293] Fix | Delete

data = fpin.read()

[294] Fix | Delete

start = data.rfind(stringEndArchive)

[295] Fix | Delete

if start >= 0:

[296] Fix | Delete

# found the magic number; attempt to unpack and interpret

[297] Fix | Delete

recData = data[start:start+sizeEndCentDir]

[298] Fix | Delete

if len(recData) != sizeEndCentDir:

[299] Fix | Delete

# Zip file is corrupted.

[300] Fix | Delete

return None

[301] Fix | Delete

endrec = list(struct.unpack(structEndArchive, recData))

[302] Fix | Delete

commentSize = endrec[_ECD_COMMENT_SIZE] #as claimed by the zip file

[303] Fix | Delete

comment = data[start+sizeEndCentDir:start+sizeEndCentDir+commentSize]

[304] Fix | Delete

endrec.append(comment)

[305] Fix | Delete

endrec.append(maxCommentStart + start)

[306] Fix | Delete

[307] Fix | Delete

# Try to read the "Zip64 end of central directory" structure

[308] Fix | Delete

return _EndRecData64(fpin, maxCommentStart + start - filesize,

[309] Fix | Delete

endrec)

[310] Fix | Delete

[311] Fix | Delete

# Unable to find a valid end of central directory structure

[312] Fix | Delete

return None

[313] Fix | Delete

[314] Fix | Delete

[315] Fix | Delete

class ZipInfo (object):

[316] Fix | Delete

"""Class with attributes describing each file in the ZIP archive."""

[317] Fix | Delete

[318] Fix | Delete

__slots__ = (

[319] Fix | Delete

'orig_filename',

[320] Fix | Delete

'filename',

[321] Fix | Delete

'date_time',

[322] Fix | Delete

'compress_type',

[323] Fix | Delete

'_compresslevel',

[324] Fix | Delete

'comment',

[325] Fix | Delete

'extra',

[326] Fix | Delete

'create_system',

[327] Fix | Delete

'create_version',

[328] Fix | Delete

'extract_version',

[329] Fix | Delete

'reserved',

[330] Fix | Delete

'flag_bits',

[331] Fix | Delete

'volume',

[332] Fix | Delete

'internal_attr',

[333] Fix | Delete

'external_attr',

[334] Fix | Delete

'header_offset',

[335] Fix | Delete

'CRC',

[336] Fix | Delete

'compress_size',

[337] Fix | Delete

'file_size',

[338] Fix | Delete

'_raw_time',

[339] Fix | Delete

)

[340] Fix | Delete

[341] Fix | Delete

def __init__(self, filename="NoName", date_time=(1980,1,1,0,0,0)):

[342] Fix | Delete

self.orig_filename = filename # Original file name in archive

[343] Fix | Delete

[344] Fix | Delete

# Terminate the file name at the first null byte. Null bytes in file

[345] Fix | Delete

# names are used as tricks by viruses in archives.

[346] Fix | Delete

null_byte = filename.find(chr(0))

[347] Fix | Delete

if null_byte >= 0:

[348] Fix | Delete

filename = filename[0:null_byte]

[349] Fix | Delete

# This is used to ensure paths in generated ZIP files always use

[350] Fix | Delete

# forward slashes as the directory separator, as required by the

[351] Fix | Delete

# ZIP format specification.

[352] Fix | Delete

if os.sep != "/" and os.sep in filename:

[353] Fix | Delete

filename = filename.replace(os.sep, "/")

[354] Fix | Delete

[355] Fix | Delete

self.filename = filename # Normalized file name

[356] Fix | Delete

self.date_time = date_time # year, month, day, hour, min, sec

[357] Fix | Delete

[358] Fix | Delete

if date_time[0] < 1980:

[359] Fix | Delete

raise ValueError('ZIP does not support timestamps before 1980')

[360] Fix | Delete

[361] Fix | Delete

# Standard values:

[362] Fix | Delete

self.compress_type = ZIP_STORED # Type of compression for the file

[363] Fix | Delete

self._compresslevel = None # Level for the compressor

[364] Fix | Delete

self.comment = b"" # Comment for each file

[365] Fix | Delete

self.extra = b"" # ZIP extra data

[366] Fix | Delete

if sys.platform == 'win32':

[367] Fix | Delete

self.create_system = 0 # System which created ZIP archive

[368] Fix | Delete

else:

[369] Fix | Delete

# Assume everything else is unix-y

[370] Fix | Delete

self.create_system = 3 # System which created ZIP archive

[371] Fix | Delete

self.create_version = DEFAULT_VERSION # Version which created ZIP archive

[372] Fix | Delete

self.extract_version = DEFAULT_VERSION # Version needed to extract archive

[373] Fix | Delete

self.reserved = 0 # Must be zero

[374] Fix | Delete

self.flag_bits = 0 # ZIP flag bits

[375] Fix | Delete

self.volume = 0 # Volume number of file header

[376] Fix | Delete

self.internal_attr = 0 # Internal attributes

[377] Fix | Delete

self.external_attr = 0 # External file attributes

[378] Fix | Delete

self.compress_size = 0 # Size of the compressed file

[379] Fix | Delete

self.file_size = 0 # Size of the uncompressed file

[380] Fix | Delete

# Other attributes are set by class ZipFile:

[381] Fix | Delete

# header_offset Byte offset to the file header

[382] Fix | Delete

# CRC CRC-32 of the uncompressed file

[383] Fix | Delete

[384] Fix | Delete

def __repr__(self):

[385] Fix | Delete

result = ['<%s filename=%r' % (self.__class__.__name__, self.filename)]

[386] Fix | Delete

if self.compress_type != ZIP_STORED:

[387] Fix | Delete

result.append(' compress_type=%s' %

[388] Fix | Delete

compressor_names.get(self.compress_type,

[389] Fix | Delete

self.compress_type))

[390] Fix | Delete

hi = self.external_attr >> 16

[391] Fix | Delete

lo = self.external_attr & 0xFFFF

[392] Fix | Delete

if hi:

[393] Fix | Delete

result.append(' filemode=%r' % stat.filemode(hi))

[394] Fix | Delete

if lo:

[395] Fix | Delete

result.append(' external_attr=%#x' % lo)

[396] Fix | Delete

isdir = self.is_dir()

[397] Fix | Delete

if not isdir or self.file_size:

[398] Fix | Delete

result.append(' file_size=%r' % self.file_size)

[399] Fix | Delete

if ((not isdir or self.compress_size) and

[400] Fix | Delete

(self.compress_type != ZIP_STORED or

[401] Fix | Delete

self.file_size != self.compress_size)):

[402] Fix | Delete

result.append(' compress_size=%r' % self.compress_size)

[403] Fix | Delete

result.append('>')

[404] Fix | Delete

return ''.join(result)

[405] Fix | Delete

[406] Fix | Delete

def FileHeader(self, zip64=None):

[407] Fix | Delete

"""Return the per-file header as a bytes object."""

[408] Fix | Delete

dt = self.date_time

[409] Fix | Delete

dosdate = (dt[0] - 1980) << 9 | dt[1] << 5 | dt[2]

[410] Fix | Delete

dostime = dt[3] << 11 | dt[4] << 5 | (dt[5] // 2)

[411] Fix | Delete

if self.flag_bits & 0x08:

[412] Fix | Delete

# Set these to zero because we write them after the file data

[413] Fix | Delete

CRC = compress_size = file_size = 0

[414] Fix | Delete

else:

[415] Fix | Delete

CRC = self.CRC

[416] Fix | Delete

compress_size = self.compress_size

[417] Fix | Delete

file_size = self.file_size

[418] Fix | Delete

[419] Fix | Delete

extra = self.extra

[420] Fix | Delete

[421] Fix | Delete

min_version = 0

[422] Fix | Delete

if zip64 is None:

[423] Fix | Delete

zip64 = file_size > ZIP64_LIMIT or compress_size > ZIP64_LIMIT

[424] Fix | Delete

if zip64:

[425] Fix | Delete

fmt = '<HHQQ'

[426] Fix | Delete

extra = extra + struct.pack(fmt,

[427] Fix | Delete

1, struct.calcsize(fmt)-4, file_size, compress_size)

[428] Fix | Delete

if file_size > ZIP64_LIMIT or compress_size > ZIP64_LIMIT:

[429] Fix | Delete

if not zip64:

[430] Fix | Delete

raise LargeZipFile("Filesize would require ZIP64 extensions")

[431] Fix | Delete

# File is larger than what fits into a 4 byte integer,

[432] Fix | Delete

# fall back to the ZIP64 extension

[433] Fix | Delete

file_size = 0xffffffff

[434] Fix | Delete

compress_size = 0xffffffff

[435] Fix | Delete

min_version = ZIP64_VERSION

[436] Fix | Delete

[437] Fix | Delete

if self.compress_type == ZIP_BZIP2:

[438] Fix | Delete

min_version = max(BZIP2_VERSION, min_version)

[439] Fix | Delete

elif self.compress_type == ZIP_LZMA:

[440] Fix | Delete

min_version = max(LZMA_VERSION, min_version)

[441] Fix | Delete

[442] Fix | Delete

self.extract_version = max(min_version, self.extract_version)

[443] Fix | Delete

self.create_version = max(min_version, self.create_version)

[444] Fix | Delete

filename, flag_bits = self._encodeFilenameFlags()

[445] Fix | Delete

header = struct.pack(structFileHeader, stringFileHeader,

[446] Fix | Delete

self.extract_version, self.reserved, flag_bits,

[447] Fix | Delete

self.compress_type, dostime, dosdate, CRC,

[448] Fix | Delete

compress_size, file_size,

[449] Fix | Delete

len(filename), len(extra))

[450] Fix | Delete

return header + filename + extra

[451] Fix | Delete

[452] Fix | Delete

def _encodeFilenameFlags(self):

[453] Fix | Delete

try:

[454] Fix | Delete

return self.filename.encode('ascii'), self.flag_bits

[455] Fix | Delete

except UnicodeEncodeError:

[456] Fix | Delete

return self.filename.encode('utf-8'), self.flag_bits | 0x800

[457] Fix | Delete

[458] Fix | Delete

def _decodeExtra(self):

[459] Fix | Delete

# Try to decode the extra field.

[460] Fix | Delete

extra = self.extra

[461] Fix | Delete

unpack = struct.unpack

[462] Fix | Delete

while len(extra) >= 4:

[463] Fix | Delete

tp, ln = unpack('<HH', extra[:4])

[464] Fix | Delete

if ln+4 > len(extra):

[465] Fix | Delete

raise BadZipFile("Corrupt extra field %04x (size=%d)" % (tp, ln))

[466] Fix | Delete

if tp == 0x0001:

[467] Fix | Delete

data = extra[4:ln+4]

[468] Fix | Delete

# ZIP64 extension (large files and/or large archives)

[469] Fix | Delete

try:

[470] Fix | Delete

if self.file_size in (0xFFFF_FFFF_FFFF_FFFF, 0xFFFF_FFFF):

[471] Fix | Delete

field = "File size"

[472] Fix | Delete

self.file_size, = unpack('<Q', data[:8])

[473] Fix | Delete

data = data[8:]

[474] Fix | Delete

if self.compress_size == 0xFFFF_FFFF:

[475] Fix | Delete

field = "Compress size"

[476] Fix | Delete

self.compress_size, = unpack('<Q', data[:8])

[477] Fix | Delete

data = data[8:]

[478] Fix | Delete

if self.header_offset == 0xFFFF_FFFF:

[479] Fix | Delete

field = "Header offset"

[480] Fix | Delete

self.header_offset, = unpack('<Q', data[:8])

[481] Fix | Delete

except struct.error:

[482] Fix | Delete

raise BadZipFile(f"Corrupt zip64 extra field. "

[483] Fix | Delete

f"{field} not found.") from None

[484] Fix | Delete

[485] Fix | Delete

extra = extra[ln+4:]

[486] Fix | Delete

[487] Fix | Delete

@classmethod

[488] Fix | Delete

def from_file(cls, filename, arcname=None, *, strict_timestamps=True):

[489] Fix | Delete

"""Construct an appropriate ZipInfo for a file on the filesystem.

[490] Fix | Delete

[491] Fix | Delete

filename should be the path to a file or directory on the filesystem.

[492] Fix | Delete

[493] Fix | Delete

arcname is the name which it will have within the archive (by default,

[494] Fix | Delete

this will be the same as filename, but without a drive letter and with

[495] Fix | Delete

leading path separators removed).

[496] Fix | Delete

"""

[497] Fix | Delete

if isinstance(filename, os.PathLike):

[498] Fix | Delete

filename = os.fspath(filename)

[499] Fix | Delete