o e$ @sddlZddlZddlZddlmZddlmZddlmZddlZGdddeZ GdddeZ dd e d e d e fd dZ   d de de dede d e f ddZd e de fddZd e de d ee eeffddZde d e fddZdS)!N)Callable) Generator)Listc@s>eZdZdZd ddZedefddZedefdd Z d S) XarInfozWInformational class which holds the details about an archive member given by a XarFile.tarcCs||_||_dSN)_infoarctype)selfZ_xarinfor r TD:\Projects\ConvertPro\env\Lib\site-packages\visualdl/component/inference/xarfile.py__init__s zXarInfo.__init__returncC|jdkr |jjS|jjSNr)r rnamefilenamer r r r r z XarInfo.namecCrr)r rsize file_sizerr r r r%rz XarInfo.sizeNr) __name__ __module__ __qualname____doc__r propertystrrintrr r r r rs rc @seZdZdZddededefddZdd Zd d Zd d Z   d dedede de fddZ dedefddZ defddZ deefddZdedeefddZdS)!XarFileac The XarFile Class provides an interface to tar/rar/zip archives. Args: name(str) : file or directory name to be archived mode(str) : specifies the mode in which the file is opened, it must be: ======== ============================================================================================== Charater Meaning -------- ---------------------------------------------------------------------------------------------- 'r' open for reading 'w' open for writing, truncating the file first, file will be saved according to the arctype field 'a' open for writing, appending to the end of the file if it exists ======== =============================================================================================== arctype(str) : archive type, support ['tar' 'rar' 'zip' 'tar.gz' 'tar.bz2' 'tar.xz' 'tgz' 'txz'], if the mode if 'w' or 'a', the default is 'tar', if the mode is 'r', it will be based on actual archive type of file rrmoder cKs(|dkr&|dvrd}d|_nH|dkrd}d|_n>|dvr"d}d|_n4||_n0|d krGt|r5d|_d }n!t|r>d |_nt|rFd |_n|d krO||_ntd||jdvrhtj ||fi||_ dS|jd krztj ||fi||_ dS|jd krtj ||fi||_ dStd|j)Nw)tar.gztgzzw:gzrtar.bz2zw:bz2)tar.xztxzzw:xzrzr:*zipZrarazUnsupported mode {})r"r$r%rr#r&zUnsupported archive type {}) r tarfile is_tarfilezipfile is_zipfilerarfile is_rarfile RuntimeErrorformatopen _archive_fpZipFileZRarFile)r rr r kwargsr r r r ?s@      zXarFile.__init__cCs|jdSr)r3closerr r r __del__gszXarFile.__del__cCs|Srr rr r r __enter__jszXarFile.__enter__cCs"|r t||||j|Sr)printr3r6)r Zexit_exceptionZ exit_valueZexit_tracebackr r r __exit__ms  zXarFile.__exit__NTarcname recursiveexcludec s|jdkr|jj||||ddS|j||rtj|s!dSg}t|D]\}}|fdd|D7}|fdd|D7}q(|D]}|rO||sOqF|j|qFdS)a Add the file `name' to the archive. `name' may be any type of file (directory, fifo, symbolic link, etc.). If given, `arcname' specifies an alternative name for the file in the archive. Directories are added recursively by default. This can be avoided by setting `recursive' to False. `exclude' is a function that should return True for each filename to be excluded. r)filterNcg|] }tj|qSr ospathjoin).0_file_dr r zXarFile.add..cr?r r@)rDZ_sub_drFr r rHrI)r r3addwriterArBisdirwalk) r rr;r<r=itemsZ_sub_ds_filesitemr rFr rJts   z XarFile.addrBcCs|j||S)z6Extract a file from the archive to the specified path.)r3extract)r rrBr r r rQszXarFile.extractcCs |j|S)z9Extract all files from the archive to the specified path.)r3 extractall)r rBr r r rRs zXarFile.extractallrcCs|jdkr |jS|jS)z+Return a list of file names in the archive.r)r r3getnamesnamelistrr r r rSs   zXarFile.getnamescCs2|jdkrt|j||jSt|j||jS)z,Return the instance of XarInfo given 'name'.r)r rr3 getmembergetinfo)r rr r r getxarinfos zXarFile.getxarinfor)NTN)rrrrrr r7r8r:boolrrJrQrRrrSrrWr r r r r,s,(  rr!rr rcKst||fi|S)zh Open a xar archive for reading, writing or appending. Return an appropriate XarFile class. )r)rr r5r r r r2sr2Trrr<r=r cCshtjtj|d}d||}t|d|d}|j|||dWd|S1s-wY|S)aK Archive a file or directory Args: name(str) : file or directory path to be archived recursive(bool) : whether to recursively archive directories exclude(Callable) : function that should return True for each filename to be excluded arctype(str) : archive type, support ['tar' 'rar' 'zip' 'tar.gz' 'tar.bz2' 'tar.xz' 'tgz' 'txz'] Returns: str: archived file path Examples: .. code-block:: python archive_path = '/PATH/TO/FILE' archive(archive_path, arcname='output.tar.gz', arctype='tar.gz') rz{}.{}r!)r r )r<r=N)rArBsplitextbasenamer1r2rJ)rr<r=r rZZsavenamefiler r r archives  r\rBcCs<t|dd}||WddS1swYdS)a% Unarchive a file Args: name(str) : file or directory name to be unarchived path(str) : storage name of archive file Examples: .. code-block:: python unarchive_path = '/PATH/TO/FILE' unarchive(unarchive_path, path='./output') r'r N)r2rR)rrBr[r r r unarchives "r^ccst|dd7}d}}|D] }|||j7}q|D]}||||||j7}|||fVqWddS1sAwYdS)a Unarchive a file and return the unarchiving progress -> Generator[filename, extrace_size, total_size] Args: name(str) : file or directory name to be unarchived path(str) : storage name of archive file Examples: .. code-block:: python unarchive_path = 'test.tar.gz' for filename, extract_size, total_szie in unarchive_with_progress(unarchive_path, path='./output'): print(filename, extract_size, total_size) r'r]rN)r2rSrWrrQ)rrBr[ total_sizeZ extract_sizerr r r unarchive_with_progresss   "r`r[cCs,tjtjtjg}|D] }||rdSq dS)z>Return True if xarfile supports specific file, otherwise FalseTF)r,r-r*r+r.r/)r[Z_x_funcZ_fr r r is_xarfiles ra)r!)TNr)rAr*r,typingrrrr.objectrrrr2rXr\r^rr`rar r r r s2    x