U
e5do � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZdddgZdZej �
d�rXdZ ne� � Z G dd� de
�Ze jd d
� �Zd d � Zdd
d�Zddd�Zdd� Zddd�Zedkr�e� dS )� N� ZipAppError�create_archive�get_interpreterz8# -*- coding: utf-8 -*-
import {module}
{module}.{fn}()
�win�utf-8c @ s e Zd ZdS )r N)�__name__�
__module__� __qualname__� r
r
� /usr/lib64/python3.8/zipapp.pyr ! s c c s8 t | ttjf�r.t| |�� }|V W 5 Q R X n| V d S �N)�
isinstance�str�os�PathLike�open)�archive�mode�fr
r
r � _maybe_open% s r c C s$ |r d|� t� d }| �|� d S )N� #!�
)�encode�shebang_encoding�write)r � interpreterZshebangr
r
r �_write_file_prefix. s r c
C s� t | d��Z}|�d�}|dkr*d}|�� t |d��&}t||� |�|� t�||� W 5 Q R X W 5 Q R X |r�t|t�r�t �
|t � |�j t j
B � d S )N�rb� r � �wb)r �read�readliner r �shutilZ copyfileobjr
r r �chmod�stat�st_mode�S_IEXEC)r Z new_archiver �srcZfirst_2Zdstr
r
r �
_copy_archive5 s
r) Fc
C s� d}t | d�r t | d�r d}nt�| �} | �� r4d}|rHt| ||� d S | �� sXtd��| d �� }|rt|rttd��|s�|s�td��d }|r�|�d �\} }
} td
d � | � d �D ��} td
d � | � d �D ��}
|
d kr�| r�|
s�td| ��t
j | | d�}|d k�r | � d�}nt |d��s"t�|�}t
|d���}t||� |�rDtjntj}tj|d|d��^}| �d�D ]4}|�| �}|d k�s�||��rf|�||�� � �qf|�r�|�d|�d�� W 5 Q R X W 5 Q R X |�r�t |d��s�|�|�� jtj B � d S )NFr! r" TzSource does not existz __main__.pyz8Cannot specify entry point if the source has __main__.pyzArchive has no entry point�:c s s | ]}|� � V qd S r �� isidentifier��.0�partr
r
r � { s z!create_archive..�.c s s | ]}|� � V qd S r r+ r- r
r
r r0 | s zInvalid entry point: )�module�fnz.pyzr r �w)� compression�*r ) �hasattr�pathlib�Path�is_filer) �existsr � partition�all�split�
MAIN_TEMPLATE�formatZ with_suffixr r �zipfileZ ZIP_DEFLATEDZ
ZIP_STOREDZZipFileZrglob� relative_tor Zas_posixZwritestrr r$ r% r& r' )�source�targetr �main�filter�
compressedZsource_is_fileZhas_mainZmain_py�mod�sepr3 Zmod_okZfn_ok�fdr5 �zZchildZarcnamer
r
r r L sX
�
�
&c
C sF t | d��2}|�d�dkr8|�� �� �t�W 5 Q R � S W 5 Q R X d S )Nr r r )r r! r" �strip�decoder )r r r
r
r r � s c C s< dd l }|�� }|jddd dd� |jddd dd� |jd d
d d d� |jd d
ddd� |jddddd� |jddd� |�| �} | jr�tj�| j�s�t d��t
| j�}t d� |p�d�� t
�d� tj�| j��r | jd k�stj�| j��r tj�| j| j��r t d��| j�r t d ��t| j| j| j| j| jd � d S ) Nr z--outputz-ozAThe name of the output archive. Required if SOURCE is an archive.)�default�helpz--pythonz-pzEThe name of the Python interpreter to use (default: no shebang line).z--mainz-mzLThe main function of the application (default: use an existing __main__.py).z
--compressz-c�
store_truezQCompress files with the deflate method. Files are stored uncompressed by default.)�actionrO z--infoFz)Display the interpreter from the archive.)rN rQ rO rC z'Source directory (or existing archive).)rO z%Can only get info for an archive filezInterpreter: {}zz-In-place editing of archives is not supportedz,Cannot change the main function when copying)r rE rG )�argparse�ArgumentParser� add_argument�
parse_args�infor �path�isfilerC �
SystemExitr �printr@ �sys�exit�outputr; �samefilerE r �python�compress)�argsrR �parserr r
r
r rE � sN
�
�
�
�
��
�
�rE �__main__)N)NNNNF)N)�
contextlibr r8 r# r% r[ rA �__all__r? �platform�
startswithr �getfilesystemencoding�
ValueErrorr �contextmanagerr r r) r r rE r r
r
r
r � s0
�
J
1