Your IP : 216.73.216.52
U
8�-d< �
@ s� d dl Z d dlZd dlmZmZmZ d dlmZ d dlm Z d dl
m Z d dl m
Z
d dlmZ d dlmZ d d lmZmZ d d
lmZ d d lmZ d d lmZmZ d d
l m Z d dl m Z eeeee j e j!eeeeg
Z"e e#d�dd�Z$d,e e%e%ee# d�dd�Z&e ee# ee# d�dd�Z'e j(G dd� de j)��Z*e ee# d�d d �Z+e ee# d�d d �Z,e ee
e*ee#e%f ee# d �d!d"�Z-e e*ee# d#�d$d%�Z.ee# e d&�d'd(�Z/e e%ej0d)�d*d+�Z1dS )-� N)�Dict�List�Type)�messages)�UAConfig)�fips)�
UAEntitlement)�CommonCriteriaEntitlement)�CISEntitlement)�ESMAppsEntitlement�ESMInfraEntitlement)�LivepatchEntitlement)�RealtimeKernelEntitlement)�ROSEntitlement�ROSUpdatesEntitlement)�EntitlementNotFoundError)�is_config_value_true)�cfg�namec C s. t D ] }||| d�jkr| S qt|��dS )a Returns a UAEntitlement class based on the provided name.
The return type is Optional[Type[UAEntitlement]].
It cannot be explicit because of the Python version on Xenial (3.5.2).
:param cfg: UAConfig instance
:param name: The name of the entitlement to return
:param not_found_okay: If True and no entitlement with the given name is
found, then returns None.
:raise EntitlementNotFoundError: If not_found_okay is False and no
entitlement with the given name is found, then raises this error.
�r N)�ENTITLEMENT_CLASSES� valid_namesr )r r � entitlement� r �@/usr/lib/python3/dist-packages/uaclient/entitlements/__init__.py�entitlement_factory s
r F)r �
allow_beta� all_names�returnc sl t � jd�}||O }t}|s*dd� |D �}|rVg }|D ]}|�|� d�j� q6t|�S t� fdd�|D ��S )a Return a list of valid (non-beta) services.
:param cfg: UAConfig instance
:param allow_beta: if we should allow beta services to be marked as valid
:param all_names: if we should return all the names for a service instead
of just the presentation_name
zfeatures.allow_betac S s g | ]}|j s|�qS r )Zis_beta��.0r r r r �
<listcomp>A s �z"valid_services.<locals>.<listcomp>r c s g | ]}|� d �j �qS )r )Zpresentation_namer r r r r! O s �)r r r �extendr �sorted)r r r Zallow_beta_cfgZ entitlements�namesr r r r �valid_services2 s"
�
��r% )r �entsr c s( t | ���fdd�� t|� fdd�d�S )zR
A function to sort entitlments for enabling that preserves invalid names
c s. z � � | �W S tk
r( t� � Y S X d S �N)�index�
ValueError�len�Zent)�valid_ents_orderedr r � sort_order_with_nonexistent_last^ s zIorder_entitlements_for_enabling.<locals>.sort_order_with_nonexistent_lastc s � | �S r' r r+ )r- r r �<lambda>d � z1order_entitlements_for_enabling.<locals>.<lambda>)�key)�entitlements_enable_orderr# )r r& r )r- r, r � order_entitlements_for_enablingV s r2 c @ s e Zd Ze� Ze� ZdS )� SortOrderN)�__name__�
__module__� __qualname__�object�REQUIRED_SERVICES�DEPENDENT_SERVICESr r r r r3 g s r3 )r r c C s t | tjd�S )zR
Return the entitlements disable order based on dependent services logic.
�r �
sort_order)�_sort_entitlementsr3 r9 r r r r �entitlements_disable_orderm s r= c C s t | tjd�S )zP
Return the entitlements enable order based on required services logic.
r: )r< r3 r8 r r r r r1 t s r1 �r �ent_clsr; �visited�orderc C sv |j |krd S |tjkr$|| �j}n
|| �j}|D ] }|j |kr2t| ||||d� q2|�t|j �� d|t|j �< d S )Nr> T)r r3 r8 Zrequired_servicesZdependent_services�_sort_entitlements_visit�append�str)r r? r; r@ rA Zcls_listZcls_dependencyr r r rB { s
�rB )r r; r c C s( g }i }t D ]}t| ||||d� q |S )Nr> )r rB )r r; rA r@ r? r r r r< � s �r< )r$ r c C sF g }| D ] }|t |ddd�kr|�|� qtt| �t|� �}||fS )z�Return a list of valid entitlement names.
:param names: List of entitlements to validate
:return: a tuple of List containing the valid and invalid entitlements
T)r r r )r% rC r# �set)r$ r Zentitlements_foundZent_name�entitlements_not_foundr r r �get_valid_entitlement_names� s � rG )r r r c C sN t ||d�}d�|�}d�tjd| d dddd��}tjjd d�| �|d
�S ) z{
Constructs the MESSAGE_INVALID_SERVICE_OP_FAILURE message
based on the attempted services and valid services.
)r r z, �
zTry �.�P F)�widthZbreak_long_wordsZbreak_on_hyphens�enable)Z operationZinvalid_service� service_msg)r% �join�textwrapZwrapr ZINVALID_SERVICE_OP_FAILURE�format)rF r r Zvalid_services_namesr rM r r r �,create_enable_entitlements_not_found_message� s
���rQ )FF)2�enumrO �typingr r r Zuaclientr Zuaclient.configr Zuaclient.entitlementsr Zuaclient.entitlements.baser Zuaclient.entitlements.ccr Zuaclient.entitlements.cisr
Zuaclient.entitlements.esmr r Z uaclient.entitlements.livepatchr
Z uaclient.entitlements.realtimer Zuaclient.entitlements.rosr r Zuaclient.exceptionsr Z
uaclient.utilr ZFIPSEntitlementZFIPSUpdatesEntitlementr rD r �boolr% r2 �unique�Enumr3 r= r1 rB r< rG Z NamedMessagerQ r r r r �<module> sl � � � % �
� �