3
e�$g�" � @ s� d dl mZ d dlmZ d dlmZ d dlmZ d dlZd dlZy d dlZW n e k
rj d dl
j ZY nX G dd� de �Z
G dd � d e �ZdS )
� )�
raise_with_tb)� rpcServer)�
up2dateErrors)� capabilitiesNc @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �_DoCallWrapperze
A callable object that will handle multiple levels of attributes,
and catch exceptions.
c C s || _ || _d S )N)�_server� _method_name)�selfZserver� method_name� r � /usr/lib/python3.6/rhnserver.py�__init__5 s z_DoCallWrapper.__init__c C s t | jd| j|f �S )z= Recursively build up the method name to pass to the server. z%s.%s)r r r )r r
r r r � __getattr__9 s z_DoCallWrapper.__getattr__c O s� t | j| j�}ytj|f|�|�S tjk
rL t| jt j
� d �� Y n� t j j
k
r� tt j
� d �}|jd�}|jd�}d}t|�dkr�|d }nt|�dkr�|d }|jd�}|dkr�ttj� � nttj|�� Y nX dS ) z3 Call the method. Catch faults and translate them. � z[()]�,� � z 'zcertificate verify failedN)�getattrr r r ZdoCall� xmlrpclibZFaultr �$_DoCallWrapper__exception_from_fault�sys�exc_info�OpenSSLZSSL�Error�str�strip�split�lenr Z SSLCertificateVerifyFailedErrorZ NetworkError)r �args�kwargs�method�error�pieces�messager r r �__call__> s$
z_DoCallWrapper.__call__c C s� |j dkrtj|j�}�n�|j d kr4tj|j�}�n�|j d krNtj|j�}�nr|j d krhtj|j�}�nX|j d kr�tj|j�}�n>|j d kr�tj|j�}�n$|j d!kr�tj|j�}�n
|j d"kr�tj|j�}�n�|j d#kr�tj|j�}�n�|j d$k�rtj |j�}�n�|j d%k�r"tj
|j�}�n�t |j �d k�rBtj |j�}�n~t |j �d
k�rbtj
|j�}�n^t |j �dk�r~tj� }�nBt |j �dk�r�tj|j�}�n"|j dk�r�tj|j�}�nt |j �dk�r�tj |j�}n�|j d&k�r�tj|j�}n�|j d'k�r tj|j�}n�|j d(k�r&tj|j�}n�|j d)k�r@tj|j�}n�|j d*k�sX|j d+k�rftj|j�}nZ|j d,k�r�tj|j�}n@|j d-k�r�tj|j�}n&|j d.k�r�tj|j�}n tj|j�}|S )/N� r �n r �
� � � � � �$ �1 �< �J �i �c �[ �j iX iY iZ i� i� i� �= � ������i�������i�i�i�i�i��i��i���i����i����i����i����i/���iD���iC���i������)Z faultCoder Z$AuthenticationOrAccountCreationErrorZ faultStringZUnknownMethodExceptionZLoginMinLengthErrorZPasswordMinLengthErrorZValidationErrorZNoBaseChannelErrorZInsuffMgmntEntsErrorZ
PasswordError�absZ
AbuseErrorZAuthenticationTicketErrorZRegistrationDeniedErrorZRhnUuidUniquenessErrorZ
DelayErrorZ InvalidRegistrationNumberErrorZNotEntitlingErrorZPasswordMaxLengthErrorZ ActivationKeyUsageLimitErrorZUnableToCreateUserZCommunicationError)r ZfaultZ exceptionr r r Z__exception_from_faultX sr
z%_DoCallWrapper.__exception_from_faultN)�__name__�
__module__� __qualname__�__doc__r
r r$ r r r r r r . s
r c @ s: e Zd ZdZd dd�Zdd� Zee�Zdd� Zd d
� Z dS ) � RhnServerz`
An rpc server object that calls doCall for you, and catches lower
level exceptions
Nc C s* |d krt j||d�| _n|| _d | _d S )N)�serverOverride�timeout)r Z getServerr �
_capabilities)r rA rB ZrpcServerOverrider r r r
� s
zRhnServer.__init__c C sL | j d krF| jj� }|d kr0| jj� | jj� }tj� | _ | j j|� | j S )N)rC r Zget_response_headersZ registrationZwelcome_messager Z CapabilitiesZpopulate)r Zheadersr r r Z__get_capabilities� s
z RhnServer.__get_capabilitiesc C s | j j||� d S )N)r �
add_header)r �key�valuer r r rD � s zRhnServer.add_headerc C s t | j|�S )z8 Return a callable object that will do the work for us. )r r )r r
r r r r � s zRhnServer.__getattr__)NNN)
r< r= r> r? r
Z _RhnServer__get_capabilities�propertyr rD r r r r r r@ � s
r@ )Zrhn.tbr Zup2date_clientr r r r r r � ImportErrorZ
xmlrpc.clientZclient�objectr r@ r r r r �<module>" s x