File: //lib64/python3.9/site-packages/rct/__pycache__/manifest_commands.cpython-39.pyc
a
X�Zh�9 � @ s� d dl Z d dlZd dlZd dlmZ d dlmZmZ d dlm Z d dl
m Z d dl m
Z
d dlmZ d dlmZ d d lmZ d
d � ZG d d
� d
e�ZG dd� de �ZG dd� de�ZG dd� de�ZdS )� N)�BytesIO)�ZipFile�
BadZipfile)� certificate)�
RCTCliCommand)�xstr)�InvalidCLIOptionError)�ourjson)�ugettextc C s0 | }|� d�D ] }||v r$|| }q dS q|S )N�.� )�split)Z json_dict�path�current�item� r �;/usr/lib64/python3.9/site-packages/rct/manifest_commands.py� get_value s
r c @ sX e Zd ZdZdZdd� Zdd� Zddd �Zd
d � Zd d
� Z dd� Z
dd� Z ddd�Z dS )�
ZipExtractAlla� extend ZipFile with a safer extractall
Zipfile() does not support extractall on python2.4, and the 2.6 versions
are known to be unsafe in how they extract files. 2.6 version does not
validate that files are within the archive root, or check that files are
created safely.
Contains helper methods for manipulating and reading
the zipfile more easily in memoryNc O sJ z t j| g|�R i |�� W n( tyD ttd�� t�d� Y n0 dS )z(
Validates the zip file
zManifest zip is invalid.� N)r �__init__r �print�_�sys�exit)�self�args�kwargsr r r r 7 s
zZipExtractAll.__init__c C s, | j d u r&t| �tj��}t|d�| _ | j S �N�r)� inner_zipr �read�RCTManifestCommand�
INNER_FILEr )r �outputr r r �_get_inner_zipA s
z ZipExtractAll._get_inner_zipFc C s~ z"t | �|��}|�� }|�� W nV tyx z |r:t�| �� �|d�}W n& tyr ttd�j |d���Y n0 Y n0 |S )NTz.Unable to find file "{file_path}" in manifest.)� file_path)
r r! �getvalue�close�KeyErrorr% �
_read_file� Exceptionr �format)r r& Zis_innerr$ �resultr r r r* G s �zZipExtractAll._read_filec C sV g }| � � }|�� D ]<}tj�|�\}}|tj�dd�krt|�dkr|�|� q|S )N�export� entitlementsr )r% �namelist�osr r
�join�len�append)r �resultsZin_zip�filenameZ read_pathZ read_filer r r �_get_entitlementsW s z ZipExtractAll._get_entitlementsc C s" t �t �|t jt jB t jB �d�S )N�wb)r1 �fdopen�open�O_RDWR�O_CREAT�O_EXCL)r r r r r �
_open_excl` s zZipExtractAll._open_exclc C s& | � |�}|�| �|�� |�� d S �N)r> �writer! r( )r Z output_pathZ archive_pathZoutfiler r r � _write_filec s
zZipExtractAll._write_filec C sl t j�|�}t j�|�}|�|�s.ttd���t j�|�}||krP| �||� nt j�|�rhttd���d S )Nz@Manifest zip attempted to extract outside of the base directory.z:Unable to trace symbolic link. Possibly circular linkage.) r1 r �abspath�
startswithr+ r �realpath�
_is_secure�islink)r �baseZnew_file� base_path�new_pathZ
real_new_pathr r r rE h s
zZipExtractAll._is_securec C s� | � ||� | �� D ]�}tj�|�\}}tj�||�}| � ||� tj�|�sXt�|� tj�||�}| � ||� tj�|�r�|r�t�|� | � ||� qd S r? )
rE r0 r1 r r
r2 �exists�makedirs�removerA )r �location� overwrite� path_nameZ directoryr6 Z new_locationr r r �
extractallt s
zZipExtractAll.extractall)F)F)
�__name__�
__module__� __qualname__�__doc__r r r% r* r7 r> rA rE rP r r r r r * s
r c @ s4 e Zd ZdZd
dd�Zdd� Zd d
� Zdd d �ZdS )r" zconsumer_export.zip�cliNFc C s t j| ||||d� d S )N��name�aliases� shortdesc�primary)r r )r rW rX rY rZ r r r r � s zRCTManifestCommand.__init__c C s t d�j| jd�S )Nz'%(prog)s {name} [OPTIONS] MANIFEST_FILE)rW )r r, rW �r r r r �
_get_usage� s z RCTManifestCommand._get_usagec C s4 | � � }|sttd���tj�|�s0ttd���d S )Nz!You must specify a manifest file.z+The specified manifest file does not exist.)�_get_file_from_argsr r r1 r �isfile)r Z
manifest_filer r r �_validate_options� s
z$RCTManifestCommand._validate_optionsc C sZ t | �� d�}|�||� |r0tj�|| j�}n| j}t |d�}|�||� t�|� d S r )r r] rP r1 r r2 r# rL )r rM rN �archiveZ
inner_filer r r �_extract_manifest� s
z$RCTManifestCommand._extract_manifest)rU NNF)F)rQ rR rS r# r r\ r_ ra r r r r r" � s
r" c @ sF e Zd Zdd� Zddd�Zdd� Zd d
� Zd d � Zd
d� Zdd� Z dS )�CatManifestCommandc C s6 t j| ddgtd�dd� | jjdddtd �d
� d S ) Nz cat-manifest�cmzPrint manifest informationTrV z --no-content�
store_trueFzskip printing Content Sets)�action�default�help�r" r r �parser� add_argumentr[ r r r r � s ��zCatManifestCommand.__init__r Tc C sz d|d }t || � |d7 }|D ]D}t|�dkrTt d||d t|d �f � q$t d||d f � q$|rvt d� d S )N� r � z%s%s: %sr z%s%sr )r r3 r )r �title�items�indentZ
whitespaceZpadr r r r �_print_section� s z!CatManifestCommand._print_sectionc C s� |� tj�dd��}t�|�}g }|�td�t|d�f� |�td�t|d�f� |�td�t|d�f� |�td �t|d
�f� | � td �|� d S ) Nr. z meta.jsonZServerZ webAppPrefixzServer Version�versionz Date Created�createdZCreatorZ
principalNamezGeneral:�
r* r1 r r2 �json�loadsr4 r r rp )r � zip_archive�part�data�to_printr r r �_print_general� s
z!CatManifestCommand._print_generalc C s� |� tj�dd��}t�|�}g }|�td�t|d�f� |�td�t|d�f� d}d|v rn|d d krnd
}|�td �|f� |�td �t|d
�f� |�td�t|d�f� |�td�t|d�f� | � td�|� d S )Nr. z
consumer.json�NamerW ZUUIDZuuidZ entitlement�contentAccessModeZorg_environmentzSimple Content AccesszContent Access Mode�Typez
type.labelzAPI URLZurlApizWeb URLZurlWebz Consumer:rs )r rv rw rx ry r| r r r �_print_consumer� s
z"CatManifestCommand._print_consumerc C s0 d }t |d�D ] }|d |kr|d } q,q|S )Nzpool.productAttributesrW �value)r )r rW rx Z return_value�attrr r r �_get_product_attribute� s z)CatManifestCommand._get_product_attributec C sJ |� � }t|�dkr0| �td�dggdd� d S |D �]}|�|�}t�|�}g }|�td�t|d�f� |�td�t|d �f� |�td
�t|d �f� |�td �t|d
�f� |�td�t|d�f� |�td�| � d|�f� |�td�| � d|�f� |�td�| � d|�f� |�td�t|d�f� |�td�t|d�f� |�td�t|d�f� |�td �t|d �f� | � d |�}|�td �|f� d }|�r�d}|�td!�|f� t
j � d"d#d$|d% �} |�td&�| f� |d' d d( d% }
t
j � d"d)d*|
�} |�td+�| f� zt
�|�| ��d,��} W n@ t
j�yt }
z$t
�td-�j| |
d.���W Y d }
~
n
d }
~
0 0 |�td/�| jf� | �td0�|dd � d1d2� |d3 d4 D �}| �td5�t|�d6d � d7|d3 v �rd8d2� |d3 d7 D �}| �td9�t|�d6d � | jj�s<d:d2� | jD �}| �td;�t|�d6d� q4td<� q4d S )=Nr zSubscriptions:�Noner Tr{ zpool.productNameZQuantityZquantityZCreatedrr z
Start DateZ startDatezEnd DateZendDatez
Service LevelZ
support_levelz Service TypeZ support_typeZ
ArchitecturesZarchZSKUzpool.productIdZContractzpool.contractNumberZOrderzpool.orderNumberZAccountzpool.accountNumber�
virt_limitz
Virt LimitFzRequires Virt-whor. r/ z%s.json�idzEntitlement FileZ certificates�serialZentitlement_certificatesz%s.pemzCertificate Filezutf-8zAUnable to read certificate file '{certificate_file}': {exception})Zcertificate_file� exceptionzCertificate Versionz
Subscription:c S s g | ]}t |d �|d f�qS �Z productIdZ productName��int��.0�ppr r r �
<listcomp> � z6CatManifestCommand._print_products.<locals>.<listcomp>ZpoolZprovidedProductszProvided Products:rl ZderivedProvidedProductsc S s g | ]}t |d �|d f�qS r� r� r� r r r r� s �zDerived Products:c S s g | ] }|j g�qS r )�url)r� r r r r r� ( r� z
Content Sets:r )r7 r3 rp r r* rt ru r4 r r� r1 r r2 r Zcreate_from_pem�decodeZCertificateExceptionr, rq �sorted�optionsZ
no_contentZcontentr )r rv r/ Zent_filerw rx ry r� Zrequire_virt_whoZentitlement_filer� Z cert_fileZcertZcer r r �_print_products� sh
��
�
z"CatManifestCommand._print_productsc C sL t | �� d�}td� ttd�� td� | �|� | �|� | �|� dS )�:
Does the work that this command intends.
r z.
+-------------------------------------------+z Manifestz.+-------------------------------------------+
N)r r] r r rz r~ r� )r Ztempr r r � _do_command- s
z CatManifestCommand._do_commandN)r T)
rQ rR rS r rp rz r~ r� r� r� r r r r rb � s
Erb c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�DumpManifestCommandc C sP t j| ddgtd�dd� | jjddtd�d � | jjd
d d d
dtd�d� d S )Nz
dump-manifestZdmz Dump the contents of a manifestTrV z
--destination� destinationz$directory to extract the manifest to)�destrg z-fz--forcerd �overwrite_filesFz overwrite files which may exist)re r� rf rg rh r[ r r r r = s( ���z DumpManifestCommand.__init__c
C s� z| � ||� W np ty� } zX|jtjkrDttd�j|jd�� n&ttd�� t|j� |jrjt|j� W Y d }~dS d }~0 0 dS )Nz?File "{filename}" exists. Use -f to force overwriting the file.)r6 z Manifest could not be written:FT) ra �EnvironmentError�errnoZEEXISTr r r, r6 �strerror)r r� rN �er r r �_extractT s ��
z DumpManifestCommand._extractc C sZ | j jr6| �| j j| j j�rVttd�j| j jd�� n | �t�� | j j�rVttd�� dS )r� z;The manifest has been dumped to the {destination} directory)r� z5The manifest has been dumped to the current directoryN) r� r� r� r� r r r, r1 �getcwdr[ r r r r� i s ��z DumpManifestCommand._do_commandN)rQ rR rS r r� r� r r r r r� < s r� )r� r1 r �ior �zipfiler r Zrhsmr Z rct.commandsr Z rct.printingr Zsubscription_manager.clir r rt Zsubscription_manager.i18nr
r r r r" rb r� r r r r �<module> s Y#