o Q`k;@sdZddlmZddlmZddlZddlZddlmZddlZddlm Z ddl m Z dd l m Z dd lmZmZdd lmZdd lmZdd lmZddlmZddlmZddlmZmZmZmZddl m!Z!ddl"m#Z#ddl$m%Z%ej&rddlm'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0ddlm1Z1ddl2m3Z3ej4dddZ5GdddeZ6Gddde#Z7ej8Gddde#Z9ej8Gd dde Z:dS)!z(Manage the filesystem in a Zip archive. )print_function)unicode_literalsN)datetime)errors)FS) write_zip) ResourceTypeSeek)Info) RawWrapper) Permissions)MemoryFS)open_fs)dirnameforcedirnormpathrelpath)datetime_to_epoch)WrapFS) url_quote) AnyBinaryIO CollectionDictListOptional SupportsIntTextTupleUnion)RawInfo)SubFSR ReadZipFS)ZboundcsFeZdZfddZd ddZd ddZejfdd Zd d Z Z S) _ZipExtFilecs>|j|_}||j|_d|_tt|||d|dS)Nrr) _zipgetinfo file_size_end_possuperr%__init__open)selffsnamer' __class__*/usr/lib/python3/dist-packages/fs/zipfs.pyr-/s z_ZipExtFile.__init__cC.|j|dur dn|}|jt|7_|SNr6)_freadr+lenr/sizebufr4r4r5r:6z_ZipExtFile.readcCr7r8)r9read1r+r;r<r4r4r5r@<r?z_ZipExtFile.read1cCst|}|tjkr||j7}|tjks|tjkr$|dkr#td|n#|tjkr:|dkr4td|||j7}n td|tjtjtj||jkrW|j |j |_ d|_| ||j|jS)aChange stream position. Change the stream position to the given byte offset. The offset is interpreted relative to the position indicated by ``whence``. Arguments: offset (int): the offset to the new position, in bytes. whence (int): the position reference. Possible values are: * `Seek.set`: start of stream (the default). * `Seek.current`: current position; offset may be negative. * `Seek.end`: end of stream; offset must be negative. Returns: int: the new absolute position. Raises: ValueError: when ``whence`` is not known, or ``offset`` is invalid. Note: Zip compression does not support seeking, so the seeking is emulated. Seeking somewhere else than the current position will need to either: * reopen the file and restart decompression * read and discard data to advance in the file rzNegative seek position {}zPositive seek position {}z+Invalid whence ({}, should be {}, {} or {}))intr Zcurrentr+set ValueErrorformatendr*r'r.r1r9r:)r/offsetwhenceZ_whencer4r4r5seekBs*     z_ZipExtFile.seekcC|jSN)r+r/r4r4r5tellwz_ZipExtFile.tell)r6) __name__ __module__ __qualname__r-r:r@r rBrHrL __classcell__r4r4r2r5r%.s    5r%c@sBeZdZdZdejddfddZejrdejddfddZ d Sd S) ZipFSaRead and write zip files. There are two ways to open a ZipFS for the use cases of reading a zip file, and creating a new one. If you open the ZipFS with ``write`` set to `False` (the default) then the filesystem will be a read only filesystem which maps to the files and directories within the zip file. Files are decompressed on the fly when you open them. Here's how you might extract and print a readme from a zip file:: with ZipFS('foo.zip') as zip_fs: readme = zip_fs.readtext('readme.txt') If you open the ZipFS with ``write`` set to `True`, then the ZipFS will be a empty temporary filesystem. Any files / directories you create in the ZipFS will be written in to a zip file when the ZipFS is closed. Here's how you might write a new zip file containing a readme.txt file:: with ZipFS('foo.zip', write=True) as new_zip: new_zip.writetext( 'readme.txt', 'This zip file was written by PyFilesystem' ) Arguments: file (str or io.IOBase): An OS filename, or an open file object. write (bool): Set to `True` to write a new zip file, or `False` (default) to read an existing zip file. compression (int): Compression to use (one of the constants defined in the `zipfile` module in the stdlib). temp_fs (str): An FS URL for the temporary filesystem used to store data prior to zipping. Futf-8temp://__ziptemp__cCs |r t||||dSt||dS)N) compressionencodingtemp_fs)rV) WriteZipFSr$)clsfilewriterUrVrWr4r4r5__new__s   z ZipFS.__new__cCsdSrJr4)r/rZr[rUrVrWr4r4r5r-s zZipFS.__init__N) rNrOrP__doc__zipfile ZIP_DEFLATEDr\typing TYPE_CHECKINGr-r4r4r4r5rR|s- rRcsfeZdZdZejddffdd ZddZdd Zd d Z d d Z fddZ   dddZ Z S)rXzA writable zip file. rSrTcsH||_||_||_||_t||_t|j|_t t | |jdSrJ) _filerUrV _temp_fs_urlr_temp_fsdictZgetmeta_metar,rXr-)r/rZrUrVrWr2r4r5r-s zWriteZipFS.__init__cCsd}||j|j|j|jS)Nz?WriteZipFS({!r}, compression={!r}, encoding={!r}, temp_fs={!r}))rDrbrUrVrc)r/tr4r4r5__repr__szWriteZipFS.__repr__cC d|jS)NzrDrbrKr4r4r5__str__ zWriteZipFS.__str__cCs |j|fSrJrdr/pathr4r4r5 delegate_paths zWriteZipFS.delegate_pathcCrIrJrmrKr4r4r5 delegate_fsrMzWriteZipFS.delegate_fscs>|sz |W|jn|jwtt|dSrJ)isclosedrrdcloser,rXrKr2r4r5rss  zWriteZipFS.closeNcCs4|st|j|p |j|p|j|p|jddSdS)a5Write zip to a file. Arguments: file (str or io.IOBase, optional): Destination file, may be a file name or an open file handle. compression (int, optional): Compression to use (one of the constants defined in the `zipfile` module in the stdlib). encoding (str, optional): The character encoding to use (default uses the encoding defined in `~WriteZipFS.__init__`). Note: This is called automatically when the ZipFS is closed. )rUrVN)rrrrdrbrUrV)r/rZrUrVr4r4r5rs zWriteZipFS.write_zip)NNN)rNrOrPr]r^r_r-rhrkrprqrsrrQr4r4r2r5rXs  rXcseZdZdZddddddddZejjd(fdd Zdd Z d d Z d d Z e ddZ d)ddZddZddZ  d*ddZd+ddZddZdd Zfd!d"Zd#d$Zd,d&d'ZZS)-r$zA readable zip file. TF)Zcase_insensitiveZnetworkZ read_onlyZsupports_renameZ thread_safeZ unicode_pathsZvirtualrScs2tt|||_||_t|d|_d|_dS)Nr&) r,r$r-rbrVr^ZipFiler' _directory_fs)r/rZrVr2r4r5r-$s  zReadZipFS.__init__cCri)NzReadZipFS({!r})rjrKr4r4r5rh-rlzReadZipFS.__repr__cCri)Nz rjrKr4r4r5rk1rlzReadZipFS.__str__cCs6tt|}|j|rt|}tjr||jS|S)z+Convert a path to a zip file name. ) rr _directoryisdirrsixPY2encoderVrnr4r4r5_path_to_zip_name5s   zReadZipFS._path_to_zip_namecCs||jE|jdurBt|_}|jD])}|}tjr&||j d}| dr3|j |ddq|j t |dd| |q|jWdS1sOwYdS)zL`MemoryFS`: a filesystem with the same folder hierarchy as the zip. Nreplace/T)recreate)check_lockrurr'ZnamelistrxrydecoderVendswithmakedirsrZcreate)r/Z_fszip_nameZ resource_namer4r4r5rv@s    $zReadZipFS._directoryNcsD||}|pd}i}|dkr'ddd|d<d|vr#dttji|d<t|S|j|}|j|jd|d<hd |s| |}z|j |Wn t yVYt|Swd|vrrj t|jrdtjntj ttjd |d<d |vrfd d jD|d <d|vrjrjdkrdtjd?d@di|d<t|S)Nr4r}T)r1is_dirZbasicdetailstype>zipaccessr)r=rZmodifiedrcs"i|] }|ds|t|qS)_) startswithgetattr).0kZzip_infor4r5 xs  z%ReadZipFS.getinfo..r permissionsi)mode)Z validatepathrAr Z directoryrvr(r1r isdisjointr{r'KeyErrorr)rZrrZ date_time __slots__Z external_attrZ create_systemr dumpr )r/roZ namespaces_pathZraw_infoZ basic_inforr4rr5r(TsR )       zReadZipFS.getinfocC|t|rJrrResourceReadOnly)r/roinfor4r4r5setinfo zReadZipFS.setinfocCs||j|SrJ)rrvlistdirrnr4r4r5rs zReadZipFS.listdircCrrJr)r/rorr~r4r4r5makedirs zReadZipFS.makedirr&r6cKsj|d|vsd|vsd|vrt||j|s t||j|r+t|||}t ||S)Nw+a) rrrrvexistsResourceNotFoundrwZ FileExpectedr{r%)r/ror bufferingkwargsrr4r4r5openbins       zReadZipFS.openbincCrrJrrnr4r4r5removerzReadZipFS.removecCrrJrrnr4r4r5 removedirrzReadZipFS.removedircs*tt|t|dr|jdSdS)Nr')r,r$rshasattrr'rKr2r4r5rss zReadZipFS.closecCs8||j|st|||}|j|}|SrJ)rrvisfilerrr{r'r:)r/rorZ zip_bytesr4r4r5 readbytess     zReadZipFS.readbytesdownloadcCs@|dkrt|jtjrt|j}t|}d||St||)Nr0z zip://{}!/{}) isinstancerbrxZ string_typesrrDrZNoURL)r/roZpurposeZ quoted_fileZ quoted_pathr4r4r5geturls    zReadZipFS.geturl)rSrJ)NF)r&r6)r)rNrOrPr]rfrZ CreateFailedZ catch_allr-rhrkr{propertyrvr(rrrrrrrsrrrQr4r4r2r5r$s:   4   );r]Z __future__rrr`r^rrxrrbasercompressrZenumsr r rr Ziotoolsr rr ZmemoryfsrZopenerrrorrrrtimerZwrapfsrZ _url_toolsrrarrrrrrrrrr r!Zsubfsr"TypeVarr#r%rRZpython_2_unicode_compatiblerXr$r4r4r4r5s>              0 NKM