o Q`t@sdZddlmZddlmZddlmZddlZddlZddlZddlZddl m Z ddl Z ddl m Z dd l mZdd lmZdd lmZmZdd lmZejrodd lmZmZmZmZmZddlmZeeeeeeefZejddfddZ   dddZ!dS)zFunctions to compress the contents of a filesystem. Currently zip and tar are supported, using the `zipfile` and `tarfile` modules from the standard library. )absolute_import)print_function)unicode_literalsN)datetime) ResourceType)relpath)datetime_to_epoch) NoSysPathMissingInfoNamespace)Walker)BinaryIOOptionalTextTupleUnion)FSutf-8c Cstj|d|dd}|p t}||j|gdd}|D]\}}t|jr(|dn|} tjs4| |d} | drL| dd d } t | } | d d } n|j pRt} | j| j| j| j| j| jf} t| | }z|jd urt|jjd >|_Wn ty~Ynw|jr|jd O_||dqz||}Wnty||||Yqw||| qWd d S1swYd S)aWrite the contents of a filesystem to a zip file. Arguments: src_fs (~fs.base.FS): The source filesystem to compress. file (str or io.IOBase): Destination file, may be a file name or an open file object. compression (int): Compression to use (one of the constants defined in the `zipfile` module in the stdlib). Defaults to `zipfile.ZIP_DEFLATED`. encoding (str): The encoding to use for filenames. The default is ``"utf-8"``, use ``"CP437"`` if compatibility with WinZip is desired. walker (~fs.walk.Walker, optional): A `Walker` instance, or `None` to use default walker. You can use this to specify which files you want to compress. wT)mode compressionZ allowZip64ZdetailsstataccessZ namespaces/replacerst_mtimeNr) zipfileZipFiler inforis_dirsixPY3encode has_namespacegettime localtimemodifiedrZutcnowZyearZmonthZdayZhourZminutesecondZZipInfo permissionsrZ external_attrr ZwritestrZ getsyspathr Z readbyteswrite)src_fsfilerencodingwalker_zipgen_walkpathr#Zzip_namerZ_mtimeZzip_timemtZzip_infoZsys_pathr8-/usr/lib/python3/dist-packages/fs/compress.py write_zip sB         "r:cCstjtjtjtjtjtjtjtj tj tj tj tj tjtjtjtj i}gd}d|p+d}t|tjtjfr>tj||d}ntj||d}t} |pMt}||j|gdd} | D]\} } t| } tjsm| |d} t| }| d r| d d | }n| j p| }t|t!rt"|}t|t#rt$|}||_%|D]\}}t&| |d d urt'||t&| |d q| d rt&| j(d d|_)| j*rtj|_+|,|q\|| j+tj |_+| j-|_-|.| }|,||Wd n1swYq\Wd d S1swYd S)aGWrite the contents of a filesystem to a tar file. Arguments: file (str or io.IOBase): Destination file, may be a file name or an open file object. compression (str, optional): Compression to use, or `None` for a plain Tar archive without compression. encoding(str): The encoding to use for filenames. The default is ``"utf-8"``. walker (~fs.walk.Walker, optional): A `Walker` instance, or `None` to use default walker. You can use this to specify which files you want to compress. ))uidr;)gidr<)unameuser)gnamegroupzw:{})r)ZfileobjrrrrrrNrri)/rZblock_special_filetarfileZBLKTYPE characterZCHRTYPEZ directoryZDIRTYPEZfifoZFIFOTYPEr1ZREGTYPEZsocketZAREGTYPEsymlinkZSYMTYPEunknownformat isinstancer%Z text_typeZ binary_typeopenr*r r#rr&r'ZTarInfor(r)r,rr floatintmtimegetattrsetattrr.rr$typeZaddfilesizeZopenbin)r0r1rr2r3Ztype_mapZtar_attrrZ_tarZ current_timer5r6r#Ztar_nameZtar_inforKZtarattrZinfoattrZbin_filer8r8r9 write_tarls`             $rP)NrN)"__doc__Z __future__rrrr*rBtypingr!rr%Zenumsrr6rr errorsr r walkr Z TYPE_CHECKINGr rrrrbaserrJZZipTime ZIP_DEFLATEDr:rPr8r8r8r9s6          O