U
|
X ±dß" ã @ s\ d Z ddlZddlZddlZddlZddlmZmZ G dd dZ dd Z
|
e dkrXe
|
dS ) z,
|
Viewer for PyInstaller-generated archives.
|
é N)ÚCArchiveReaderÚZlibArchiveReaderc @ sX e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d ZdddZ dddZ
|
dd Z d S )Ú ArchiveViewerc C s. || _ || _|| _|| _g | _| jr*d| _d S )NF)ÚfilenameÚinteractive_modeÚrecursive_modeÚ
|
brief_modeÚstack)Úselfr r r r © r ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\PyInstaller/utils/cliutils/archive_viewer.pyÚ__init__ s zArchiveViewer.__init__c C s& | | j¡}tj | j¡}| j ||f¡ | js8| ¡ S | ||¡ zt
|
d d d¡}W n& t k
|
r~ t tjd Y q"Y nX |s| ¡ qD|d ¡ }|dkr¨q"qD|dkrº| ¡ qD|dkrÖ| j|dd
qD|dkrò| j|dd
qD|d kr| jd
|
\}}| ||¡ qD| ¡ qDd S ) Nz? é ©Úfiler ÚQÚUÚOÚXÚSéÿÿÿÿ)Ú_open_toplevel_archiver ÚosÚpathÚbasenamer Úappendr Ú_non_interactive_processingÚ_show_archive_contentsÚinputÚsplitÚEOFErrorÚprintÚsysÚstderrÚ _print_usageÚupperÚ_move_up_the_stackÚ_open_embedded_archiveÚ _extract_file)r
|
ÚarchiveÚ archive_nameÚtokensÚcommandr r r Úmain$ s6
|
|
zArchiveViewer.mainc C sÒ d}| j rÎ| j ¡ \}}|d7 }|dkr0td | ||¡ | jsDqt|tr|j ¡ D ]r\}^ }}|dkrXz| |¡}W n< t
|
k
|
r¸ } z td|d| t j d W 5 d }~X Y nX | j ||f¡ qXqd S )Nr r Ú Úzú Could not open embedded archive ú: r )r Úpopr! r r Ú
|
isinstancer ÚtocÚitemsÚopen_embedded_archiveÚ Exceptionr" r# r ) r
|
Z archive_countr* r) ÚnameÚ_ÚtypecodeZembedded_archiveÚer r r r P s"
|
,z)ArchiveViewer._non_interactive_processingc C sJ t dtjd t dtjd t dtjd t dtjd t dtjd d S )NzU: go up one levelr z/O <name>: open embedded archive with given namez&X <name>: extract file with given namez-S: list the contents of current archive againzQ: quit)r! r" r# )r
|
r r r r$ h s
|
zArchiveViewer._print_usagec C sF t | jdkr4| j ¡ | jd \}}| ||¡ ntdtjd d S )Nr r zAlready in the top archive!r )Úlenr r2 r r! r" r# )r
|
r* r) r r r r& o s
|
|
z ArchiveViewer._move_up_the_stackc C sP t j |¡s,td| dtjd t d¡ |dd
¡ dkrHt|S t |S )NzArchive z does not exist!r r éüÿÿÿz.pyz)
|
r r Úisfiler! r" r# ÚexitÚlowerr r )r
|
r r r r r w s
|
z$ArchiveViewer._open_toplevel_archiveNc
|
C s¸ |s t d}| ¡ }|s d S | jd \}}t|dsFtdtjd d S z| |¡}W nB tk
|
r } z$td|d| tjd W Y ¢d S d }~X Y nX | j ||f¡ |
|
||¡ d S )Nz Open name? r r6 z+Archive does not support embedded archives!r r0 r1 ) r Ústripr Úhasattrr! r" r# r6 r7 r r )r
|
r* r9 Zparent_archiver) r; r r r r' s
|
z$ArchiveViewer._open_embedded_archivec
|
C sì |s t d}| ¡ }| jd \}}zHt|tr:| |¡}n.t|trT|j|dd}ntdt| dW nB t k
|
r¬ } z$t
|
d|d|d | t j d
|
W 5 d }~X Y nX t d }|sÈt
|
t | n t|d }| |¡ W 5 Q R X d S ) NzExtract name? r T)Úrawz Extraction from archive type z not implemented!z!Failed to extract data for entry z from r1 r zOutput filename? Úwb)r rA r r3 r Úextractr ÚNotImplementedErrorÚtyper7 r! r" r# ÚreprÚopenÚwrite)r
|
r8 r* r) Údatar; r Úfpr r r r( s"
|
|
2 zArchiveViewer._extract_filec
|
C sB t |trtd|d | jr@|j ¡ D ]}td| q*nRtd |j ¡ D ]>\}\}}}}}td| d| d| d| d|d| qRnªt |trtd|d | jrÖ|j ¡ D ]}td| qÀnBtd |j ¡ D ].\}\} }}td| d| d| d| qèn$td| d td t| d
|
d S ) Nz Contents of z (PKG/CArchive):ú zE position, length, uncompressed_length, is_compressed, typecode, namez, z (PYZ):z# is_package, position, length, namez
|
(unknown)z2FIXME: implement content listing for archive type ú!) r3 r r! r r4 Úkeysr5 r rG )
|
r
|
r* r) r8 ÚpositionÚlengthZuncompressed_lengthZ is_compressedr: Ú
|
is_packager r r r · s$
|
0 $z$ArchiveViewer._show_archive_contents)N)N) Ú__name__Ú
|
__module__Ú __qualname__r r- r r$ r& r r' r( r r r r r r s ,
|
|
r c C sÄ t ¡ } | jddddddd | jdd ddd
|
d d | jd d ddddd tj | ¡ | jdddd | ¡ }tj | |¡ z&t|j |j
|
|j |j d}| ¡ W n tk
|
r¾ tdY nX d S )Nz-lz--listFÚ
|
store_trueÚ listing_modez:List the archive contents and exit (default: %(default)s).)ÚdefaultÚactionÚdestÚhelpz-rz --recursiveÚ recursivezHRecursively print an archive log (default: %(default)s). Implies --list.z-bz--briefÚbriefzOWhen displaying archive contents, show only file names. (default: %(default)s).r Z pyi_archivez PyInstaller archive to process.)Úmetavarr[ )r r r r zAborted by user.)ÚargparseÚArgumentParserÚ add_argumentÚ PyInstallerÚlogZ __add_optionsÚ
|
parse_argsZ__process_optionsr r rW r\ r] r- ÚKeyboardInterruptÚ
|
SystemExit)ÚparserÚargsZviewerr r r ÚrunÏ sV úúú ýü ri Ú__main__) Ú__doc__r_ r r" ZPyInstaller.logrb ZPyInstaller.archive.readersr r r ri rS r r r r Ú<module> s 90
|
