o pf@sdZddlmZmZmZddlZddlZddlZddlZddl Z ddl m Z ddl m Z ddl mZmZddlmZdd lmZdd lmZejrgdd lmZmZmZmZmZdd lmZeeeeeeefZe jd dfddZ  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_importprint_functionunicode_literalsN)datetime) ResourceType)MissingInfoNamespace NoSysPath)relpath)datetime_to_epoch)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 compression allowZip64detailsstataccess namespaces/replacerst_mtimeNr) zipfileZipFiler infor is_dirsixPY3encode has_namespacegettime localtimemodifiedrutcnowyearmonthdayhourminutesecondZipInfo permissionsr external_attrrwritestr getsyspathr readbyteswrite)src_fsfilerencodingwalker_zipgen_walkpathr&zip_namer _mtimezip_timemtzip_infosys_pathrKD/home/ertert/spirit/venv/lib/python3.10/site-packages/fs/compress.py write_zipsB         "rMcCstjtjtjtjtjtjtjtj 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)aWrite the contents of a filesystem to a tar 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 (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. ))uidrN)gidrO)unameuser)gnamegroupzw:{})r)fileobjrrrrrr Nrri)/rblock_special_filetarfileBLKTYPE characterCHRTYPE directoryDIRTYPEfifoFIFOTYPEr?REGTYPEsocketAREGTYPEsymlinkSYMTYPEunknownformat isinstancer( text_type binary_typeopenr-r r&r r)r*TarInfor+r,r/rr floatintmtimegetattrsetattrr8rr'typeaddfilesizeopenbin)r>r?rr@rAtype_maptar_attrr_tar current_timerCrDr&tar_nametar_informtarattrinfoattrbin_filerKrKrL write_tarks`             $r})NrN)"__doc__ __future__rrrtypingr(rWr-r$renumsrerrorsrr rDr r walkr TYPE_CHECKINGr rrrrbaserrlZipTime ZIP_DEFLATEDrMr}rKrKrKrLs2       O