File: //lib/python3.9/site-packages/up2date_client/__pycache__/rhnserver.cpython-39.pyc
a
c8h;% � @ s� d dl mZ d dlmZ d dlmZ d dlmZ d dlZd dlZz d dlZW n e yj d dl
m Z Y n0 G dd� de �Z
G dd � d e �ZG d
d � 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.9/site-packages/up2date_client/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�}ztj|g|�R i |��W S tjyR t| �t �
� d �� Y n� t j j
y� tt �
� d �}|�d�}|�d�}d}t|�dkr�|d }nt|�dkr�|d }|�d�}|dkr�tt�� � ntt�|�� Y n0 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 dkrt�|j�}�n�|j dkr4t�|j�}�n�|j dkrNt�|j�}�nr|j dkrht�|j�}�nX|j dkr�t�|j�}�n>|j dkr�t�|j�}�n$|j dkr�t�|j�}�n
|j dkr�t�|j�}�n�|j d kr�t�|j�}�n�|j d
k�rt� |j�}�n�|j d k�r"t�
|j�}�n�t |j �d k�rBt� |j�}�n~t |j �d
k�rbt�
|j�}�n^t |j �dk�r~t�� }�nBt |j �dk�r�t�|j�}�n"|j dk�r�t�|j�}�nt |j �dk�r�t� |j�}n�|j dk�r�t�|j�}n�|j dk�r t�|j�}n�|j dk�r&t�|j�}n�|j dk�r@t�|j�}n�|j dk�sX|j dk�rft�|j�}nZ|j dk�r�t�|j�}n@|j dk�r�t�|j�}n&|j dk�r�t�|j�}n t�|j�}|S )N������i�������i�i�i�i�i��i��i����1 �< �J �i �c �[ i����i����i����i����i/���iD���iC���i������)Z faultCoder Z$AuthenticationOrAccountCreationErrorZ faultStringZUnknownMethodExceptionZLoginMinLengthErrorZPasswordMinLengthErrorZValidationErrorZNoBaseChannelErrorZInsuffMgmntEntsErrorZ
PasswordError�absZ
AbuseErrorZAuthenticationTicketErrorZRegistrationDeniedErrorZRhnUuidUniquenessErrorZ
DelayErrorZ InvalidRegistrationNumberErrorZNotEntitlingErrorZPasswordMaxLengthErrorZ ActivationKeyUsageLimitErrorZUnableToCreateUserZCommunicationError)r
Zfault� exceptionr r r
Z__exception_from_faultX s|
�
� � � � 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
NFc C s, |d u r t j|||d�| _n|| _d | _d S )N)�serverOverride�timeout� registration)r Z getServerr �
_capabilities)r
r8 r9 �rpcServerOverrider: r r r
r � s �
zRhnServer.__init__c C sL | j d u rF| j�� }|d u r0| j�� | j�� }t�� | _ | j �|� | j S r )r; r Zget_response_headersr: Zwelcome_messager Z CapabilitiesZpopulate)r
Zheadersr r r
Z__get_capabilities� s
z RhnServer.__get_capabilitiesc C s | j �||� d S r )r �
add_header)r
�key�valuer r r
r= � 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 � s zRhnServer.__getattr__)NNNF)
r3 r4 r5 r6 r Z _RhnServer__get_capabilities�propertyr r= r r r r r
r7 � s �
r7 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)r8 r9 r< r: )�superrA r )r
r8 r9 r< �� __class__r r
r � s
�z RegistrationRhnServer.__init__)NNN)r3 r4 r5 r6 r �
__classcell__r r rC r
rA � s rA )Zrhn.tbr Zup2date_clientr r r r r r � ImportErrorZ
xmlrpc.clientZclient�objectr r7 rA r r r r
�<module>" s x+