3
e�$gϮ � @ s( d dl mZ d dlZd dlZejddd�Zeed�s<eje_ejZd dl Z d dl
Z
d dl m Z m
Z
d dl mZ d d l mZ d d
l mZ d d l mZ d d l mZ d d
lmZ d dl mZ d dlmZ d dlmZ d dlmZ d dl m Z m Z d dl T ej � Z!ej"� Z#dd� Z$dd� Z%dd� Z&dd� Z'd d � Z(G d d � d e)�Z*G d d!� d!�Z+G d"d#� d#�Z,G d$d%� d%�Z-G d&d'� d'�Z.G d(d)� d)�Z/G d*d+� d+�Z0G d,d-� d-�Z1G d.d/� d/�Z2G d0d1� d1�Z3G d2d3� d3�Z4G d4d5� d5�Z5G d6d7� d7�Z6G d8d9� d9�Z7G d:d;� d;�Z8G d<d=� d=�Z9d>d?� Z:e;d@k�r$e:� dS )A� )�geteuidNzrhn-client-toolsT)Zfallback�ugettext)�rhnreg�hardware)�
up2dateErrors)� up2dateUtils)�pkgUtils)�
up2dateLog)�config)�convert_url_from_puny)� up2dateAuth)�rpclib)�idn_puny_to_unicode)�sstr)�PM_PLUGIN_NAME�PM_PLUGIN_CONF)�*c C s& t j| tt�td| �tt�g� d S )Nz%s)�snack�ButtonChoiceWindowr �ERROR�BACK)�screen�errmsg� r �/usr/lib/python3.6/tui.py� ErrorWindow/ s r c C s8 t j| tt�td| �tt�g� | j� tjd� d S )Nz%s� )r r r Z FATAL_ERROR�OK�finish�sys�exit)r r r r r �FatalErrorWindow3 s r! c C s. t j| tt�td| �tt�g� | j� d S )Nz%s)r r r �WARNINGr r )r r r r r �
WarningWindow9 s r# c C s� t j| tt�tt�d tt� d tt� d tt� d tt� d tt � d tt
� d tt � d tt � d tt
� d tt� tt�tt�gdd�}|tt�j� kr�| j� dS dS d S )N�
z
�F )�widthr r )r r r Z CONFIRM_QUITZCONFIRM_QUIT_SURE�WHY_REGISTER_SEC�WHY_REGISTER_SEC_TXT�WHY_REGISTER_DLD�WHY_REGISTER_DLD_TXT�WHY_REGISTER_SUPP�WHY_REGISTER_SUPP_TXT�WHY_REGISTER_COMP�WHY_REGISTER_COMP_TXTZCONFIRM_QUIT_WILLNOT�WHY_REGISTER_TIPZCONTINUE_REGISTERINGZREGISTER_LATER2�lowerr )r �buttonr r r �ConfirmQuitWindow? s
vr2 c
G s� y ||� }W n� t jk
rD t| ttj� d � tj� d �Y n� t jk
rp t| |j � tj� d �Y n� t j
k
r� tj� d }t | |j d t t
j� t
j� t
j� f � Y n: t jk
r� tj� d }t| |j d t � |�Y nX |S )Nr r r$ z
)r �CommunicationErrorr ZHOSTED_CONNECTION_ERRORr
�
getServerlURLr �exc_infoZ SSLCertificateVerifyFailedErrorr ZNoBaseChannelErrorr! ZBASECHANNELERRORr �getArchZ getOSRelease�
getVersionZSSLCertificateFileNotFoundZ SSL_CERT_FILE_NOT_FOUND_ERRER)r �funcZparams�results�er r r �tui_call_wrapperV s(
r; c @ s e Zd ZdS )�WindowSkipExceptionN)�__name__�
__module__� __qualname__r r r r r< n s r< c @ s( e Zd Zd Zdd� Zdd� Zdd� ZdS )�AlreadyRegisteredWindowc C s< t j� s|jrt� �|| _|| _tjj� }t j
j t j
� �}|d d d }|d d d }tj| jtt�dd�}tj| jtt�dftt�dfg�| _|j| jdddd� tj|d d |d d
ttd td � d
t| jj� d td� d
| d td� d
| d t d �dd�}|j|dddd� || _ d S )Nr Zusername� system_idr � �nextr )�growx� � z
zSpacewalk Location:� r$ zLogin:z
System ID:)�padding)r r r r ) r Z
registered�testr< r �tuir �_snack�sizer
Z xmlrpclib�loadsr Z getSystemId�GridFormr �SYSTEM_ALREADY_SETUP� ButtonBar�YES_CONT� NO_CANCEL�bb�add�TextboxZSYSTEM_ALREADY_REGISTERED�_r � serverURL� SYSTEM_ALREADY_REGISTERED_CONT�g) �selfr rJ rL Z systemIdXmlZ oldUsernameZ oldsystemId�toplevel�tbr r r �__init__t s&
Tz AlreadyRegisteredWindow.__init__c C s d S )Nr )rZ r r r � saveResults� s z#AlreadyRegisteredWindow.saveResultsc C s6 t jd| j � | jj� }| jj|�}|dkr2dS |S )Nz
Running %s�F12rC )�log� log_debug�namerY �runOncerS �
buttonPressed)rZ �resultr1 r r r �run� s
zAlreadyRegisteredWindow.runN)r= r> r? rb r] r^ rf r r r r r@ q s r@ c @ s0 e Zd Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�SatelliteUrlWindowc C s� || _ || _d| j_t| jj�| _tj| j�}|| jkrR|| _tj | j� t
j � t j
j� }t j|tt�dd�}t}t}t}t j|d d dt|�ddd�} |j| dddd� t jtd��} |j| dd� t jd d�}
t jt|��} |
j| ddddd
� t jd �| _| jj | j� |
j| jdddd� t jt|��} |
j| ddddd
� t jd �| _ | j j |j � |
j| j dddd� |j|
dd � t j |tt �d ftt!�d
ftt"�dfg�| _#|j| j#ddddd� || _$d S )Nr r � �
� )�scroll�wrap)�
anchorLeft� rB )rH � anchorRight�( rC �back�cancel)rH rD )r r r r )r r r r )r r r r )%r rJ �alreadyRegisteredr rW �serverr �makeNiceServerUrlr
� setServerURL�cfg�saver rK rL rN r ZSATELLITE_URL_WINDOWZSATELLITE_URL_TEXTZSATELLITE_URL_PROMPTZSATELLITE_URL_PROMPT2rU rT �Label�Grid�setField�Entry�urlEntry�set�sslEntry� sslCACertrP �NEXTr �CANCELrS rY ) rZ r rJ �fixed_server_urlrL r[ Z prompt_textZ url_labelZ ssl_label�label�gridr r r r] � sT
zSatelliteUrlWindow.__init__c C s� | j j� dkrBtj| jtt�tt�tt�gd� | j j
| j � dS | j j� d d� dkr�| j j� dkr�tj| jtt�tt �tt�gd� | j j
| j � dS dS )Nrn )�buttonsr � Zhttpsr )
r} �valuer r r r r ZSATELLITE_REQUIREDr rY �
setCurrentr Z SSL_REQUIRED)rZ r r r �validateFields� s z!SatelliteUrlWindow.validateFieldsc C s\ | j j� }tj|�}||kr |}|| j_| jj� | j_tj |� tj
| jj� � t j � d S )N)
r} r� r ru rJ rW r r� r
rv Z setSSLCACertrw rx )rZ Z serverEntryr� r r r r^ � s
z SatelliteUrlWindow.saveResultsc C sl t jd| j � | jj� d}x>|s\| jj� }| jj|�}|dkrFd}|dkrX| j � }q P q W | jj
� |S )Nz
Running %sr r_ rC ) r` ra rb r �refreshrY rf rS rd r� � popWindow)rZ �validre r1 r r r rf s
zSatelliteUrlWindow.runN)r= r> r? rb r] r� r^ rf r r r r rg � s
A
rg c @ s( e Zd Zd Zdd� Zdd� Zdd� ZdS )�*AlreadyRegisteredSubscriptionManagerWindowc C s� t j� s|jrt� �|| _|| _tjj� }tj | jt
t �dd�}tj | jt
t
�dft
t�dfg�| _|j| jdddd� tj|d d |d d t
td t d t d
�dd�}|j|ddd d � || _d S )
Nr rB rC r r )rD rE rF z
r$ )rH )r r r r )r Zrhsm_registeredrI r< r rJ r rK rL rN r rO rP rQ rR rS rT rU r" Z RHSM_SYSTEM_ALREADY_REGISTEREDrX rY )rZ r rJ rL r[ r\ r r r r] s
z3AlreadyRegisteredSubscriptionManagerWindow.__init__c C s d S )Nr )rZ r r r r^ 2 s z6AlreadyRegisteredSubscriptionManagerWindow.saveResultsc C s6 t jd| j � | jj� }| jj|�}|dkr2dS |S )Nz
Running %sr_ rC )r` ra rb rY rc rS rd )rZ re r1 r r r rf 5 s
z.AlreadyRegisteredSubscriptionManagerWindow.runN)r= r> r? rb r] r^ rf r r r r r� s r� c @ s( e Zd Zd Zdd� Zdd� Zdd� ZdS )�
ConnectWindowc C s� || _ || _tjj� }t| jj�| _td | _ tj
| j t t �dd�}t
| j d }| j rf|t| j 7 }tj|d d |d d t |�dd�}|j|dddd� || _d S ) NZ httpProxyr z
r rE rF )rH )r r r r )r rJ r rK rL r rW rt rw �proxyrN r ZCONNECT_WINDOWZCONNECT_WINDOW_TEXTZCONNECT_WINDOW_TEXT2rU rT rY )rZ r rJ rL r[ �textr\ r r r r] C s
zConnectWindow.__init__c C sP t jd| j � | jj� }| jj� yt| jtj � W n
dS | jj
� dS )Nz
Running %srq rC ) r` ra rb rY �drawr r� r; r ZgetCapsr� )rZ re r r r rf \ s
zConnectWindow.runc C s d S )Nr )rZ r r r r^ q s zConnectWindow.saveResultsN)r= r> r? rb r] rf r^ r r r r r� @ s r� c @ s( e Zd Zd Zdd� Zdd� Zdd� ZdS )� StartWindowc C s� || _ || _tjj� }tj| j tt�dd�}tt�}tj |d d |d d |dd�}|j
|ddd d� tj | j tt �dftt
�dftt�d fg�| _|j
| jdddd
� || _d S ) Nr rB r ri � )rH � why_registerrC rr )rD )r r r r )r rJ r rK rL rN r ZSTART_REGISTER_WINDOWZSTART_REGISTER_TEXTrU rT rP Z WHY_REGISTERr� r� rS rY )rZ r rJ rL r[ Zstart_register_textr\ r r r r] w s
"
zStartWindow.__init__c C s d S )Nr )rZ r r r r^ � s zStartWindow.saveResultsc C sX t jd| j � | jj� }| jj|�}|dkr2dS |dkrTt| j| j �}|j
� |S |S )Nz
Running %sr_ rC r� ) r` ra rb rY rc rS rd �WhyRegisterWindowr rJ rf )rZ re r1 Z why_reg_winr r r rf � s
zStartWindow.runN)r= r> r? rb r] r^ rf r r r r r� t s r� c @ s e Zd Zd Zdd� Zdd� ZdS )r� c C s� || _ || _tjj� }tj| j tt�dd�}td t d t
d t d t d t
d t d t d t d t }tj|d d |d d t|�dd�}|j|ddd d� tj| j tt�d fg�| _|j| jdddd
� || _d S ) Nr rB z
r$ r ri r� )rH rq )rD )r r r r )r rJ r rK rL rN r ZWHY_REGISTER_WINDOWZWHY_REGISTER_TEXTr' r( r) r* r+ r, r- r. r/ rU rT rP Z
BACK_REGISTERrS rY )rZ r rJ rL r[ Zwhy_register_textr\ r r r r] � s
F&zWhyRegisterWindow.__init__c C s* t jd| j � | jj� }| jj|�}|S )Nz
Running %s)r` ra rb rY rc rS rd )rZ re r1 r r r rf � s
zWhyRegisterWindow.runN)r= r> r? rb r] rf r r r r r� � s r� c @ s0 e Zd Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�
InfoWindowc C s || _ || _d| j_| jj| _tjj� }tj|t t
�dd�}t | j�}| j}|| jkrb|d| 7 }t | }t
}t} tj|d d dt |�ddd�}
|j|
dddd� tjd d�} tjt |��}
| j|
ddddd
� tjd �| _| jj|j� | j| jdddd� tjt t��}
| j|
ddddd
� ytjd dd �| _W n& tk
�r\ tjd dd
�| _Y nX | jj|j� | j| jdddd� |j| dd� tj |d d t | ��}
|j|
dd dd� tj |t t �dft t �dft t �dfg�| _!|j| j!ddddd� || _"d S )Nr r rh z (%s)ri rj )rk rl )rm rB )rH ro rF )�password)ZhiddenrC rq rr )rH rD )r r r r )r r r r )r r r r )#r rJ rs rW rt r rK rL rN r ZREGISTER_WINDOWr Z LOGIN_PROMPTZLOGINZ LOGIN_TIPrU rT rz ry r{ r| �
userNameEntryr~ �userNameZPASSWORD�
passwordEntry� TypeErrorr� �TextboxReflowedrP r� r r� rS rY ) rZ r rJ rL r[ Zdecoded_serverZurlZ login_promptZ login_labelZ login_tipr� r� r r r r] � sV
zInfoWindow.__init__c
C sL | j j� dkrBtj| jtt�tt�tt�gd� | j j
| j � dS | j j� dkr�tj| jtt�tt �tt�gd� | j j
| j � dS y t
j| j j� | j j� �| j_W n� tjk
�r tj� d }tj| jttd��ttd��t|j� ttd��gd� | j j
| j � dS tjk
�rF tj� d }t| jtd�|j � Y nX dS ) Nrn )r� r r �Errorz The server indicated an error:
r z?There was an error communicating with the registration server:
)r� r� r r r r r Z
USER_REQUIREDr rY r� r� ZPASSWORD_REQUIREDr Z reserveUserrJ rs r ZValidationErrorr r5 rV r r3 r! )rZ r: r r r r� s. 8 zInfoWindow.validateFieldsc C s | j j� | j_| jj� | j_d S )N)r� r� rJ r� r� r� )rZ r r r r^ % s zInfoWindow.saveResultsc C sl t jd| j � | jj� d}x>|s\| jj� }| jj|�}|dkrFd}|dkrX| j � }q P q W | jj
� |S )Nz
Running %sr r_ rC ) r` ra rb r r� rY rf rS rd r� r� )rZ r� re r1 r r r rf ) s
zInfoWindow.runN)r= r> r? rb r] r� r^ rf r r r r r� � s
@ r� c @ s0 e Zd Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�OSReleaseWindowc C sx || _ tj� s tjd� t� �tj|j|j�| _ t
| j d �dk rRtjd� t� �|| _ t j
j� | _d| _t j| j tt�dd�}|| _t j| jd d tt��| _|j| jdddd � tt�}| j jr�t j|d dd
�| _nt j|d �| _|j| jddd dd � t jtt ��| _ |j| j dd dd � t j | jd d
d| jd d d�| _ |j| j dd� xRt | j d j!� dd� d�D ]4\}}|| j d k�r�|d }| j j"d| |� �qtW t j| jd d tt#��| _$|j| j$dddd � tt%�}| j j&�rt j|| jdd
�| _&nt j|| j�| _&|j| j&ddd dd � t j'|tt(�dftt)�dftt*�dfg�| _+|j| j+dddd � | j j,� d S ) Nz5Server does not support EUS, skipping OSReleaseWindow�channelsr z3No available EUS channels, skipping OSReleaseWindowF� r ri )rm )�isOn)rH rm rB � )r& rj c S s | d S )Nr r )�ar r r �<lambda>j s z*OSReleaseWindow.__init__.<locals>.<lambda>)�keyZreceiving_updatesr rG rh r� rC rq rr � )rD )r r r r )r r r r )-rJ r Zserver_supports_eusr` ra r< ZgetAvailableChannelsr� r� �available_channels�lenr r rK rL Z
selectChannelrN r ZSELECT_OSRELEASErY r� Z
OS_VERSIONZostextrT ZLIMITED_UPDATES�limited_updates_buttonZSingleRadioButtonry Z
MINOR_RELEASEZsublabelZListbox� channelList�sorted�items�appendZCHANNEL_PAGE_TIPZtipZ ALL_UPDATES�all_updates_buttonrP r� r r� rS r� )rZ r rJ r[ Z optiontext1r� r� Z optiontext2r r r r] @ sb
$
zOSReleaseWindow.__init__c C sp t jd| j � | jj� d}xB|dkr`| jj� }| jj|�}|dkrJd}|dkr\| j � }q P q W | jj
� |S )Nz
Running %srr r_ rC ) r` ra rb r r� rY rf rS rd r� r� )rZ r� re r1 r r r rf � s
zOSReleaseWindow.runc C s� d}d}| j j� r&| jj� | jd k}tt�}|rdtj| j |tt
�| jj� tt �tt �gd�}|S | j
j� sr|r�tj| j |tt�tt �tt �gd�}|S |S )N�okFZdefault_channel)r� )r� �selectedr� �currentr� r Z CONFIRM_OS_RELEASE_SELECTIONr r r ZCONFIRM_OS_WARNINGr r� r� ZCONFIRM_OS_ALL)rZ ZmsgboxZ
later_release�titler r r r� � s
z OSReleaseWindow.validateFieldsc C sz | j jjdd � | jj� rVtjd| jj� � | jj� | j jd< | jj� | j _d| j _ | j j� rv| j j� | j _ d| j _d S )N�channelzSelected Channel %sr )
rJ �other�popr� r� r` ra r� r� r� )rZ r r r r^ � s
zOSReleaseWindow.saveResultsN)r= r> r? rb r] rf r� r^ r r r r r� = s
Mr� c @ s( e Zd Zd Zdd� Zdd� Zdd� ZdS )�HardwareWindowc C s� || _ || _tjj� }tj� \}}|d k rB|| jjd< || jjd< tj � |_tj
|t t �dd�}tj
dt t��}|j|dddd� tjdd�}tjt td ���} |j| ddd.dd
� tjd �| _|j| jdddd� |j|dddd� |j�rtjt t�dd �| _ntjt t��| _|j| jddd/dd
� tjt t��} |j| dddd0d� tjdd�}d}
|
td�t tj � � d 7 }
tjt td���| _ |j| j ddd1dd
� tjt tj � ��| _ |j| j dddd� |
td�7 }
x.|jD ]$} | d dk�r�|
| d d 7 }
�q�W |
td�7 }
xb|jD ]X} | d dk�rt | d �} |
| d 7 }
|j dk�r`| jj!|j � n| jj!t | �� �qW |
td �7 }
x6|jD ],} | d dk�r�|
td �| d d 7 }
�q�W |
td �7 }
xT|jD ]J} | d dk�r�| d �r�|
| d d 7 }
n| d! �r�|
| d! d 7 }
�q�W |
td"�7 }
x2|jD ](} | d d#k�r0|
td$�| d% 7 }
�q0W tj
d&t |
��}
|j|
dd� tj
|d d' t t"��| _#|j| j#dd(d2dd
� tj$|t t%�d)ft t&�d*ft t'�d+fg�| _(|j| j(dd,d3dd-� || _)d S )4N� virt_uuid� virt_typer r� r% r )rm rB z
Profile name:)rH ro rp )r� )rH rm rj )rm rH rh rn z Version: z z CPU model: �classZCPUZmodelr$ z
Hostname: ZNETINFOZhostnamez CPU speed: z%d MHzZspeedz IP Address: ZipaddrZip6addrzMemory: ZMEMORYz %s megabytesZram�P ri r� rC rq rr r� )rH rD )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 rJ r rK rL r Z
get_virt_infor� r ZHardwarerN r ZHARDWARE_WINDOWr� ZHARDWARE_WINDOW_DESC1rT rz ry rV r{ r| � profileEntry�includeHardware�CheckboxZHARDWARE_WINDOW_CHECKBOX�hardwareButtonZDESELECTr r7 Z versionLabelZ
versionLabel2r � profileNamer~ ZHARDWARE_WINDOW_DESC2ZadditionalHWLabelrP r� r r� rS rY )rZ r rJ rL r� r� r[ r� r� r� Z
hardware_textZhwZunicode_hostnamer\ r r r r] � s�
zHardwareWindow.__init__c C s | j j� | j_| jj� | j_d S )N)r� r� rJ r� r� r� r� )rZ r r r r^ 5 s zHardwareWindow.saveResultsc C s6 t jd| j � | jj� }| jj|�}|dkr2dS |S )Nz
Running %sr_ rC )r` ra rb rY rc rS rd )rZ re r1 r r r rf 9 s
zHardwareWindow.runN)r= r> r? rb r] r^ rf r r r r r� � s nr� c @ s0 e Zd Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�PackagesWindowc C s� || _ || _tjj� }tj|tt�dd�}|| _tj |d d tt
��}|j |dddd� tj tt
�d�| _|j | jddddd� tjtt��}|j |dddd� tj|d d d�| _|j | jdd � |jg k�rJtj|tt�dd�| _tjd
d �| _| jj | jdd� | jj� | j j� d}tjd �r2d}tj |d
�|_| j j � xL|jD ]B}| jj dt|d �t|d �t|d �f t|d �dd� �qRW tj |tt �dftt!�dftt"�dfg�| _#|j | j#ddddd� d S )Nr r� r ri )rm )rH rm rB � rj rp �d Z supportsExtendedPackageProfile)r6 z%s-%s-%srb �version�release)�itemr� rC rq rr rh )rH rD )r r r r )r r r r )$r rJ r rK rL rN r ZPACKAGES_WINDOWrY r� ZPACKAGES_WINDOW_DESC1rT r� ZPACKAGES_WINDOW_DESC2�packagesButtonry ZPACKAGES_WINDOW_UNCHECKZ CheckboxTree� packageListZPACKAGES_WINDOW_PKGLIST�pwin�Scale�scaler� r� r rw r ZgetInstalledPackageListr� r� rP r� r r� rS ) rZ r rJ rL r[ r� r� r6 �packager r r r] E sJ
zPackagesWindow.__init__c C s4 | j jt|d | d �� | jj� | jj� d S )Ng �?r� )r� r~ �intr� r� r r� )rZ �amount�totalr r r �setScale{ s
zPackagesWindow.setScalec C sH | j j� | j_| jj� }x*| jjD ] }|d |kr"| jjj|� q"W d S )Nrb )r� r� rJ �includePackagesr� Z getSelection�selectedPackagesr� )rZ Z selectionZpkgr r r r^ � s
zPackagesWindow.saveResultsc C s6 t jd| j � | jj� }| jj|�}|dkr2dS |S )Nz
Running %sr_ rC )r` ra rb rY rc rS rd )rZ re r1 r r r rf � s
zPackagesWindow.runN)r= r> r? rb r] r� r^ rf r r r r r� B s
6r� c @ s( e Zd Zd Zdd� Zdd� Zdd� ZdS )�
SendWindowc C s� || _ || _tjj� }tj|tt�dd�}tj|d d tt ��}|j
|dd� tj |tt �dftt
�dftt�dfg�| _|j
| jddd dd� || _d S )
Nr rB r � rC rq rr )rH rD )r r r r )r rJ r rK rL rN r Z SEND_WINDOWr� ZSEND_WINDOW_DESCrT rP r� r r� rS rY )rZ r rJ rL r[ r� r r r r] � s
zSendWindow.__init__c C s d S )Nr )rZ r r r r^ � s zSendWindow.saveResultsc C s6 t jd| j � | jj� }| jj|�}|dkr2dS |S )Nz
Running %sr_ rC )r` ra rb rY rc rS rd )rZ re r1 r r r rf � s
zSendWindow.runN)r= r> r? rb r] r^ rf r r r r r� � s r� c @ s0 e Zd Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�
SendingWindowc C sP || _ || _tjj� }tj|tt�dd�| _tj dd�| _
| jj | j
dd� d S )Nr rp r� r ) r rJ r rK rL rN r ZSENDING_WINDOWr� r� r� rT )rZ r rJ rL r r r r] � s
zSendingWindow.__init__c % C s� t jd| j � | jj� | jj� d }yPt jdt| jj � � t
j | jj | jj
| jj| jj d�}|j}t|d �}W �n( tjk
r� tj� d }t| jtd�|j � Y n� tjk
r� tj� d }t| jtd�|j � Y n� tjk
�r tj� d }t| jtd�|j � Y n� tjk
�rV tj� d }t| jtd�|j � Y nL tjk
�rx t| jt � Y n* t j tj� � t| jtd�� Y nX t
j |��s�t| jtd�� | j dd � t
j � | _!| j d
d � | jj"�rbyt
j#|| jj$� W nb tjk
�r8 tj� d }t| jtd �|j � Y n* t j tj� � t| jtd �� Y nX | j d
d � | jj%�r�yt
j&|| jj'� W nb tjk
�r� tj� d }t| jtd�|j � Y n* t j tj� � t| jtd�� Y nX d }y t(j)� }W n, tjk
�r, t| jtj� d � Y nX t
j*|� yt
j+� \| j_,| j_-W nH t.k
�r� tj� d }t/| jtd�t0t1f |j � d| j_2Y nX t
j3� | j d d � | jj4� || j_5dS )Nz
Running %sz other is %s)r� rA r z Problem registering system:
zProblem registering system.z&Problem writing out system id to disk.rh rB z"Problem sending hardware profile:
z!Problem sending hardware profile.rj z Problem sending package list:
z Problem sending package list.z%Could not open %s
%s is not enabled.
rC )6r` ra rb r� r� r r� �strrJ r� r ZregisterSystem2r� r� r� ZrawDictr r r3 r r5 r! rV r ZRhnUuidUniquenessErrorZInsuffMgmntEntsErrorZRegistrationDeniedErrorZ ActivationKeyUsageLimitErrorZACT_KEY_USAGE_LIMIT_ERRORZ
log_exceptionZ
writeSystemIdr� Z
getOemInfo�oemInfor� Z sendHardwarer r� Z sendPackagesr� r ZupdateLoginInfoZ sendVirtInfoZ pluginEnable�pm_plugin_present�pm_plugin_conf_changed�IOErrorr# r r �pm_plugin_conf_errorZspawnRhnCheckForUIr� �reg_info)rZ r� ZsystemIdr: Zlir r r rf � s�
zSendingWindow.runc C s d S )Nr )rZ r r r r^ . s zSendingWindow.saveResultsc C s4 | j jt|d | d �� | jj� | jj� d S )Ng �?r� )r� r~ r� r� r� r r� )rZ r� r� r r r r� 1 s
zSendingWindow.setScaleN)r= r> r? rb r] rf r^ r� r r r r r� � s
kr� c @ s( e Zd Zd Zdd� Zdd� Zdd� ZdS )� FinishWindowc C s� || _ || _tjj� }tj|tt�dd�}tj|d d tt ��}|j
|dd� tj |tt d��dfg�| _
|j
| j
ddddd� || _d S ) Nr rB r � ZFinishrC )rH rD )r r r r )r rJ r rK rL rN r Z
FINISH_WINDOWr� ZFINISH_WINDOW_TEXT_TUIrT rP rV rS rY )rZ r rJ rL r[ r� r r r r] : s
zFinishWindow.__init__c C s d S )Nr )rZ r r r r^ M s zFinishWindow.saveResultsc C s6 t jd| j � | jj� }| jj|�}|dkr2dS |S )Nz
Running %sr_ rC )r` ra rb rY rc rS rd )rZ re r1 r r r rf Q s
zFinishWindow.runN)r= r> r? rb r] r^ rf r r r r r� 7 s r� c @ s( e Zd Zd Zdd� Zdd� Zdd� ZdS )� ReviewWindowc
C s" || _ || _|j| _tjj� }tj|tt�dd�}d}| jj sJ|t
d 7 }| jj r^|t d 7 }| jj
rr|td 7 }|td 7 }t| jd �dkr�d}x | jd D ]}||d 7 }q�W td t d d }tjd | jj � |t7 }||| d 7 }t| jd
�dk�rld} xP| jd
D ]B}
|
d k�r2| td 7 } n$|
d k�rJ| td 7 } n | |
d 7 } �qW |t| d 7 }t| jd
�dk�r�d} x | jd
D ]} | | 7 } �q�W |t| 7 }tj|d d |d d t|�dd�| _ |j | j dddd� tj |tt �dfg�| _ |j | j ddddd� || _!d S )Nr rB rn z
r� r r$ z%s
zserver type is %s Z system_slotsZenterprise_entitledZvirtualization_hostZuniversal_activation_keyri r� )rH rC )rH rD )r r r r )r r r r )"r rJ r� r rK rL rN r Z
REVIEW_WINDOWr� ZPM_PLUGIN_WARNINGr� ZPM_PLUGIN_CONF_ERRORr� ZPM_PLUGIN_CONF_CHANGEDZREVIEW_WINDOW_PROMPTr� ZCHANNELS_TITLEZ OK_CHANNELSr` ra �
serverTypeZCHANNELS_SAT_WARNINGZ
MANAGEMENTZVIRTZSLOTSZACTIVATION_KEYrU Z
review_windowrT rP r rS rY )
rZ r rJ rL r[ Zreview_window_textZ channel_listr� r� Z slot_listZslotZ act_key_listZact_keyr r r r] ] sT
(zReviewWindow.__init__c C s dS )Nr r )rZ r r r r^ � s zReviewWindow.saveResultsc C sN t jd| j � | jj� }| jj|�}|dkr2d}| jjs>d}| jj rJd}|S )Nz
Running %sr_ rC r )
r` ra rb rY rc rS rd rJ r� r� )rZ re r1 r r r rf � s
zReviewWindow.runN)r= r> r? rb r] r^ rf r r r r r� Z s ?r� c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�TuiZRHN_REGISTER_TUIc C s� || _ || _tjj� | _| j� d| _ytj� | _ W n$ t
j k
rX t |t
d�� Y nX tttttttttttttg
| _t j � d | _ t d s�t j dd� t d | _!d S )Nr z"You specified an invalid protocol.z Only https and http are allowed.r� z'/usr/share/rhn/CLN-ORG-TRUSTED-SSL-CERTzBYou specified an invalid protocol.Only https and http are allowed.)"r rI r rK rL � drawFramers r Z
getServerTyper� r ZInvalidProtocolErrorr! rV r� r@ r� rg r� r� r� r� r� r� r� r� r� �windowsr
r4 rW rw r~ r� )rZ r rI r r r r] � s6 z Tui.__init__c C s | j j� d S )N)r r )rZ r r r �__del__� s z Tui.__del__c C s4 t | _| jjddt| j�� | jjttd��� d S )Nr zL <Tab>/<Alt-Tab> between elements | <Space> selects | <F12> next screen)ZCOPYRIGHT_TEXTZ welcomeTextr Z drawRootTextr Z pushHelpLinerV )rZ r r r r� � s z
Tui.drawFramec C s~ d| _ d| _i | _ddddddddddd�
| _i | _d| jd< d| _d| _d| _d| _d| _ g | _
g | _ d| _ d| _
d| _d S )Nrn )
Zentitlement_numZregistration_numZ
first_name� last_nameZcompanyZaddressZcity�state�zipZcountryZregistration_numberr r )r� r� r� Z productInfor� r� r� r� r� r� r� r� r� r� r� )rZ r r r � initResults� s2
zTui.initResultsc C sR t jd| j � | j� d}�z"d}�x|t| j�k �r<d }y| j| | j| �}W n0 tk
r� |dkrt|d }n|d }w(Y nX t jd| � |j� }t jd| � |dkr�|dkr�|d }|dkr�| j| jt jkr�|d8 }d}q(|d kr�d S |d
k�r t jd � t
| j�dk�r:d S q(|d kr(|d }|j � d}q(W W d | jj � X d S )
Nz
Running %sZforwardr r z index is %sz Result %srq Zbackwardr rr zCaught a cancel requestrC )
r` ra rb r� r� r� r r< rf r� r2 r^ r )rZ � direction�index�winre r r r rf � sF
zTui.runN) r= r> r? rb r] r� r� r� rf r r r r r� � s !!r� c C s~ d} t j t jt j� ttj�dkrBtjd dks>tjd dkrBd} tj� }t� dkrh| rht |t
d�� t || �}|j � d S )Nr r z-tz--testz2You must run the RHN registration program as root.)
�signal�SIGINT�SIG_IGNr� r �argvr Z SnackScreenr r! rV r� rf )rI r rJ r r r �main5 s
r� �__main__)<�osr r �gettextZ translation�t�hasattrr rV r r� Zup2date_clientr r r r r r r
Zup2date_client.configr r Zrhnr
Zrhn.connectionsr Zrhn.i18nr Zup2date_client.pmPluginr r Z up2date_client.rhnreg_constantsZinitLogr` ZinitUp2dateConfigrw r r! r# r2 r; � Exceptionr< r@ rg r� r� r� r� r� r� r� r� r� r� r� r� r� r� r= r r r r �<module> sd
/w)4))w ~P$ #S