3
\� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZdddgZdZej j
d�rXdZ nej � 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}()
�winzutf-8c @ s e Zd ZdS )r N)�__name__�
__module__� __qualname__� r r � /usr/lib64/python3.6/zipapp.pyr ! s c c sF t | tj�rt| �} t | t�r<t| |�� }|V W d Q R X n| V d S )N)�
isinstance�pathlib�Path�str�open)�archive�mode�fr r r
� _maybe_open% s
r c C s$ |r d|j t� d }| j|� dS )zWrite a shebang line.s #!�
N)�encode�shebang_encoding�write)r � interpreterZshebangr r r
�_write_file_prefix0 s r c C s� t | d��Z}|jd�}|dkr*d}|j� t |d��&}t||� |j|� tj||� W dQ R X W dQ R X |r�t|t�r�t j
|t j |�j t j
B � dS )z8Copy an application archive, modifying the shebang line.�rb� s #!� �wbN)r �read�readliner r �shutilZ copyfileobjr r �os�chmod�stat�st_mode�S_IEXEC)r Z new_archiver �srcZfirst_2�dstr r r
�
_copy_archive7 s
r( c C s� d}t | d�r t | d�r d}ntj| �} | j� r4d}|rHt| ||� dS | j� sXtd��| d j� }|rt|rttd��|pz|s�td ��d}|r�|jd
�\}}} td d � |j d
�D ��}
tdd � | j d
�D ��} |d
ko�|
o�| s�td| ��t
j || d�}|dk�r | j d�}nt |d��s"tj|�}t
|d��~} t| |� tj| d��\}
tj| �}x2|jd�D ]$}t|j|��}|
jt|�|� �q\W |�r�|
jd|jd�� W dQ R X W dQ R X |�r�t |d� �r�|j|j� jtjB � dS )ab Create an application archive from SOURCE.
The SOURCE can be the name of a directory, or a filename or a file-like
object referring to an existing archive.
The content of SOURCE is packed into an application archive in TARGET,
which can be a filename or a file-like object. If SOURCE is a directory,
TARGET can be omitted and will default to the name of SOURCE with .pyz
appended.
The created application archive will have a shebang line specifying
that it should run with INTERPRETER (there will be no shebang line if
INTERPRETER is None), and a __main__.py which runs MAIN (if MAIN is
not specified, an existing __main__.py will be used). It is an error
to specify MAIN for anything other than a directory source with no
__main__.py, and it is an error to omit MAIN if the directory has no
__main__.py.
Fr r TNzSource does not existz __main__.pyz8Cannot specify entry point if the source has __main__.pyzArchive has no entry point�:c s s | ]}|j � V qd S )N)� isidentifier)�.0�partr r r
� <genexpr>| s z!create_archive.<locals>.<genexpr>�.c s s | ]}|j � V qd S )N)r* )r+ r, r r r
r- } s zInvalid entry point: )�module�fnz.pyzr r �w�*zutf-8)�hasattrr r
�is_filer( �existsr � partition�all�split�
MAIN_TEMPLATE�formatZ with_suffixr r �zipfileZZipFileZrglobr Z relative_tor Zwritestrr r" r# r$ r% )�source�targetr �mainZsource_is_fileZhas_mainZmain_py�mod�sepr0 Zmod_okZfn_ok�fd�z�rootZchildZarcnamer r r
r N sP
&c C s: t | d��&}|jd�dkr,|j� j� jt�S W d Q R X d S )Nr r s #!)r r r �strip�decoder )r r r r r
r � s c C s ddl }|j� }|jddddd� |jdddd d� |jd
d dd d� |jd
dddd� |jddd� |j| �} | jr�tjj| j�s�t d��t
| j�}t dj |p�d�� t
jd� tjj| j��r| jdks�tjj| j�r�tjj| j| j�r�t d��| j�rt d��t| j| j| j| jd� dS )z�Run the zipapp command line interface.
The ARGS parameter lets you specify the argument list directly.
Omitting ARGS (or setting it to None) works as for argparse, using
sys.argv[1:] as the argument list.
r Nz--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--infoF�
store_truez)Display the interpreter from the archive.)rF �actionrG r< z'Source directory (or existing archive).)rG z%Can only get info for an archive filezInterpreter: {}z<none>z-In-place editing of archives is not supportedz,Cannot change the main function when copying)r r> )�argparse�ArgumentParser� add_argument�
parse_args�infor! �path�isfiler< �
SystemExitr �printr: �sys�exit�outputr5 �samefiler> r �python)�argsrJ �parserr r r r
r> � s6
r> �__main__)N)NNN)N)�
contextlibr! r r r# rS r; �__all__r9 �platform�
startswithr �getfilesystemencoding�
ValueErrorr �contextmanagerr r r( r r r> r r r r r
�<module> s(
G
-