3
e�$g2 � @ sF d Z ddlZddlZddlZddlZddlmZmZ ddlm Z m
Z
yddl m Z m
Z
W n$ ek
r| ddlm Z m
Z
Y nX ddlZejddd�Zeed �s�eje_ejZd:d;d<d=d>d?d@dAdBdCdDdEdFd d d d d!gfd"�Zd#d d$gZG d%d&� d&�ZG d'd(� d(�Zd)d*� Zd+d,� Zd-d.� Z d/d0� Z d1d2� Z d3d4� Z d5d6� Z dGd8d9�Z!dS )Hzj
This module includes the Config and Up2date Config classes use by the
up2date agent to hold config info.
� N)�idn_ascii_to_puny�idn_puny_to_unicode)�ustr�sstr)�urlsplit�
urlunsplitzrhn-client-toolsT)Zfallback�ugettext�Use a HTTP Proxy�Remote server URL�)https://xmlrpc.cln.cloudlinux.com/XMLRPC/�Mirror list URL�>https://repo.cloudlinux.com/cloudlinux/mirrorlists/cln-mirrors�#Whether or not debugging is enabled�Location of system id�/etc/sysconfig/rhn/systemid�4Override the automatically determined system version� �;HTTP proxy in host:port format, e.g. squid.example.com:3128�'The username for an authenticated proxy�.The password to use for an authenticated proxy�$To use an authenticated proxy or not�BNumber of attempts to make at network connections before giving up� �)The CA cert used to verify the ssl server�'/usr/share/rhn/CLN-ORG-TRUSTED-SSL-CERT�Disable the reboot actionzDConfig options that can not be overwritten by a config update action� sslCACert� serverURL�disallowConfChanges�noReboot)Z enableProxyr � mirrorURL�debug� systemIdPathZversionOverride� httpProxy� proxyUser�
proxyPasswordZenableProxyAuthZnetworkRetriesr r r r" ZtmpDirc @ sl e Zd ZdZddd�Zddd�Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )�
ConfigFilez;class for handling persistent config options for the clientNc C s i | _ || _| jr| j� d S )N)�dict�fileName�load)�self�filename� r, � /usr/lib/python3.6/config.py�__init__J s zConfigFile.__init__c C s� |r
|| _ | j d krd S tj| j tj�s,d S t| j d�}d}�x�|j� D �]�}|jd�dkr^qH||j� }|spqH|d dkr�|d d� j� }qHnd}|jdd�}t |�dkr�qH|d j� }t
|d j� �}d }|jd �} | dkr�|d | � }|}d }|�rx|jd
�}
|dk�r"t |j t
j� ��}nVt |
�dk�rVy t|�}W n tk
�rR Y nX n"|
d dk�rjg }nd
d� |
D �}||f} || jk�r�| j| } |d k �r�|| d f} |d k �r�| d |f} | | j|< qHW |j� d S )N�rr �#r r �\�=� z [comment]�;r$ r% c S s g | ]}|j � r|j � �qS r, )�strip)�.0�valr, r, r- �
<listcomp>� s z#ConfigFile.load.<locals>.<listcomp>���r9 r9 )r$ r% )r( �os�access�R_OK�open� readlines�findr5 �split�lenr r �encode�locale�getpreferredencoding�int�
ValueErrorr' �close) r* r+ �fZ multiline�liner@ �key�value�comment�pos�valuesZnewvalr, r, r- r) P sd
zConfigFile.loadc C sJ | j d krd S tj| j tj�sTtjtjj| j �tj�sTttdtjj| j � �� d S t| j d d�}tj | j d t
dd�� |j d� |j d� |j d� x�| j j
� D ]�}| j | \}}|j td ||f �� t|�tg �kr�|g}|tkr�ttjj|�}|j td
|d jtt|��f �� |j d � q�W |j� tj| j d | j � d S )
Nz%s was not foundz.new�wZ0644� zI# Automatically generated Red Hat Update Agent config file, do not edit.
z# Format: 1.0
r z%s[comment]=%s
z%s=%s
r4 �
)r( r: r; r< �path�dirname�print�_r= �chmodrE �writer' �keysr �type� FileOptions�map�abspath�join�strrG �rename)r* rH rJ rL rK r, r, r- �save� s,
"zConfigFile.savec C s
|| j kS )N)r' )r* �namer, r, r- � __contains__� s zConfigFile.__contains__c C s || kS )Nr, )r* ra r, r, r- �has_key� s zConfigFile.has_keyc C s
| j j� S )N)r' rX )r* r, r, r- rX � s zConfigFile.keysc C s dd� | j j� D �S )Nc S s g | ] }|d �qS )r r, )r6 �ar, r, r- r8 � s z%ConfigFile.values.<locals>.<listcomp>)r' rN )r* r, r, r- rN � s zConfigFile.valuesc C s | j j|� d S )N)r' �update)r* r' r, r, r- re � s zConfigFile.updatec C s || j kr| j | d S d S )Nr )r' )r* ra r, r, r- � __getitem__� s
zConfigFile.__getitem__c C s0 || j kr| j | }nd}|d |f| j |< d S )Nr )NN)r' )r* ra rK r7 r, r, r- � __setitem__� s
zConfigFile.__setitem__c C s || j kr| j | d S dS )Nr r )r' )r* ra r, r, r- �info� s
zConfigFile.info)N)N)�__name__�
__module__� __qualname__�__doc__r. r) r` rb rc rX rN re rf rg rh r, r, r, r- r& H s
H$r& c @ sv e Zd Zd dd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS ) �ConfigNc C s. t � | _| jjt� |r$| jj|� i | _d S )N)r& �storedre �Defaultsr) �runtime)r* r+ r, r, r- r. � s
zConfig.__init__c C s || j krdS || jkr dS dS )NTF)rp rn )r* ra r, r, r- rb � s
zConfig.__contains__c C s || kS )Nr, )r* ra r, r, r- rc � s zConfig.has_keyc C s8 t | jj� �}x$| jj� D ]}||kr|j|� qW |S )N)�listrp rX rn �append)r* �ret�kr, r, r- rX � s
z Config.keysc C s* g }x | j � D ]}|j| j|�� qW |S )N)rX rr rf )r* rs rt r, r, r- rN � s z
Config.valuesc C s. g }x$| j � D ]}|j|| j|�f� qW |S )N)rX rr rf )r* rs rt r, r, r- �items s z Config.itemsc C s t | j� �S )N)rA rX )r* r, r, r- �__len__ s zConfig.__len__c C s || j |< d S )N)rp )r* ra rK r, r, r- rg s zConfig.__setitem__c C s, || j kr| j | S || jkr(| j| S d S )N)rp rn )r* ra r, r, r- rf s
zConfig.__getitem__c C s | j j|�S )N)rn rh )r* ra r, r, r- rh s z Config.infoc C s | j j� d S )N)rn r` )r* r, r, r- r` s z Config.savec C s8 | j j|� x&| j j� D ]}|| jkr(q| j|= qW d S )N)rn r) rX rp )r* r+ rt r, r, r- r) ! s
z Config.loadc C s || j |< || jkr | j|= d S )N)rn rp )r* ra rK r, r, r- �set+ s
z
Config.set)N)ri rj rk r. rb rc rX rN ru rv rg rf rh r` r) rw r, r, r, r- rm � s
rm c C s^ t � } d}| d }|rZ|dd� dkr2|dd� }|jd�}tt|d ��|d< dj|�}|S )zl returns proxy string in format hostname:port
hostname is converted to Punycode (RFC3492) if needed
Nr# � zhttp://�:r )�initUp2dateConfigr@ r^ r r] )�cfg�proxyZ proxyHost�partsr, r, r- �getProxySetting2 s
r~ c C s: t | �}tt|d tt|d ��|d |d |d f��S )z? returns url where hostname is converted to Punycode (RFC3492) r r r3 � � )r r r r r )�url�sr, r, r- �convert_url_to_punyC s r� c C s6 t | �}tt|d t|d �|d |d |d f��S )zZ returns url where hostname is converted from Punycode (RFC3492). Returns unicode string. r r r3 r r� )r r r r )r� r� r, r, r- �convert_url_from_punyH s r� c
C sv t d } | d krd} | jd�rVt| jdd�d�� }ttj|j� �}dd� |D �S Q R X tj | �}dd� |j
j d �D �S )
Nr z>https://repo.cloudlinux.com/cloudlinux/mirrorlists/cln-mirrorszfile://r r/ c S s g | ]}|rt |��qS r, )r� )r6 �mirrorr, r, r- r8 T s z+getServerlURLFromMirror.<locals>.<listcomp>c S s g | ]}|rt |��qS r, )r� )r6 r� r, r, r- r8 V s rQ ) r{ �
startswithr= �replacer[ r^ r5 r> �requests�get�textr@ )r� Z
mirrorlistZmirrorsZrequestr, r, r- �getServerlURLFromMirrorM s
r� c C s> t � } t| d �tg �kr,dd� | d D �S t| d �gS dS )z� return list of serverURL from config
Note: in config may be one value or more values, but this
function always return list
r c S s g | ] }t |��qS r, )r� )r6 �ir, r, r- r8 a s z!getServerlURL.<locals>.<listcomp>N)rz rY r� )r{ r, r, r- �
getServerlURLX s r� c C s t � }|jd| � dS )z Set serverURL in config r N)rz rw )r r{ r, r, r- � setServerURLe s r� c C s t � }|jd| � dS )z Set sslCACert in config r N)rz rw )r r{ r, r, r- � setSSLCACertj s r� �/etc/sysconfig/rhn/up2datec C sP yt W n tk
r da Y nX t dkrLt| �a dt d< tjj� rLdt d< t S )z<This function is the right way to get at the up2date config.NF�isattyT)r{ � NameErrorrm �sys�stdoutr� )Zcfg_filer, r, r- rz p s
rz )r r )r
r )r r
)r r )r r )r r )r r )r r )r r )r r )r r )r r )r r )r� )"rl r: r� rC r� Zrhn.connectionsr r Zrhn.i18nr r Zurlparser r � ImportErrorZ urllib.parse�gettextZ translation�t�hasattrr rU ro rZ r& rm r~ r� r� r� r� r� r� rz r, r, r, r- �<module> sp
O