3

��`Y/�@s4
dZd
dl
Z
d
dlZd
dlZd
dlmZmZ
d
dlmZm	Z	
yd
dl
mZmZ
Wn$e
k
rt


d
dlmZmZ
YnXd
dlZejddd�Zeed	�s�eje_ejZd6d7d8d9d:d;d<d=d>d?d@dAdddddgfd �
Zd!dd"gZGd#d$�d$�ZGd%d&�d&�Zd'd(�Zd)d*�Zd+d,�Zd-d.�Zd/d0�Zd1d2�ZdBd4d5�
ZdS)Czj
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://your.server.url.here/XMLRPC�#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/RHN-ORG-TRUSTED-SSL-CERT�Disable the reboot actionzDConfig options that can not be overwritten by a config update action�	sslCACert�	serverURL�disallowConfChanges�noReboot)
ZenableProxyr�debug�systemIdPathZversionOverride�	httpProxy�	proxyUser�
proxyPasswordZenableProxyAuthZnetworkRetriesrrrrZtmpDirc@sleZ
dZd
Zddd�
Zddd�
Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�Zdd�Zdd�Z
dd�ZdS)�
ConfigFilez;class for handling persistent config options for the clientNcCsi|_|
|_
|j
r|j�
dS)
N)�dict�fileName�load)�self�filename�r)�/usr/lib/python3.6/config.py�__init__Gs



zConfigFile.__init__cCs�
|
r
|
|_|jdkrdSt
j|jt
j�s,dSt|jd
�}d}�
x�|j�D�
]�}|jd�
dkr^qH||j�}|spqH|ddkr�|dd�j�}qHnd}|jdd�}t	|�
dkr�qH|dj�}t
|dj��
}d}|jd	�
}	|	dkr�|d|	�}|}d}|�
rx|jd
�
}
|dk�
r"t|jt
j��
�
}nVt	|
�
dk�
rVyt|�
}Wntk
�
rR


YnXn"|
ddk�
rjg}nd
d�|
D�
}||f}||jk�
r�|j|}|dk	�
r�||df}|dk	�
r�|d|f}||j|<qHW|j�
dS)N�
rr�
#r
r�
\�
=�z	[comment]�
;r!r"c
Ssg|]}
|
j�r|
j��qSr))
�strip)�.0�valr)r)r*�
<listcomp>�sz#ConfigFile.load.<locals>.<listcomp>���r6r6)r!r")r%�os�access�R_OK�open�	readlines�findr2�split�lenrr�encode�locale�getpreferredencoding�int�
ValueErrorr$�close)r'r(�
fZ	multiline�liner=�key�value�comment�pos�valuesZnewvalr)r)r*r&Msd











































zConfigFile.loadc
	CsJ
|jdkrdSt
j|jt
j�sTt
jt
jj|j�
t
j�sTttd
t
jj|j�
�
�

dSt|jdd�}
t
j	|jdt
dd��
|
jd�

|
jd�

|
jd�

x�|jj
�D]�}|j|\}}|
jtd	||f�
�

t|�
tg�
kr�|g
}|tkr�tt
jj|�}|
jtd
|djtt|��
f�
�

|
jd�

q�W|
j�
t
j|jd|j�
dS)
Nz%s was not foundz.new�
wZ0644�zI# Automatically generated Red Hat Update Agent config file, do not edit.
z# Format: 1.0
rz%s[comment]=%s
z%s=%s
r1�

)r%r7r8r9�path�dirname�print�
_r:�chmodrB�writer$�keysr�type�FileOptions�map�abspath�join�strrD�rename)r'rErGrIrHr)r)r*�save�s,


	















"


zConfigFile.savecCs
|
|jkS)
N)
r$)r'�namer)r)r*�__contains__�s
zConfigFile.__contains__cCs|
|kS)
Nr))r'r^r)r)r*�has_key�szConfigFile.has_keyc


Cs
|jj
�S)
N)r$rU)
r'r)r)r*rU�s
zConfigFile.keysc

Csd
d�|jj
�D�
S)Nc
Ssg|]}
|
d�qS)
rr))r3�
ar)r)r*r5�sz%ConfigFile.values.<locals>.<listcomp>)r$rK)
r'r)r)r*rK�s
zConfigFile.valuescCs|jj
|
�

dS)
N)r$�update)r'r$r)r)r*rb�s
zConfigFile.updatecCs|
|jkr|j|
d
SdS)Nr)
r$)r'r^r)r)r*�__getitem__�s



zConfigFile.__getitem__cCs0|
|jkr|j|
}nd}|d
|f|j|
<dS)Nr
)NN)
r$)r'r^rHr4r)r)r*�__setitem__�s



zConfigFile.__setitem__cCs|
|jkr|j|
d
SdS)Nr
r)
r$)r'r^r)r)r*�info�s



zConfigFile.info)
N)
N)�__name__�
__module__�__qualname__�__doc__r+r&r]r_r`rUrKrbrcrdrer)r)r)r*r#Es



H$r#c@sveZ
dZddd�
Zdd�Zdd�Zdd	�Zd
d�Zdd
�Zdd�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zd
S)�ConfigNcCs.t�|_
|j
jt�

|
r$|j
j|
�

i|_dS)
N)r#�storedrb�Defaultsr&�runtime)r'r(r)r)r*r+�s





zConfig.__init__cCs |
|jkrd
S|
|j
krd
SdS)NTF)rmrk)r'r^r)r)r*r_�s







zConfig.__contains__cCs|
|kS)
Nr))r'r^r)r)r*r`�szConfig.has_keyc
Cs8t|j
j��
}
x$|jj�D]}||
kr|
j|�

qW|
S)
N)�listrmrUrk�append)r'�ret�
kr)r)r*rU�s





zConfig.keysc
Cs*g}
x |j�D]}|
j
|j|�
�

qW|
S)
N)rUrorc)r'rprqr)r)r*rK�s



z
Config.valuesc
Cs.g}
x$|j�D]}|
j
||j|�
f�

qW|
S)
N)rUrorc)r'rprqr)r)r*�items
s



zConfig.itemsc

Cst|j
��
S)
N)r>rU)
r'r)r)r*�__len__
s
zConfig.__len__cCs||j|
<dS)
N)
rm)r'r^rHr)r)r*rd
s
zConfig.__setitem__cCs,|
|jkr|j|
S|
|j
kr(|j
|
SdS)
N)rmrk)r'r^r)r)r*rc
s









zConfig.__getitem__cCs|jj
|
�
S)
N)rkre)r'r^r)r)r*re
s
zConfig.infoc


Cs|jj
�
dS)
N)rkr])
r'r)r)r*r]
s
zConfig.savecCs8|jj
|
�

x&|jj�D]}||jkr(q|j|=qWdS)
N)rkr&rUrm)r'r(rqr)r)r*r&
s




zConfig.loadcCs ||j|
<|
|j
kr|j
|
=dS)
N)rkrm)r'r^rHr)r)r*�set(
s



z
Config.set)
N)rfrgrhr+r_r`rUrKrrrsrdrcrer]r&rtr)r)r)r*rj�s

		
rjcCs^t�}d
}
|d}|rZ|d
d�dkr2|dd
�}|j
d�
}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[rrZ)�cfg�proxyZ	proxyHost�partsr)r)r*�getProxySetting/
s








r{c
Cs:t|�
}
t
t|
d
tt|
d�
�
|
d|
d|
df�
�
S)z? returns url where hostname is converted to Punycode (RFC3492) r
rr0��)rrrrr)�url�
sr)r)r*�convert_url_to_puny@
s
r�c
Cs6t|�
}
t
t|
d
t|
d�
|
d|
d|
df�
�
S)zZ returns url where hostname is converted from Punycode (RFC3492). Returns unicode string. r
rr0r|r})rrrr)r~rr)r)r*�convert_url_from_punyE
s
r�c
Cs>t�}t
|d
�
t
g�
kr,dd�|d
D�
St|d
�
g
SdS)z� return list of serverURL from config
        Note: in config may be one value or more values, but this
        function always return list
    rc
Ssg|]}
t|
�
�qSr))
r�)r3�
ir)r)r*r5S
sz!getServerlURL.<locals>.<listcomp>N)rwrVr�)
rxr)r)r*�
getServerlURLJ
s
r�c
Cst�}
|
j
d
|�
dS)z Set serverURL in config rN)rwrt)rrxr)r)r*�setServerURLW
s
r�c
Cst�}
|
j
d
|�
dS)z Set sslCACert in config rN)rwrt)rrxr)r)r*�setSSLCACert\
s
r��/etc/sysconfig/rhn/up2datec

CsPyt
Wnt
k
r 


d
aYnXtd
krLt|�
adtd<tjj�rLdtd<tS)z<This function is the right way to get at the up2date config.NF�isattyT)rx�	NameErrorrj�sys�stdoutr�)
Zcfg_filer)r)r*rwb
s








rw)r	r
)r
r)rr
)r
r)rr)rr)rr)rr)rr
)rr)rr)rr
)
r�) rir7r�r@Zrhn.connectionsrrZrhn.i18nrrZurlparserr�ImportErrorZurllib.parse�gettextZtranslation�
t�hasattrrrRrlrWr#rjr{r�r�r�r�r�rwr)r)r)r*�<module>sh


































O