관리-도구
편집 파일: spacewalk.cpython-36.pyc
3 %T�f�6 � @ sf 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#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'Leapp upgrade is running - using cache.z>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) �configR s zSpacewalk.configc C s t jd� d S )Nz$/usr/sbin/clnreg_ks --strict-edition)r �system)r# r( r( r) �clnregW s zSpacewalk.clnregTc C s� i }d }d}d }d }| j � }|s(|}�n�tjjd�rFtjt� |}�n�tjj � | _ t | j �}| j d f}d}xl|s�ytj j | jjd�}d}W qn tjk r� } z*|dkr�| j� d}wntjdtt| � d S d } ~ X qnX qnW |� s�tjdtt� | ji � d S ytjj| jjd�} W n� tjk �rF } ztjdtt| � d S d } ~ X nX tjk �rt tjdt t � | ji � d S tj k �r� tjd t t t t� d S X d| _!tj"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-� �qJW t.|| j&j||j|||||j/|j0d � �}| j1|� �q�W tj2|� d S ) Nr z/etc/cln_leapp_in_progressZuseNoSSLForPackagesF)�timeoutTz%s %s %sz%s %sz %s %s %s %s�version�label) r �proxyr- � sslcacert� force_http�cached_version� login_info�gpgcheckr )3�_read_channels_filer r �isfiler r �LEAPP_IN_PROGRESS�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 � SpacewalkRepor5 r �addr! )r# Z networkingZenabled_channelsr1 r2 Z proxy_urlr4 Zcached_channelsZ clnreg_tried�eZ svrChannels�channelrH Z channel_idZ channel_dictZcached_channelr3 r r$ r% r& �repor( r( r) r" Z 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_deltar9 ZrhnPackageInfoZlogDeltaPackagesZupdatePackageProfiler- r r; r r< r= �PROFILE_NOT_SENT)r# �deltarL 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_fileZcontentZchannelsrL 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)rU r rW �dumprY rZ r[ r\ )r# �varr^ rL 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( )rd r. re rf rg )rh ri r( r( r) rj � s )ZaddedZremoved)r rS Z install_setZ remove_set)r# rR r( r( r) rP � s z Spacewalk._make_package_delta)T) �__name__� __module__� __qualname__rd r r* r, r" rS r6 r@ rP � __classcell__r( r( )r' r) r 7 s Yr c sD e Zd ZdZdddddgZ� fdd�Zd d � Zd� fd d � Z� ZS )rJ 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 rd c s g | ]}|d � j �qS )z /GET-REQ/)�id)rh �url)r# r( r) rj � s z*SpacewalkRepo.__init__.<locals>.<listcomp>rp r1 r0 Z gpg_key_urlr. r3 rO r4 r r- r5 r2 r �set_http_headers) r rJ r r rI rd Zbaseurlr1 r0 �get_gpg_key_urlsZgpgkey�InvalidGpgKeyLocationr r �GPG_KEY_REJECTEDr Zi18nZucdZmetadata_expirer4 Z keepaliveZ bandwidthZretriesZthrottler- r5 r2 �enable�disable�hasattr�create_http_headersrq )r# rM ZoptsrL � http_headers)r' )r# r) r � s8 zSpacewalkRepo.__init__c C s� g }| j s|S xb| jD ]X}|| j kr8t| }tjj|��| j | dkrV|jd| � q|jd|| j | f � qW | js�|jd� |S )N� z*%s: X-libcurl-Empty-Header-Workaround: *z%s: %sz.X-RHN-Transport-Capability: follow-redirects=3)Nrz )r4 �needed_headers�MISSING_HEADERr �ErrorZ RepoError�appendr2 )r# ry �headerr< r( r( r) rx s z!SpacewalkRepo.create_http_headersTc s0 t t| �j||�}| j� }|r,|jtj|� |S )N)r rJ �_handle_new_remoterx Zsetopt�librepoZLRO_HTTPHEADER)r# ZdestdirZ mirror_setupZhandlery )r' r( r) r� & s z SpacewalkRepo._handle_new_remote)T) rk rl rm �__doc__r{ r rx r� rn r( r( )r' r) rJ � s %rJ 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_urlrs )Zkey_url_stringZkey_urls�key_urlr( r( r) rr 2 s rr c @ s e Zd ZdS )rs N)rk rl rm r( r( r( r) rs A s rs 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� D 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) r: R s r: )/Z __future__r r Zdnfpluginscorer r r Zdnf.exceptionsr[ rW r� r r Zdnf.conf.configr Zup2date_client.up2dateAuthr9 Zup2date_client.configZup2date_client.rhnChannelZ up2date_client.rhnPackageInfoZrhn.i18nr r r r> rB r= r? rA rC rD rF rt rQ r| r8 r ZPluginr rN ZReporJ rr � Exceptionrs r� r: r( r( r( r) �<module> sJ -O