File: //lib/python3.9/site-packages/up2date_client/__pycache__/config.cpython-39.pyc
a
c8h�9 � @ sf d Z ddlZddlZddlZddlZddlmZmZ ddlm Z m
Z
zddl m Z m
Z
W n" eyz ddlm Z m
Z
Y n0 ddlZejddd�Zeed �s�eje_ejZd
d d d
dddddddddddg d�fd�Zg d�ZG d d � d �ZG d d � d �Zd d!� Zd"d#� Zd$d%� Z d&d'� Z d(d)� Z d:d+d,�Z d;d-d.�Z d<d/d0�Z!d1d2� Z"d3d4� Z#d5d6� Z$d=d8d9�Z%dS )>zj
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)�fallback�ugettext)zUse a HTTP Proxyr )zRemote server URLz)https://xmlrpc.cln.cloudlinux.com/XMLRPC/)z&Remote server URL for access over IPv6z.https://ipv6.xmlrpc.cln.cloudlinux.com/XMLRPC/)zMirror list URL�>https://repo.cloudlinux.com/cloudlinux/mirrorlists/cln-mirrors)z#Whether or not debugging is enabledr )zLocation of system idz/etc/sysconfig/rhn/systemid)z4Override the automatically determined system version� )z;HTTP proxy in host:port format, e.g. squid.example.com:3128r )z'The username for an authenticated proxyr )z.The password to use for an authenticated proxyr )z$To use an authenticated proxy or notr )zBNumber of attempts to make at network connections before giving up� )z)The CA cert used to verify the ssl serverz'/usr/share/rhn/CLN-ORG-TRUSTED-SSL-CERT)zDisable the reboot actionr zDConfig 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 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| �� d S �N)�dict�fileName�load��self�filename� r �9/usr/lib/python3.9/site-packages/up2date_client/config.py�__init__C s zConfigFile.__init__c C s� |r
|| _ | j d u rd S t�| j tj�s,d S t| j d�}d}|�� D �]�}|�d�dkrZqD||�� }|slqD|d dkr�|d d� �� }qDnd}|�dd�}t |�d kr�qD|d �� }t
|d �� �}d }|�d
�} | dkr�|d | � }|}d }|�rr|�d �}
|d v �r t |� t
�� ��}nTt |
�dk�rPz t|�}W n t�yL Y n0 n"|
d dk�rdg }nd
d� |
D �}||f} || jv �r�| j| } |d u�r�|| d f} |d u�r�| d |f} | | j|< qD|�� d S )N�rr �#r ����\�=r � z [comment]�;)r r c S s g | ]}|� � r|� � �qS r )�strip)�.0�valr r r! �
<listcomp>� � z#ConfigFile.load.<locals>.<listcomp>)r �os�access�R_OK�open� readlines�findr* �split�lenr r �encode�locale�getpreferredencoding�int�
ValueErrorr �close) r r �fZ multiline�liner5 �key�value�comment�pos�valuesZnewvalr r r! r I sd
zConfigFile.loadc C sF | j d u rd S t�| j tj�sTt�tj�| j �tj�sTttdtj�| j � �� d S t| j d d�}t� | j d t
dd�� |� d� |� d� |� d� | j �
� D ]�}| j | \}}|� td ||f �� t|�tg �kr�|g}|tv r�ttjj|�}|� td
|d �tt|��f �� |� d � q�|�� t�| 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
r) �
)r r/ r0 r1 �path�dirname�print�_r2 �chmodr: �writer �keysr �type� FileOptions�map�abspath�join�strr< �rename)r r= r? rA r@ r r r! �save� s,
" zConfigFile.savec C s
|| j v S r �r �r �namer r r! � __contains__� s zConfigFile.__contains__c C s || v S r r rW r r r! �has_key� s zConfigFile.has_keyc C s
| j �� S r )r rM �r r r r! rM � s zConfigFile.keysc C s dd� | j �� D �S )Nc S s g | ] }|d �qS )r r )r+ �ar r r! r- � r. z%ConfigFile.values.<locals>.<listcomp>)r rC r[ r r r! rC � s zConfigFile.valuesc C s | j �|� d S r )r �update)r r r r r! r] � s zConfigFile.updatec C s || j v r| j | d S d S )Nr rV rW r r r! � __getitem__� s
zConfigFile.__getitem__c C s0 || j v r| j | }nd}|d |f| j |< d S )N)NNr rV )r rX r@ r, r r r! � __setitem__� s
zConfigFile.__setitem__c C s || j v r| j | d S dS )Nr r rV rW r r r! �info� s
zConfigFile.info)N)N)�__name__�
__module__� __qualname__�__doc__r" r rU rY rZ rM rC r] r^ r_ r` r r r r! 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 � | _| j�t� |r$| j�|� i | _d S r )r �storedr] �Defaultsr �runtimer r r r! r" � s
zConfig.__init__c C s || j v rdS || jv r dS dS )NTF�rh rf rW r r r! rY � s
zConfig.__contains__c C s || v S r r rW r r r! rZ � s zConfig.has_keyc C s4 t | j�� �}| j�� D ]}||vr|�|� q|S r )�listrh rM rf �append�r �ret�kr r r! rM � s
z Config.keysc C s& g }| � � D ]}|�| �|�� q |S r �rM rk r^ rl r r r! rC � s z
Config.valuesc C s* g }| � � D ]}|�|| �|�f� q |S r ro rl r r r! �items� s z Config.itemsc C s t | �� �S r )r6 rM r[ r r r! �__len__ s zConfig.__len__c C s || j |< d S r )rh �r rX r@ r r r! r_ s zConfig.__setitem__c C s, || j v r| j | S || jv r(| j| S d S r ri rW r r r! r^
s
zConfig.__getitem__c C s | j �|�S r )rf r` rW r r r! r` s z Config.infoc C s | j �� d S r )rf rU r[ r r r! rU s z Config.savec C s4 | j �|� | j �� D ]}|| jvr&q| j|= qd S r )rf r rM rh )r r rn r r r! r s
z Config.loadc C s || j |< || jv r | j|= d S r )rf rh rr r r r! �set# s
z
Config.set)N)ra rb rc r" rY rZ rM rC rp rq r_ r^ r` rU r rs r r r r! re � s
re c C s^ t � } d}| d }|rZ|dd� dkr2|dd� }|�d�}tt|d ��|d< d�|�}|S )zkreturns proxy string in format hostname:port
hostname is converted to Punycode (RFC3492) if needed
Nr � zhttp://�:r )�initUp2dateConfigr5 rS r rR )�cfg�proxyZ proxyHost�partsr r r! �getProxySetting* s
rz 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 r( � � )r r r r r ��url�sr r r! �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 r( r{ r| )r r r r r} r r r! �convert_url_from_punyB s r� c C s� t d } | d u rd} | �d�rnt| �dd�d��.}ttj|�� �}dd� |D �W d � S 1 sd0 Y t� | �}dd� |j
� d �D �S )
Nr r
zfile://r r# c S s g | ]}|rt |��qS r �r� �r+ Zmirrorr r r! r- O r. z+getServerlURLFromMirror.<locals>.<listcomp>c S s g | ]}|rt |��qS r r� r� r r r! r- Q r. rF ) rw �
startswithr2 �replacerP rS r* r3 �requests�get�textr5 )r~ Z
mirrorlistZmirrorsZrequestr r r! �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� )r+ �ir r r! r- ^ r. z$processServerURL.<locals>.<listcomp>N)�
isinstancerj r� )Z serverUrlr r r! �processServerURLT s
r� Fc C s^ t � }t|d �}t|d �}||f}||f}| r6|S ddlm} |�� }|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)rv r� Zup2date_clientr� ZgetPreferredInterface)� registrationrw Zipv4_urlZipv6_urlZipv4_primary_pairZipv6_primary_pairr� Zpreferred_interfacer r r! �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 r! � 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 r! �getFallbackServerURL� s r� c C s t � }|�d| � dS )zSet serverURL in configr N�rv rs �r rw r r r! � setServerURL� s r� c C s t � }|�d| � dS )zSet serverURLipv6 in configr Nr� r� r r r! �setServerURLipv6� s r� c C s t � }|�d| � dS )zSet sslCACert in configr
Nr� )r
rw r r r! � setSSLCACert� s r� �/etc/sysconfig/rhn/up2datec C sN zt W n ty da Y n0 t dkrJt| �a dt d< tj�� rJdt d< t S )z<This function is the right way to get at the up2date config.NF�isattyT)rw � NameErrorre �sys�stdoutr� )Zcfg_filer r r! rv � s
rv )F)F)F)r� )&rd r/ r� r8 r� Zrhn.connectionsr r Zrhn.i18nr r �urlparser r � ImportError� urllib.parse�gettext� translation�t�hasattrr rJ rg rO r re rz r� r� r� r� r� r� r� r� r� r� rv r r r r! �<module> sb
�� O
'