Current File : //lib/python3.9/site-packages/cockpit/_vendor/bei/__pycache__/beipack.cpython-39.pyc
a
��h� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZm Z m
Z
m Z m Z m
Z
ddlmZ eed�dd�Zeed�dd �Zeed�d
d �Zee e ed �d
d�Zee e ed �dd�Zeeef e e ed�dd�Zd(eeef e e eed�dd�Zd)e
e e e eeef d�dd�Zeeeeef d�d d �Zeeeef d �d d �Z eeeef d!�d"d#�Z dd$�d%d&�Z e d'k�r�e � dS )*� N)�Dict�Iterable�List�Optional�Set�Tuple� )�read_data_file)�data�returnc C s8 d}|| vr d| | | S d| vr0d| d S t | �S )Nz'''�rz"""zr""")�repr)r
Z triplequote� r �?/usr/lib/python3.9/site-packages/cockpit/_vendor/bei/beipack.py�
escape_string s r c C s dt | �d�� S )N�b�ascii�r �decode�r
r r r �ascii_bytes_repr( s r c C s t | �d��d S )N�utf-8z.encode('utf-8')r r r r r �utf8_bytes_repr, s r )r
�importsr c C s* |� d� t�| ��d��� }d|� d�S )Nz from binascii import a2b_base64r z a2b_base64("z"))�add�binasciiZ
b2a_base64r �strip)r
r Zencodedr r r �base64_bytes_repr0 s
r c C sF z
t | �W S ty Y n0 z
t| �W S ty: Y n0 t| |�S )N)r �UnicodeDecodeErrorr r )r
r r r r �
bytes_repr8 s
r )�contentsr r c s$ dd� � fdd�| �� D �� d S )Nz{
� c 3 s. | ]&\}}d t |�� dt|� �� d�V qdS )z z: z,
N)r
r )�.0�k�v�r r r � <genexpr>P s �z dict_repr.<locals>.<genexpr>�})�join�items)r r r r% r � dict_reprN s ���r* r! )r �
entrypoint�argsr c C s� t d�}dd� |�� D �}|�d� dh}t| |�}|�|� |�d|� d�� |r�|�d�\}}|�d |� d
|� d �� |�d |� d
�� d�dd� |D ��S )a� Creates a beipack with the given `contents`.
If `entrypoint` is given, it should be an entry point which is run as the
"main" function. It is given in the `package.module:func format` such that
the following code is emitted:
from package.module import func as main
main()
Additionally, if `args` is given, it is written verbatim between the parens
of the call to main (ie: it should already be in Python syntax).
zbeipack_loader.pyc S s g | ] }|r|�qS r r �r" �liner r r �
<listcomp>f � zpack.<locals>.<listcomp>r! z
import sysz&sys.meta_path.insert(0, BeipackLoader(z))�:zfrom z import z as mainzmain(�)c s s | ]}|� d �V qdS )�
Nr r- r r r r& s r0 zpack.<locals>.<genexpr>)r �
splitlines�appendr* �extend�splitr( ) r r+ r, �loader�linesr Z contents_txt�package�mainr r r �packU s
r< )� filenames� relative_tor c C sT i }| D ]F}t |d��(}|�� |tjj||d�< W d � q1 sD0 Y q|S )N�rb)�start)�open�read�os�path�relpath)r= r> r �filename�filer r r �collect_contentsv s
8rH )�name� recursiver c sV dd l }ddlm} t|ttttf d�� �fdd� �t�| �dd�|j �
| ���S )Nr )� Traversable)rD �entryr c 3 sX |� � D ]J}| � d|j� �}|�� r4||�� fV q� r|jdkr�||�E d H qd S )N�/� __pycache__)�iterdirrI �is_file�
read_bytes)rD rL �itemZitemname�rJ �walkr r rT � s z collect_module.<locals>.walk�.rM ) Zimportlib.resourcesZimportlib.resources.abcrK �strr r �bytes�dict�replaceZ resources�files)rI rJ � importlibrK r rS r �collect_module� s $r\ )rF r c C sZ i }t �| ��8}|jD ] }d|jv r&q|�|�||j< qW d � n1 sL0 Y |S )Nz .dist-info/)�zipfile�ZipFileZfilelistrF rB )rF r rG rL r r r � collect_zip� s
0r_ )rD r c C sR t �� �6}dd l}|�| �}|�d|�}t|�W d � S 1 sD0 Y d S )Nr �wheel)�tempfileZTemporaryDirectory�buildZProjectBuilderr_ )rD Ztmpdirrb Zbuilderr` r r r �collect_pep517� s
rc )r c C s� t �� } | jdddd� | jddddd � | jd
d d� | jd d
dd� | jddddd� | jddddd� | jddg dd � | jd d dg d d � | jd d!dg d"d#� | jd$d%d&d'� | �� }t|j|jd(�}|jD ]}|�t |�� q�|j
D ]}|�t |d)d*�� q�|j D ]}|�t
|�� �q t||j|j��d+�}|j�rXd,|j�d-� d. | }|j�rptj|tjd/�}|j�r�t|jd0��}|�|� W d � n1 �s�0 Y n,|j�r�t�d1��r�t � d2� t j j �|� d S )3Nz--pythonz-pz5add a #!python3 interpreter line using the given path)�helpz--xzz-J�
store_truez compress the output with `xz`)�actionrd z--topdirz6toplevel directory (paths are stored relative to this)z--outputz-oz(write output to a file (default: stdout)z--mainz-mz MODULE:FUNCz)use FUNC from MODULE as the main function)�metavarrd z --main-argsZARGSz$arguments to main() in Python syntaxr! )rg rd �defaultz--moduler5 z'collect installed modules (recursively))rf rh rd z--zipz-zz'include files from a zipfile (or wheel)z--build�DIRz%PEP-517 from a given source directory)rg rf rh rd rZ �*z files to include in the beipack)�nargsrd )r> T)rJ r s #!r �
)�preset�wbr z1refusing to write compressed output to a terminal) �argparse�ArgumentParser� add_argument�
parse_argsrH rZ Ztopdir�zip�updater_ �moduler\ rb rc r< r; Z main_args�encodeZpython�xz�lzma�compress�PRESET_EXTREME�outputrA �writerC �isatty�sys�exit�stdout�buffer)�parserr, r rG rI rD �resultr r r r; � sd �
���
��
� � ��
,
r; �__main__)Nr! )N) ro r rx rC r~ ra r] �typingr r r r r r r
r rV r rW r r r r r* r< rH �boolr\ r_ rc r; �__name__r r r r �<module> s<
� � " �� 4
Mini Shell