Edit File: spacewalk.cpython-36.pyc
3 ��co5 � @ 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.jsonz7CloudLinux Network based repositories will be disabled.z4CloudLinux Network channel support will be disabled.z@There was an error communicating with CloudLinux Network server.z=This system is not registered with CloudLinux Network server.z.This system is not subscribed to any channels.z SystemId could not be acquired.z%You can use rhn_register to register.z@This system is receiving updates from CloudLinux Network server.z�For security reasons packages from CloudLinux Network 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.z=Missing required login information for CloudLinux Network: %sz>Spacewalk plugin has to be run under with the root privileges.c sZ e Zd ZdZ� fdd�Zdd� Zdd� Zdd d �Zd d � Zd d� Z dd� 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.configc C s t jd� d S )NzI/usr/sbin/clnreg_ks --serverUrl=https://xmlrpc.cln.cloudlinux.com/XMLRPC/)r �system)r# r( r( r) �clnregV s zSpacewalk.clnregTc C s� i }d }d}d }d }| j � }|s(|}�n�tjj� | _t| j�}| jd f}d}xl|s�ytjj| jj d�}d}W qP t j k r� } z*|dkr�| j � d}wPt jdtt| � d S d } ~ X qPX qPW |s�t jdtt� | ji � d S ytjj| jj d�} W n� t jk �r& } zt jdtt| � d S d } ~ X nX t jk �rT t jdtt� | ji � d S t jk �r| t jdtttt� d S X d| _ t j t � x,| D ]$} | d �r�t | j � �|| d < �q�W | j|� | j!j"} x�|j � D ]�\} }|j#| �}d }|�r�|j#d �}t$| j�}| | j%j&� k�rF| j%j | �}x |D ]\}}|j'||t(� �q*W t)|| j!j||j |||||j*|j+d � �}| j,|� �q�W t j-|� d S ) Nr ZuseNoSSLForPackagesF)�timeoutTz%s %s %sz%s %sz %s %s %s %s�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, r �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 � SpacewalkRepor5 r �addr! )r# Z networkingZenabled_channelsr1 r2 Z proxy_urlr4 Zcached_channelsZ clnreg_tried�eZ svrChannels�channelrF Z channel_idZ channel_dictZcached_channelr3 r r$ r% r& �repor( r( r) r" Y 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_deltar7 ZrhnPackageInfoZlogDeltaPackagesZupdatePackageProfiler- r r9 r r: r; �PROFILE_NOT_SENT)r# �deltarJ 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_fileZcontentZchannelsrJ r( r( r) r6 � 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)rS r rU �dumprW rX rY rZ )r# �varr\ rJ 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( )rb r. rc rd re )rf rg r( r( r) rh � s )ZaddedZremoved)r rQ Z install_setZ remove_set)r# rP r( r( r) rN � s z Spacewalk._make_package_delta)T) �__name__� __module__� __qualname__rb r r* r, r" rQ r6 r> rN � __classcell__r( r( )r' r) r 6 s Ur c sD e Zd ZdZdddddgZ� fdd�Zd d � Zd� fd d � Z� ZS )rH 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 rb c s g | ]}|d � j �qS )z /GET-REQ/)�id)rf �url)r# r( r) rh � s z*SpacewalkRepo.__init__.<locals>.<listcomp>rn r1 r0 Z gpg_key_urlr. r3 rM r4 r r- r5 r2 r �set_http_headers) r rH r r rG rb Zbaseurlr1 r0 �get_gpg_key_urlsZgpgkey�InvalidGpgKeyLocationr r �GPG_KEY_REJECTEDr Zi18nZucdZmetadata_expirer4 Z keepaliveZ bandwidthZretriesZthrottler- r5 r2 �enable�disable�hasattr�create_http_headersro )r# rK ZoptsrJ � 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)Nrx )�needed_headersr4 �MISSING_HEADERr �ErrorZ RepoError�appendr2 )r# rw �headerr: r( r( r) rv s z!SpacewalkRepo.create_http_headersTc s0 t t| �j||�}| j� }|r,|jtj|� |S )N)r rH �_handle_new_remoterv Zsetopt�librepoZLRO_HTTPHEADER)r# ZdestdirZ mirror_setupZhandlerw )r' r( r) r~ s z SpacewalkRepo._handle_new_remote)T) ri rj rk �__doc__ry r rv r~ rl r( r( )r' r) rH � s %rH 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_urlrq )Zkey_url_stringZkey_urls�key_urlr( r( r) rp + s rp c @ s e Zd ZdS )rq N)ri rj rk r( r( r( r) rq : s rq 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� = 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) r8 K s r8 ).Z __future__r r Zdnfpluginscorer r r Zdnf.exceptionsrY rU r r r Zdnf.conf.configr Zup2date_client.up2dateAuthr7 Zup2date_client.configZup2date_client.rhnChannelZ up2date_client.rhnPackageInfoZrhn.i18nr r r r< r@ r; r= r? rA rB rD rr rO rz r ZPluginr rL ZReporH rp � Exceptionrq r� r8 r( r( r( r) �<module> sH )M