o Q`+<@sdZddlmZddlmZddlZddlZddlZddlmZddlm Z m Z ddl Z ddl m Z dd lmZdd lmZdd lmZdd l mZmZdd lmZddlmZddlmZddlmZddlmZddl m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&ddl'm(Z(ej)rddlm*Z*ddlm+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3ddlm4Z4ddl5m6Z6ej7dddZ8gdZ9e j:rddZ;nddZ;Gdd d e(Ze j=Gd#ddeZ?e@d$krRdd%lAmBZBet ||||dSt ||dS)Nname) compressionr.temp_fs)r.) isinstancesix text_type binary_typeospath expandusergetattrZ iteritems_compression_formatsendswithr)r') clsfilewriter:r.r;filenamecomp extensionsr/r/r0__new__ss      z TarFS.__new__cCdSr*r/)selfrGrHr:r.r;r/r/r0__init__s zTarFS.__init__)FNr6r7) __name__ __module__ __qualname____doc__rDrLtyping TYPE_CHECKINGrOr/r/r/r0r(Bs$+  r(csdeZdZdZ   dfdd ZddZd d Zd d Zd dZfddZ   dddZ Z S)r)zA writable tar file. Nr6r7csH||_||_||_||_t||_t|j|_t t | |jdSr*) _filer:r. _temp_fs_urlr_temp_fsdictZgetmeta_metasuperr)rO)rNrGr:r.r; __class__r/r0rOs zWriteTarFS.__init__cCsd}||j|j|j|jS)Nz?WriteTarFS({!r}, compression={!r}, encoding={!r}, temp_fs={!r}))formatrVr:r.rW)rNtr/r/r0__repr__szWriteTarFS.__repr__cC d|jS)Nzr^rVrNr/r/r0__str__r2zWriteTarFS.__str__cCs |j|fSr*rXrNrAr/r/r0 delegate_paths zWriteTarFS.delegate_pathcCs|jSr*rercr/r/r0 delegate_fsszWriteTarFS.delegate_fscs>|sz |W|jn|jwtt|dSr*)isclosedr rXcloser[r)rcr\r/r0rjs  zWriteTarFS.closecCs4|st|j|p |j|p|j|p|jddSdS)a)Write tar to a file. Arguments: file (str or io.IOBase, optional): Destination file, may be a file name or an open file object. compression (str, optional): Compression to use (one of the constants defined in `tarfile` in the stdlib). encoding (str, optional): The character encoding to use (default uses the encoding defined in `~WriteTarFS.__init__`). Note: This is called automatically when the TarFS is closed. )r:r.N)rir rXrVr:r.)rNrGr:r.r/r/r0r s zWriteTarFS.write_tar)Nr6r7)NNN) rPrQrRrSrOr`rdrgrhrjr __classcell__r/r/r\r0r)s  r)csBeZdZdZddddddddZejejej ej ej ej ej ejejejejejejejejejejeji ZZejjd0fdd Zedd Zd d Zd d Ze j!r]ddZ"ddZ#nddZ"ddZ#d1ddZ$ddZ%ddZ&ddZ'ddZ(  d2dd Z)d3d#d$Z*d%d&Z+d'd(Z,fd)d*Z-d+d,Z.d4d.d/Z/Z0S)5r'zA readable tar file. TF)Zcase_insensitiveZnetworkZ read_onlyZsupports_renameZ thread_safeZ unicode_pathsZvirtualr6csXtt|||_||_t|tjtjfrt j |dd|_ nt j |dd|_ d|_ dS)Nrmode)Zfileobjrn) r[r'rOrVr.r<r=r>r?tarfileopen_tar_directory_cache)rNrGr.r\r/r0rOs zReadTarFS.__init__csB|jdur|jfdd|jDfdd}t||_|jS)zLazy directory cache.Nc3s$|] }|jd|fVqdS)/N)r8strip).0info)_decoder/r0 s z/ReadTarFS._directory_entries..c 3sBD]\}}zt|}Wn tyYqw|r||fVqdSr*)rr )r8rv_name)_directory_entriesr/r0 _list_tars    z/ReadTarFS._directory_entries.._list_tar)rrrwrqr)rNr{r/)rwrzr0rzs    zReadTarFS._directory_entriescCra)NzReadTarFS({!r})rbrcr/r/r0r`*r2zReadTarFS.__repr__cCra)Nz rbrcr/r/r0rd.r2zReadTarFS.__str__cC ||jSr*)encoder.rNsr/r/r0_encode4r2zReadTarFS._encodecCr|r*)decoder.r~r/r/r0rw8r2zReadTarFS._decodecC|Sr*r/r~r/r/r0r>cCrr*r/r~r/r/r0rwBrNcsjt||}|p d}i}|s'ddd|d<d|vr#dttji|d<t|Sz d}|j|WntyN||sAt |d}t |t j _ Ynwt|jd|d<d|vryjt|jj d |d<|syj|dd <d |vr|sjjtjd jjd |d <d|vr|st|j|d<|dfddtDt|S)Nr/r9T)r8is_dirZbasicZdetailstypeF)sizerZmodifiedaccessrm)gidgroup permissionsuidusertarcs,i|]}|dr|ddt|qS)isis_) startswithreplacerC)rukr-r/r0 ss z%ReadTarFS.getinfo..) r validatepathintr directoryrzKeyErrorisdirrResourceNotFoundrorDIRTYPErrrwr8rtype_mapmtimergnamerrndumprunamer1r.updatedirr)rNrAZ namespaces_pathZraw_infoZimplicitr/rr0getinfoFsT+          zReadTarFS.getinfocsJt||z|jWSty$tfdd|jDYSw)Nc3s|]}t|VqdSr*)r)rur8rr/r0rxsz"ReadTarFS.isdir..)rrrzrranyrfr/rr0r|s  zReadTarFS.isdircCs4t||}z|j|WStyYdSwNF)rrrzisfiler)rNrArr/r/r0rs  zReadTarFS.isfilecC|t|r*checkrResourceReadOnly)rNrArvr/r/r0setinfo zReadTarFS.setinfocsXt||||tjurt|fdd|jD}dd|D}tt |S)Nc3s$|] }t|rt|VqdSr*)rr)runrr/r0rxs   z$ReadTarFS.listdir..css$|] }t|rt|dVqdS)rN)rr)ruchildr/r/r0rxs") rrZgettyper rrZDirectoryExpectedrzlistrfromkeys)rNrAchildrenZcontentr/rr0listdirs  zReadTarFS.listdircCrr*r)rNrArZrecreater/r/r0makedirs zReadTarFS.makedirrlc Kst||}d|vsd|vsd|vrt|z|j|}Wnty1tt|dYnw| s;t |t t t |j|}tjrPdd}||_|S)Nw+acSrMr*r/r/r/r/r0_flushsz!ReadTarFS.openbin.._flush)rrrrrzrr=Z raise_fromrrZ FileExpectedrrrrqZ extractfilePY2flush) rNrArn bufferingoptionsrr-rwrr/r/r0openbins   zReadTarFS.openbincCrr*rrfr/r/r0removerzReadTarFS.removecCrr*rrfr/r/r0 removedirrzReadTarFS.removedircs*tt|t|dr|jdSdS)Nrq)r[r'rjhasattrrqrcr\r/r0rjs zReadTarFS.closecCs|jjSr*)rqclosedrcr/r/r0riszReadTarFS.iscloseddownloadcCs>|dkrt|jtjrt|j}t|}d||St||)NZfsz tar://{}!/{})r<rVr=Z string_typesrr^r )rNrAZpurposeZ quoted_fileZ quoted_pathr/r/r0geturls    zReadTarFS.geturl)r6r*r)rlr)r)1rPrQrRrSrZroZBLKTYPEr Zblock_special_fileZCHRTYPE characterrrZFIFOTYPEZfifoZREGTYPErGZAREGTYPEZSYMTYPEsymlinkZCONTTYPEZLNKTYPEZ_typemaprrZ CreateFailedZ catch_allrOpropertyrzr`rdr=rrrwrrrrrrrrrrjrirrkr/r/r\r0r'sZ     6  __main__)renderz tests.tarrsz/tests/ztests/ttt/settings.iniz TarFS.tarT)rHzfoo/barzfoo/bar/baz.txtz Hello, World)JrSZ __future__rrr@rorT collectionsrrrr=r9rbaser compressr Zenumsr r r rvrZiotoolsrZopenerrrrZ _url_toolsrrArrrrrrZwrapfsrrUrrrrrrr r!r"r#r$Zsubfsr%TypeVarr&__all__rr1r(Zpython_2_unicode_compatibler)r'rPZfs.treerZtar_fsprintrZreadtextreprmakedirsZ writetextr/r/r/r0sl               ,  \L s    $