File: //lib/python3.6/site-packages/dnf-plugins/__pycache__/spacewalk.cpython-36.pyc
3
��`�3 � @ s^ d dl mZ d dl mZ d dlmZmZ d dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl m Z d dl
mZ d dlZd dlZd dlZd dlZd dlmZ d dlmZ d Zed
�Zed �Zed �Zed
�Zed�Z ed�Z ed�Z ed�Z ed�Z ed�Z!ed�Z"ed�Z#G dd� dej$�Z%G dd� dej&j'�Z(dd� Z)G d d � d e*�Z+d d � Z,d d!� Z-dS )"� )�absolute_import)�unicode_literals)�_�loggerN)�copy)�PRIO_PLUGINCONFIG)�ustr)�
up2dateErrorsz_spacewalk.jsonzCRed Hat Satellite or Spacewalk based repositories will be disabled.z@Red Hat Satellite or Spacewalk channel support will be disabled.zLThere was an error communicating with Red Hat Satellite or Spacewalk server.zIThis system is not registered with Red Hat Satellite or Spacewalk server.z.This system is not subscribed to any channels.z SystemId could not be acquired.z%You can use rhn_register to register.zLThis system is receiving updates from Red Hat Satellite or Spacewalk server.z�For security reasons packages from Red Hat Satellite or Spacewalk based repositories can be verified only with locally installed gpg keys. GPG key '%s' has been rejected.z.Package profile information could not be sent.zIMissing required login information for Red Hat Satellite or Spacewalk: %sz>Spacewalk plugin has to be run under with the root privileges.c sR e Zd ZdZ� fdd�Zdd� Zddd�Zd d
� Zd d � Zd
d� Z dd� Z
� Z S )� SpacewalkZ spacewalkc s� t t| �j||� || _|| _tjj| jjj t
�| _ d| _ i | _
t| jj�| _| j| j�| _d| jj� kr�| jjd�}x |D ]\}}| jj||t� q|W tjj� s�tjt� d| j_| jjs�d S tj d� | j � d S )NF�mainz initialized Spacewalk plugin) �superr
�__init__�base�cli�os�path�join�confZ
persistdir�STORED_CHANNELS_NAME�stored_channels_path�connected_to_spacewalk� up2date_cfgr Z read_config�parser�sections�items�
_set_valuer �dnf�utilZ am_i_rootr �warning� MUST_BE_ROOT�enabled�debug�activate_channels)�selfr r �options�key�value)� __class__� � /usr/lib/python3.6/spacewalk.pyr
: s(
zSpacewalk.__init__c C s | j js d S d| jj_d S )NT)r r r ZdemandsZ root_user)r# r( r( r) �configQ s zSpacewalk.configTc ) C sd i }d }d}d }d }| j � }|s(|}�nvtjj� | _t| j�}| jd f}ytjj| jj d�}W n2 t
j k
r� } zt j
dtt|� d S d }~X nX |s�t j
dtt� | ji � d S ytjj| jj d�} W n� t
jk
� r� } zt j
dtt|� d S d }~X nX t
jk
�r, t j
dtt� | ji � d S t
jk
�rT t j
dtttt� d S X d| _t j t � x,| D ]$}
|
d �rlt |
j � �||
d < �qlW | j|� | j j!} x�|j � D ]�\} }
|j"| �}d }|�r�|j"d�}t#| j�}| | j$j%� k�r | j$j | �}x |D ]\}}|j&||t'� �qW t(|
| j j||j |||||j)|j*d
� �}| j+|� �q�W t j,|� d S ) Nr ZuseNoSSLForPackages)�timeoutz%s
%s
%sz%s
%sz %s %s
%s
%sT�version�label) r �proxyr+ � sslcacert�
force_http�cached_version�
login_info�gpgcheckr )-�_read_channels_file�up2date_clientr* ZinitUp2dateConfigr �get_ssl_ca_certZ up2dateAuthZ getLoginInfor r+ r �RhnServerExceptionr �error�COMMUNICATION_ERROR� RHN_DISABLED�NOT_REGISTERED_ERROR�_write_channels_fileZ
rhnChannelZgetChannelDetailsZCommunicationErrorZNoChannelsError�NOT_SUBSCRIBED_ERROR�CHANNELS_DISABLEDZNoSystemIdError�NO_SYSTEM_ID_ERROR�USE_RHNREGISTERr �info�UPDATES_FROM_SPACEWALK�dictr r �repos�getr r r r r �
SpacewalkRepor3 r �addr! )r# Z
networkingZenabled_channelsr/ r0 Z proxy_urlr2 Zcached_channels�eZ svrChannels�channelrD Z
channel_idZ channel_dictZcached_channelr1 r r$ r% r& �repor( r( r) r" V s~
zSpacewalk.activate_channelsc C s� | j js dS | jsdS | jd dkr8| j� }tjj|� ytjj| j j d� W n4 t
j k
r� } zt j
dtt|� W Y dd}~X nX dS )z, Update system's profile after transaction. NZwriteChangesToLog� )r+ z%s
%s
%s)r r r r �_make_package_deltar5 ZrhnPackageInfoZlogDeltaPackagesZupdatePackageProfiler+ r r7 r r8 r9 �PROFILE_NOT_SENT)r# �deltarH r( r( r) � transaction� s zSpacewalk.transactionc C s� y.t | jd��}|j� }tj|�}|S Q R X W nX ttfk
rb } z|jtjkrR� W Y d d }~X n& tj j
k
r� } zW Y d d }~X nX i S )N�r) �openr �read�json�loads�FileNotFoundError�IOError�errno�ENOENT�decoderZJSONDecodeError)r# �
channels_fileZcontentZchannelsrH r( r( r) r4 � s
z Spacewalk._read_channels_filec C sf y,t | jd��}tj||dd� W d Q R X W n4 ttfk
r` } z|jtjkrP� W Y d d }~X nX d S )N�w� )�indent)rQ r rS �dumprU rV rW rX )r# �varrZ rH r( r( r) r< � s z Spacewalk._write_channels_filec C s. dd� | j jjD �dd� | j jjD �d�}|S )Nc S s$ g | ] }|j |j|j|j|jf�qS r( )�namer, �release�epoch�arch)�.0�pr( r( r) �
<listcomp>� s z1Spacewalk._make_package_delta.<locals>.<listcomp>c S s$ g | ] }|j |j|j|j|jf�qS r( )r` r, ra rb rc )rd re r( r( r) rf � s )ZaddedZremoved)r rO Z install_setZ
remove_set)r# rN r( r( r) rL � s
z Spacewalk._make_package_delta)T) �__name__�
__module__� __qualname__r` r
r* r" rO r4 r< rL �
__classcell__r( r( )r' r) r
6 s
Mr
c sD e Zd ZdZdddddgZ� fdd�Zd d
� Zd� fd d
� Z� ZS )rF zB
Repository object for Spacewalk. Uses up2date libraries.
zX-RHN-Server-IdzX-RHN-Auth-User-Idz
X-RHN-AuthzX-RHN-Auth-Server-TimezX-RHN-Auth-Expire-Offsetc sT t t� �jt|d �|jd�� t|d �� _� fdd�|d D �� _|jd�� _|jd�� _yt |d �� _
W n< t k
r� } z t j
ttjj|�� g � _
W Y d d }~X nX |d
|jd �kr�d � _|jd
�� _d� _d� _d � _d� _|jd�� _|jd�� _|jd�� _|jd��r$� j� n� j � t � d��rP� j � }|�rP� j |� d S )Nr- r r` c s g | ]}|d � j �qS )z /GET-REQ/)�id)rd �url)r# r( r) rf � s z*SpacewalkRepo.__init__.<locals>.<listcomp>rl r/ r. Z gpg_key_urlr, r1 rK r2 r r+ r3 r0 r �set_http_headers) r rF r
r rE r` Zbaseurlr/ r. �get_gpg_key_urlsZgpgkey�InvalidGpgKeyLocationr r �GPG_KEY_REJECTEDr Zi18nZucdZmetadata_expirer2 Z keepaliveZ bandwidthZretriesZthrottler+ r3 r0 �enable�disable�hasattr�create_http_headersrm )r# rI ZoptsrH � http_headers)r' )r# r) r
� s8
zSpacewalkRepo.__init__c C s| g }xb| j D ]X}|| jkr.t| }tjj|��| j| dkrL|jd| � q |jd|| j| f � q W | jsx|jd� |S )N� z*%s:
X-libcurl-Empty-Header-Workaround: *z%s: %sz.X-RHN-Transport-Capability: follow-redirects=3)Nrv )�needed_headersr2 �MISSING_HEADERr �ErrorZ RepoError�appendr0 )r# ru �headerr8 r( r( r) rt s
z!SpacewalkRepo.create_http_headersTc s0 t t| �j||�}| j� }|r,|jtj|� |S )N)r rF �_handle_new_remotert Zsetopt�librepoZLRO_HTTPHEADER)r# ZdestdirZ mirror_setupZhandleru )r' r( r) r| s
z SpacewalkRepo._handle_new_remote)T) rg rh ri �__doc__rw r
rt r| rj r( r( )r' r) rF � s %rF c C s* | j � }x |D ]}t|�st|��qW |S )a
Parse the key urls and validate them.
key_url_string is a space seperated list of gpg key urls that must be
located in /etc/pkg/rpm-gpg/.
Return a list of strings containing the key urls.
Raises InvalidGpgKeyLocation if any of the key urls are invalid.
)�split�is_valid_gpg_key_urlro )Zkey_url_stringZkey_urls�key_urlr( r( r) rn s
rn c @ s e Zd ZdS )ro N)rg rh ri r( r( r( r) ro / s ro c C sP | j d�}t|�dkrdS |\}}|j� dkr2dS tjj|�}|jd�sLdS dS )Nz://� F�filez/etc/pki/rpm-gpg/T)r �len�lowerr r �normpath�
startswith)r� Z proto_split�protor r( r( r) r� 2 s
r� c C s4 d| ko| d st �| d }t|�tkr0|d S |S )NZ sslCACertr )ZBadSslCaCertConfig�type�list)r Zca_certsr( r( r) r6 @ s r6 ).Z
__future__r r Zdnfpluginscorer r r Zdnf.exceptionsrW rS r} r r Zdnf.conf.configr Zup2date_client.up2dateAuthr5 Zup2date_client.configZup2date_client.rhnChannelZ up2date_client.rhnPackageInfoZrhn.i18nr r r r: r> r9 r; r= r? r@ rB rp rM rx r ZPluginr
rJ ZReporF rn � Exceptionro r� r6 r( r( r( r) �<module> sH M