3
e�$g � @ s� d dl mZ d dl mZ y d dlZW n ek
r@ d dlZY nX d dlZejddd�Ze ed�sjeje_
ej
Z dd idd
idd
idd
idd
idd
idd
idd
id �Z d d
� Z
G dd� dej�ZdS )� )�config)�
up2dateErrorsNzrhn-client-toolsT)Zfallback�ugettext�versionZ21�1)� caneatCheese�supportsAutoUp2dateOptionzregistration.finish_messagez xmlrpc.packages.extended_profilezregistration.delta_packagesz registration.update_contact_infoz$registration.extended_update_supportzregistration.smbiosc C s� d }| j d�}g }x�|D ]�}y dd� |j dd�D �\}}W n tk
rP wY nX |j dd�\}}|d
dkrvtd� |d d � }||d �}|j||f� qW |S ) N�,c S s g | ] }|j � �qS � )�strip)�.0�ir
r
�"/usr/lib/python3.6/capabilities.py�
<listcomp># s z parseCap.<locals>.<listcomp>�=� �(�)z2something broke in parsing the capabilited headers)r �value���r )�split�
ValueError�print�append) Z capstringr Zcaps�capslist�capZ key_version�keyr �datar
r
r �parseCap s
r c @ sN e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z ddd�Z
dS )� Capabilitiesc C s& t j j| � i | _t| _tj� | _d S )N)�UserDict�__init__� missingCaps�
neededCapsr ZinitUp2dateConfig�cfg)�selfr
r
r r! : s zCapabilities.__init__c C sJ xD|j � D ]8\}}|j� dkr
t|�}x|D ]\}}|| j|< q,W q
W d S )Nzx-rhn-server-capability)�items�lowerr r )r% Zheadersr �valr r r r
r
r �populateC s
zCapabilities.populatec C sx |j d�}|dkrF|jd�}|d }|d }tt|�t|�d �}|S |jd�}t|�dkrndd� |D �}|S t|�gS )N�-r r �:c S s g | ] }t |��qS r
)�int)r �ar
r
r r W s z0Capabilities.parseCapVersion.<locals>.<listcomp>)�findr �ranger, �len)r% Z
versionString�index�rng�start�endZversionsZversr
r
r �parseCapVersionK s
z Capabilities.parseCapVersionc C sZ || j kr"td�| }d | j|< n4| j | }t|d �| j|d �krV| j| | j|< d S )NzXThis client requires the server to support %s, which the current server does not supportr )r �_r" r, r5 r# )r% r ZcapvalueZerrstrr r
r
r � validateCap\ s
zCapabilities.validateCapc C s2 x$| j j� D ]}| j|| j | � q W | j� d S )N)r# �keysr7 �workaroundMissingCaps)r% r r
r
r �validateh s zCapabilities.validatec C s, || j kr d| j|< | j |= n
d| j|< d S )Nr r )�tmpCapsr$ )r% r Z
configItemr
r
r � setConfign s
zCapabilities.setConfigc C sZ | j | _d}|| jkr | j|= n dddddddd �}x |j� D ]}| j||| � q>W g }g }t| j�r�x>| jD ]4}| j| }|d kr�|j||f� qp|j||f� qpW d
}g }t|��rxH|D ]@\}}|d ||d | j| d f }|j||| j| d
�� q�W t|��r@x0|D ](\}}|d| }|j||d
d
�� �qW t|��rVtj||��d S )Nr r ZsupportsFinishMessageZsupportsUpdateContactInfoZsupportsDeltaPackagesZ supportsExtendedPackageProfileZ supportsEUSZsupportsSMBIOS)r zregistration.finish_messagez registration.update_contact_infozregistration.delta_packagesz xmlrpc.packages.extended_profilez$registration.extended_update_supportzregistration.smbios� z3Needs %s of version: %s but server has version: %s
r )ZcapName�capInfoZ
serverVersionz5Needs %s but server does not support that capability
) r" r; r8 r< r0 r r r ZServerCapabilityError) r% r Z
capsConfigMapr" ZwrongVersionCapsr r> Z errStringZ errorListr
r
r r9 u sF
z"Capabilities.workaroundMissingCapsNc C sR |dkst |�j� st�|| jkr&dS |rN| j| }t|�| j|d �krNdS dS )a� Checks if the server supports a capability and optionally a version.
Returns True or False.
This complements the neededCaps mechanism provided by this module.
Using hasCapability makes it easier to do something only if the server
supports it or to put workaround code in the user of this class. The
neededCaps mechanism makes it easier to put workaround code in this
module, which makes sense if it is to be shared.
'capability' should be a string such as 'registration.foobar'. It can
be a capability in 'neededCaps' above or one that isn't there. 'version'
can be a string (where isdigit() is True) or an int.
NFr T)�str�isdigit�AssertionErrorr r, r5 )r% Z
capabilityr r r
r
r �
hasCapability� s
zCapabilities.hasCapability)N) �__name__�
__module__� __qualname__r! r) r5 r7 r: r< r9 rB r
r
r
r r 9 s <r )Zup2date_clientr r r � ImportError� collections�gettextZ translation�t�hasattrr r6 r# r r r
r
r
r �<module> s(