3
�Xj%" � @ s� d dl mZ d dl mZ d dlZd dlZy d dlZW n ek
rR d dljZY nX G dd� de �Z
G dd� de �Z G dd � d e �Z dS )
� )� rpcServer)�
up2dateErrorsNc @ 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__2 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__6 s z_DoCallWrapper.__getattr__c O s� t | j| j�}ytj|f|�|�S tjk
rH | jtj � d ��Y n� t
j j k
r� t
tj � d �}|jd�}|jd�}d}t|�dkr�|d }nt|�dkr�|d }|jd�}d|kr�tj� �n
tj|��Y nX dS ) z3 Call the method. Catch faults and translate them. � z[()]�,� � z 'zcertificate verify failedN)�getattrr r r ZdoCall� xmlrpclibZFault�$_DoCallWrapper__exception_from_fault�sys�exc_info�requests�
exceptionsZSSLError�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_faultR 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� Zdd � ZdS ) � RhnServerz`
An rpc server object that calls doCall for you, and catches lower
level exceptions
NFc C s& |d kr t j|||d�| _n|| _d S )N)�serverOverride�timeout� registration)r Z getServerr )r r? r@ �rpcServerOverriderA r r r
r � s zRhnServer.__init__c C s | j j||� d S )N)r �
add_header)r �key�valuer r r
rC � s zRhnServer.add_headerc C s t | j|�S )z6Return a callable object that will do the work for us.)r r )r r r r r
r � s zRhnServer.__getattr__)NNNF)r: r; r< r= r rC r r r r r
r> � s
r> c s" e Zd ZdZd� fdd� Z� ZS )�RegistrationRhnServerz�
A specialized RhnServer subclass for handling registration-related calls.
Intended to be used as a shorthand for registration tasks instead of
the plain RhnServer.
Nc s t t| �j|||dd� d S )NT)r? r@ rB rA )�superrF r )r r? r@ rB )� __class__r r
r � s
z RegistrationRhnServer.__init__)NNN)r: r; r< r= r �
__classcell__r r )rH r
rF � s rF )
Zup2date_clientr r r r r � ImportErrorZ
xmlrpc.clientZclient�objectr r> rF r r r r
�! s u