File: //opt/saltstack/salt/lib/python3.10/site-packages/relenv/__pycache__/common.cpython-310.pyc
o
�N�gp; � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
Z
ddlZdZe�e��� jZdZdZdZdZdZed ed
ed iZe
jdkrcej�� d d
d Z n ej�� d d Z e�ej � de ���� Z e � d�Z!e � d�Z"e
jdkr�e!Z#ne"Z#e�$e%�Z&G dd� de'�Z(e#fdd�Z)dd� Z*d?dd�Z+d?dd �Z,G d d � d �Z-d?d d �Z.d@d!d"�Z/d@d#d$�Z0d%d&� Z1d'd(� Z2d?d)d*�Z3d+d,� Z4d-d.� Z5dAd0d1�Z6dBd3d4�Z7dCd7d8�Z8d9d:� Z9d;d<� Z:d=d>� Z;dS )Dz/
Common classes and values used around relenv.
� Nz0.18.1�linux�win32�darwinz10.15)z5packages.broadcom.com/artifactory/saltproject-genericzrepo.saltproject.iozwoz.io)�x86_64Zaarch64)r Zarm64)Zamd64Zx86ZAppDataZLocal�relenvz.localZ RELENV_DATAzP#!/bin/sh
"true" ''''
"exec" "$(dirname "$(readlink -f "$0")"){}" "$0" "$@"
'''
a� #!/bin/sh
"true" ''''
TARGET_FILE=$0
cd "$(dirname "$TARGET_FILE")" || return
TARGET_FILE=$(basename "$TARGET_FILE")
# Iterate down a (possible) chain of symlinks
while [ -L "$TARGET_FILE" ]
do
TARGET_FILE=$(readlink "$TARGET_FILE")
cd "$(dirname "$TARGET_FILE")" || return
TARGET_FILE=$(basename "$TARGET_FILE")
done
PHYS_DIR=$(pwd -P)
REALPATH=$PHYS_DIR/$TARGET_FILE
"exec" "$(dirname "$REALPATH")"{} "$REALPATH" "$@"
'''c @ s e Zd ZdZdS )�RelenvExceptionz9
Base class for exeptions generated from relenv.
N)�__name__�
__module__� __qualname__�__doc__� r r �A/opt/saltstack/salt/lib/python3.10/site-packages/relenv/common.pyr _ s r c C s |� | ��� S )z%
Return a formatted shebang.
)�format�strip)�pythonZtplr r r
�format_shebange s r c C s t �� } | �� S )z%
Return the current machine.
)�platform�machine�lower)r r r r
�
build_archl s r c C s" | dur
t �| ��� }|S t}|S )a
Get the root directory that all other relenv working directories should be based on.
:param root: An explicitly requested root directory
:type root: str
:return: An absolute path to the relenv root working directory
:rtype: ``pathlib.Path``
N)�pathlib�Path�resolve�
MODULE_DIR)�root�baser r r
� work_roott s
�r c C s. t |�}|tkr|d�| � }|S || }|S )a`
Get the absolute path to the relenv working directory of the given name.
:param name: The name of the directory
:type name: str
:param root: The root directory that this working directory will be relative to
:type root: str
:return: An absolute path to the requested relenv working directory
:rtype: ``pathlib.Path``
z_{})r r r )�namer r r r r
�work_dir� s �r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �WorkDirsz�
Simple class used to hold references to working directories relenv uses relative to a given root.
:param root: The root of the working directories tree
:type root: str
c C sT || _ td| j �| _tdt�| _tdt�| _tdt�| _tdt�| _tdt�| _d S )N� toolchain�build�src�logs�download) r r �toolchain_config�DATA_DIRr r! r" r# r$ )�selfr r r r
�__init__� s zWorkDirs.__init__c C s" | j | j| j| j| j| j| jd�S )z[
Return an object used for pickling.
:return: The picklable state
�r r% r r! r" r# r$ r) )r' r r r
� __getstate__� s �zWorkDirs.__getstate__c C sJ |d | _ |d | _|d | _|d | _|d | _|d | _|d | _dS ) z�
Unwrap the object returned from unpickling.
:param state: The state to unpickle
:type state: dict
r r% r r! r" r# r$ Nr) )r' �stater r r
� __setstate__� s
zWorkDirs.__setstate__N)r r r
r r( r* r, r r r r
r � s
r c C s t t| ��S )z�
Returns a WorkDirs instance based on the given root.
:param root: The desired root of relenv's working directories
:type root: str
:return: A WorkDirs instance based on the given root
:rtype: ``relenv.common.WorkDirs``
)r r )r r r r
� work_dirs� s
r- c C s" t |�}| r|jd�| � S |jS )aF
Get a the toolchain directory, specific to the arch if supplied.
:param arch: The architecture to get the toolchain for
:type arch: str
:param root: The root of the relenv working directories to search in
:type root: str
:return: The directory holding the toolchain
:rtype: ``pathlib.Path``
z {}-linux-gnu)r- r r )�archr �dirsr r r
�
get_toolchain� s r0 c C sX |st j}| s
t� } |dkr| � d�S |dkr | � d�S |dkr%| � d�S td|� ���)a�
Get the target triplet for the specified machine and platform.
If any of the args are None, it will try to deduce what they should be.
:param machine: The machine for the triplet
:type machine: str
:param plat: The platform for the triplet
:type plat: str
:raises RelenvException: If the platform is unknown
:return: The target triplet
:rtype: str
r z-macosr z-winr z
-linux-gnuzUnknown platform )�sysr r r )r �platr r r
� get_triplet� s
r3 c C s2 | dkrdS | dkr dS | dkrdS t d| � ���)zC
Convert platform from build to the value of sys.platform.
z linux-gnur Zmacosr �winr zUnkown platform )r )r2 r r r
�plat_from_triplet s r5 c C sx g } t t�}t�|j�D ]-\}}}|D ]%}|�d�r8|dd� }|�dd�\}}|�dd�\}}| �|||f� qq | S )zJ
Return a list of version, architecture and platforms for builds.
�.tar.xzNi�����-� )r- r& �os�walkr! �endswith�split�append) Zbuildsr/ r �files�file�version�tripletr. r2 r r r
�list_archived_builds s
��rB c C s& | st � } tt�}| � d�}|j| S )z�
Finds a the location of an archived build.
:param triplet: The build triplet to find
:type triplet: str
:return: The location of the archived build
:rtype: ``pathlib.Path``
r6 )r3 r- r& r! )rA r/ �archiver r r
�archived_build$ s
rD c C sp |� d�rd}n|� d�rd}n
|� d�rd}nd}t�||��}|�| � W d � dS 1 s1w Y dS ) z�
Extract an archive to a specific location.
:param to_dir: The directory to extract to
:type to_dir: str
:param archive: The archive to extract
:type archive: str
Ztgzzr:gz�xzzr:xz�bz2zr:bz2�rN)r; �tarfile�open�
extractall)Zto_dirrC Z read_type�tr r r
�extract_archive5 s
"�rL c C s t j�|t j�| ��S )a
Get the full path to where the url will be downloaded to.
:param url: The url to donwload
:type url: str
:param dest: Where to download the url to
:type dest: str
:return: The path to where the url will be downloaded to
:rtype: str
)r9 �path�join�basename)�url�destr r r
�get_download_locationJ s rR � c C sl ddl }d}z'z
|jj| |d�}W n ty$ Y W |r"|�� dS dS w W |r,|�� dS |r5|�� w w )z+
Check that the url returns a 200.
r N��timeoutFT)�urllib.request�request�urlopen� Exception�close)rP rU �urllib�finr r r
� check_urlY s" ����
�r] � c
C sR ddl }ddl}t�� }|dk rd}d}||k ra|d7 }z
|jj| |d�}W n6 |jj|jjtj j
fy\ } z!||krEt d| � d|� ���t �
d| � t�|d � W Y d}~nd}~ww ||k st �d | � z4d} d
}
|�|
�} | r�| |
7 } t�� | dkr�t �d | | � t�� }|�| � |�d �} | ssW |�� n|�� w t �d
| � dS )zn
Fetch the contents of a url.
This method will store the contents in the given file like object.
r Nr8 rT zError fetching url � zUnable to connect %s�
z
url opened %si � z%s > %di ( zDownload complete %s)Z urllib.errorrV �timerW rX �errorZ HTTPErrorZURLError�httpZclientZRemoteDisconnectedr �log�debug�sleep�info�read�writerZ ) rP �fp�backoffrU r[ �last�nr\ �exc�total�size�blockr r r
� fetch_urlk sL � ���
��rr T�< c C s� t | |�}|rtd| � d|� �� t|d�}z
t| |||� W |S tyM } z!|r8td| � d|� �tjdd� zt�|� W � t yH Y � w d}~ww ) a�
Download the url to the provided destination.
This method assumes the last part of the url is a filename. (https://foo.com/bar/myfile.tar.xz)
:param url: The url to download
:type url: str
:param dest: Where to download the url to
:type dest: str
:param verbose: Print download url and destination to stdout
:type verbose: bool
:raises urllib.error.HTTPError: If the url was unable to be downloaded
:return: The path to the downloaded content
:rtype: str
z Downloading z -> �wbzUnable to download: r_ T)r? �flushN)
rR �printrI rr rY r1 �stderrr9 �unlink�OSError)rP rQ �verboserk rU �localZfoutrn r r r
� download_url� s&
� ����r| c O s t �dd�| d �� tj|d< tj|d< d|vr d|d< tjdkrutj| i |��}t� � }|�
|j tj � |�
|j
tj � d}|rt|�� D ]+\}}|j�� }|sUd } n |�d
�r`|d d � }|j|j u rlt �|� qFt �|� qF|sBnyd
d� }dd� } tj| i |��}t�� }
tj||j |
dfd�} tj||j
|
dfd�} tj| ||
fd�}
| �� | �� |
�� |
�� \}}|dkr�t �|d d � � n|dkr�t �|d d � � n
|dkr�t �d� nq�|
�� | �� | �� |�� |jdk�rt d� d�| d ����|S )aV
Run a command.
Run the provided command, raising an Exception when the command finishes
with a non zero exit code. Arguments are passed through to ``subprocess.run``
:return: The process result
:rtype: ``subprocess.CompletedProcess``
:raises RelenvException: If the command finishes with a non zero exit code
zRunning command: %sr_ r �stdoutrw �universal_newlinesTr F�
N���c S sX t � }t| j|�D ]}||u s|dkr n
|r |�||f� q t�d||� | �� d S )N� zstream close %r %r)�object�iter�readline�putrd re rZ )�stream�queue�typeZNOOP�liner r r
�enqueue_stream� s � z runcmd.<locals>.enqueue_streamc S s | � � |�d� d S )N)�xr� )�waitr� )�processr� r r r
�enqueue_process� s z runcmd.<locals>.enqueue_processr8 )�target�args� r� zprocess queue endzBuild cmd '{}' failed) rd re rN �
subprocess�PIPEr1 r �Popen� selectors�DefaultSelector�registerr} �
EVENT_READrw �select�fileobjr� r; rg rb r� ZQueue� threading�Thread�start�getr� �
returncoder r )r� �kwargs�pZsel�ok�keyZval1r� r� r� �q�toZte�tp�kindr r r
�runcmd� sf
��
�
r� c C s� t �|�}t �|�}t �| �}z|�|�}W n ty! td��w zt jdd� |�|�jD �� }W || S ty@ td��w )zM
Return a relativized path to the given scripts_dir and interpreter.
z$interperter not relative to root_dirc s s � | ]}d V qdS )z..Nr )�.0r� r r r
� <genexpr> s � z'relative_interpreter.<locals>.<genexpr>z$scripts_dir not relative to root_dir)r r � relative_to�
ValueError�parts)�root_dirZ scripts_dir� interpreter�scriptsZinterpr Z relinterpZ
relscriptsr r r
�relative_interpreter s
� ��r� c
C s� g }t t�tj��� t�tj��� h�}| D ]%}|D ] }zt�|��� }|�|� |�t |�� W q t
y: Y qw qdt j v rRt j d �
t j�D ]}|�|� qJ|S )zW
Sanitize `sys.path` to only include paths relative to the onedir environment.
Z
PYTHONPATH)�tupler r r1 �prefixr � base_prefixr� r= �strr� r9 �environr< �pathsep)Zsys_path_entriesZ
__sys_pathZ__valid_path_prefixesZ__pathZ__valid_path_prefixZ__resolved_pathr r r
�sanitize_sys_path" s( ��
��
r� )N)NN)rS )r^ rS )Tr^ rs )<r Z http.clientrc Zloggingr9 r r r� r� r� r1 rH �textwrapr� ra � __version__r �__file__r �parentr ZLINUXZWIN32ZDARWINZMACOS_DEVELOPMENT_TARGETZ CHECK_HOSTSZarches�homeZDEFAULT_DATA_DIRr� r� r& �dedentZSHEBANG_TPL_LINUXZSHEBANG_TPL_MACOSZ SHEBANG_TPLZ getLoggerr rd rY r r r r r r r- r0 r3 r5 rB rD rL rR r] rr r| r� r� r� r r r r
�<module> sv �
��
1
-#U