JezK
Edit File: rhnserver.cpython-36.pyc
3 7 f�" � @ 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)�self�server� method_name� r �</usr/lib/python3.6/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�}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 �doCall� xmlrpclib�Faultr �$_DoCallWrapper__exception_from_fault�sys�exc_info�OpenSSL�SSL�Error�str�strip�split�lenr � SSLCertificateVerifyFailedError� 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������)� faultCoder �$AuthenticationOrAccountCreationError� faultString�UnknownMethodException�LoginMinLengthError�PasswordMinLengthError�ValidationError�NoBaseChannelError�InsuffMgmntEntsError� PasswordError�abs� AbuseError�AuthenticationTicketError�RegistrationDeniedError�RhnUuidUniquenessError� DelayError� InvalidRegistrationNumberError�NotEntitlingError�PasswordMaxLengthError� ActivationKeyUsageLimitError�UnableToCreateUser�CommunicationError)r �fault� 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 � getServerr � _capabilities)r r^ r_ ZrpcServerOverrider r r r � s zRhnServer.__init__c C sL | j d krF| jj� }|d kr0| jj� | jj� }tj� | _ | j j|� | j S )N)ra r �get_response_headers� registrationZwelcome_messager � Capabilities�populate)r �headersr 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 rg � 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) rY rZ r[ r\ r Z _RhnServer__get_capabilities�propertyr rg r r r r r r] � s r] )�rhn.tbr �up2date_clientr r r r r r � ImportError� xmlrpc.client�client�objectr r] r r r r �<module>" s x