| Current File : //lib/python3.6/site-packages/up2date_client/__pycache__/config.cpython-36.pyc |
3
c8h�9 � @ sn 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_ejZdEdFdGdHdIdJdKdLdMdNdOdPdQdRd d d!d"d#gfd$�Zd%d d&gZG d'd(� d(�ZG d)d*� d*�Zd+d,� Zd-d.� Zd/d0� Z d1d2� Z d3d4� Z dSd6d7�Z dTd8d9�Z dUd:d;�Z!d<d=� Z"d>d?� Z#d@dA� Z$dVdCdD�Z%dS )Wzj
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/�&Remote server URL for access over IPv6�.https://ipv6.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 �
serverURLipv6� 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__C 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/ r0 �
<listcomp>� s z#ConfigFile.load.<locals>.<listcomp>���r< r< )r' r( )r+ �os�access�R_OK�open� readlines�findr8 �split�lenr r �encode�locale�getpreferredencoding�int�
ValueErrorr* �close) r- r. �fZ multiline�linerC �key�value�comment�pos�valuesZnewvalr/ r/ r0 r, I 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
r7 �
)r+ r= r> r? �path�dirname�print�_r@ �chmodrH �writer* �keysr �type� FileOptions�map�abspath�join�strrJ �rename)r- rK rM rO rN r/ r/ r0 �save� s,
"zConfigFile.savec C s
|| j kS )N)r* )r- �namer/ r/ r0 � __contains__� s zConfigFile.__contains__c C s || kS )Nr/ )r- rd r/ r/ r0 �has_key� s zConfigFile.has_keyc C s
| j j� S )N)r* r[ )r- r/ r/ r0 r[ � s zConfigFile.keysc C s dd� | j j� D �S )Nc S s g | ] }|d �qS )r r/ )r9 �ar/ r/ r0 r; � s z%ConfigFile.values.<locals>.<listcomp>)r* rQ )r- r/ r/ r0 rQ � s zConfigFile.valuesc C s | j j|� d S )N)r* �update)r- r* r/ r/ r0 rh � s zConfigFile.updatec C s || j kr| j | d S d S )Nr )r* )r- rd r/ r/ r0 � __getitem__� s
zConfigFile.__getitem__c C s0 || j kr| j | }nd}|d |f| j |< d S )Nr )NN)r* )r- rd rN r: r/ r/ r0 � __setitem__� s
zConfigFile.__setitem__c C s || j kr| j | d S dS )Nr r )r* )r- rd r/ r/ r0 �info� s
zConfigFile.info)N)N)�__name__�
__module__� __qualname__�__doc__r1 r, rc re rf r[ rQ rh ri rj rk r/ r/ r/ r0 r) @ 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) �storedrh �Defaultsr, �runtime)r- r. r/ r/ r0 r1 � s
zConfig.__init__c C s || j krdS || jkr dS dS )NTF)rs rq )r- rd r/ r/ r0 re � s
zConfig.__contains__c C s || kS )Nr/ )r- rd r/ r/ r0 rf � s zConfig.has_keyc C s8 t | jj� �}x$| jj� D ]}||kr|j|� qW |S )N)�listrs r[ rq �append)r- �ret�kr/ r/ r0 r[ � s
z Config.keysc C s* g }x | j � D ]}|j| j|�� qW |S )N)r[ ru ri )r- rv rw r/ r/ r0 rQ � s z
Config.valuesc C s. g }x$| j � D ]}|j|| j|�f� qW |S )N)r[ ru ri )r- rv rw r/ r/ r0 �items� s z Config.itemsc C s t | j� �S )N)rD r[ )r- r/ r/ r0 �__len__ s zConfig.__len__c C s || j |< d S )N)rs )r- rd rN r/ r/ r0 rj s zConfig.__setitem__c C s, || j kr| j | S || jkr(| j| S d S )N)rs rq )r- rd r/ r/ r0 ri
s
zConfig.__getitem__c C s | j j|�S )N)rq rk )r- rd r/ r/ r0 rk s z Config.infoc C s | j j� d S )N)rq rc )r- r/ r/ r0 rc s z Config.savec C s8 | j j|� x&| j j� D ]}|| jkr(q| j|= qW d S )N)rq r, r[ rs )r- r. rw r/ r/ r0 r, s
z Config.loadc C s || j |< || jkr | j|= d S )N)rq rs )r- rd rN r/ r/ r0 �set# s
z
Config.set)N)rl rm rn r1 re rf r[ rQ rx ry rj ri rk rc r, rz r/ r/ r/ r0 rp � s
rp c C s^ t � } d}| d }|rZ|dd� dkr2|dd� }|jd�}tt|d ��|d< dj|�}|S )zkreturns proxy string in format hostname:port
hostname is converted to Punycode (RFC3492) if needed
Nr& � zhttp://�:r )�initUp2dateConfigrC ra r r` )�cfg�proxyZ proxyHost�partsr/ r/ r0 �getProxySetting* 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 r6 � � )r r r r r )�url�sr/ r/ r0 �convert_url_to_puny< s r� c C s6 t | �}tt|d t|d �|d |d |d f��S )zXreturns url where hostname is converted from Punycode (RFC3492). Returns unicode string.r r r6 r� r� )r r r r )r� r� r/ r/ r0 �convert_url_from_punyB 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 r2 c S s g | ]}|rt |��qS r/ )r� )r9 �mirrorr/ r/ r0 r; O s z+getServerlURLFromMirror.<locals>.<listcomp>c S s g | ]}|rt |��qS r/ )r� )r9 r� r/ r/ r0 r; Q s rT ) r~ �
startswithr@ �replacer^ ra r8 rA �requests�get�textrC )r� Z
mirrorlistZmirrorsZrequestr/ r/ r0 �getServerlURLFromMirrorH s
r� c C s& t | t�rdd� | D �S t| �gS dS )z�
Internal function to process server URL to Punycode format.
Processes both single URLs and lists of URLs.
:param serverUrl: URL or list of URLs to process.
:return: List of processed URLs in Punycode format.
c S s g | ] }t |��qS r/ )r� )r9 �ir/ r/ r0 r; ^ s z$processServerURL.<locals>.<listcomp>N)�
isinstancert r� )Z serverUrlr/ r/ r0 �processServerURLT s
r� Fc C s^ t � }t|d �}t|d �}||f}||f}| r6|S ddlm} |j� }|dkrZ|rZ|S |S )a�
Return a pair of server URLs (primary and fallback) based on the preferred interface.
:param registration: If True, it indicates that the URLs are being fetched for registration purposes.
If that is the case, the function will always use IPv4 URLs with IPv6 as fallback,
regardless of the preferred_interface setting.
Assumed false by default, expected to be specified explicitly when needed.
:return: Pair of server URL configs, first is the primary, second is the fallback.
r r" r )� up2dateAuthZIPv6)r} r� Zup2date_clientr� ZgetPreferredInterface)� registrationr~ Zipv4_urlZipv6_urlZipv4_primary_pairZipv6_primary_pairr� Zpreferred_interfacer/ r/ r0 �getServerURLPairc s r� c C s t | d�d S )az
Return the primary server URL from config based on preferred_interface.
If preferred_interface=IPv6 in system_id, returns serverURLipv6 if available.
Otherwise returns normal serverURL.
Note: the config may have one value or a list of values, but this function always returns a list.
:return: List of server URLs with hostnames converted to Punycode.
)r� r )r� )r� r/ r/ r0 � getServerURL� s
r� c C s t | d�d S )z�
Determine the fallback server URL from system_id.
The fallback server URL is the one that is *not* set as preferred_interface.
If preferred_interface is IPv6, it returns serverURL (i.e. IPv4 host), otherwise serverURLipv6.
)r� r )r� )r� r/ r/ r0 �getFallbackServerURL� s r� c C s t � }|jd| � dS )zSet serverURL in configr N)r} rz )r r~ r/ r/ r0 � setServerURL� s r� c C s t � }|jd| � dS )zSet serverURLipv6 in configr" N)r} rz )r r~ r/ r/ r0 �setServerURLipv6� s r� c C s t � }|jd| � dS )zSet sslCACert in configr N)r} rz )r r~ r/ r/ r0 � setSSLCACert� 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~ � NameErrorrp �sys�stdoutr� )Zcfg_filer/ r/ r0 r} � s
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 )r r )F)F)F)r� )&ro r= r� rF r� Zrhn.connectionsr r Zrhn.i18nr r Zurlparser r � ImportErrorZ urllib.parse�gettextZ translation�t�hasattrr rX rr r] r) rp r� r� r� r� r� r� r� r� r� r� r� r} r/ r/ r/ r0 �<module> sb
O
'