File: //lib/python3/dist-packages/uaclient/entitlements/__pycache__/__init__.cpython-38.pyc
U
��Jh@" �
@ s2 d dl Z d dlZd dlmZ d dlmZmZmZmZ d dl m
Z
d dl m Z d dl
mZ d dlmZ d dlmZ d d lmZ d d
lmZ d d lmZ d d lmZmZ d d
l m Z d dl m Z d dl m!Z! d dl"m#Z# d dl$m%Z%m&Z& d dl'm(Z( eeeeeej)ej*ej+e e e!e%e&g
Z,d5e e-e-e.e.eee- ed�dd�Z/d6e e.ee- d�dd�Z0e ee- ee- d�d d �Z1e j2G d d � d e j3��Z4e ee- d �d!d"�Z5e ee- d �d#d$�Z6e ee e4ee-e.f ee- d%�d&d'�Z7e e4ee- d(�d)d*�Z8ee- e d+�d,d-�Z9e e
j:d �d.d/�Z;e e.d �d0d1�Z<d7e e-d2�d3d4�Z=dS )8� N)� defaultdict)�Dict�List�Optional�Type)�
exceptions)�UAConfig)�fips)�AnboxEntitlement)�
UAEntitlement)�CommonCriteriaEntitlement)�CISEntitlement)�ApplicabilityStatus)�ESMAppsEntitlement�ESMInfraEntitlement)�LandscapeEntitlement)�LivepatchEntitlement)�RealtimeKernelEntitlement)�RepoEntitlement)�ROSEntitlement�ROSUpdatesEntitlement)�EntitlementNotFoundError� F)�cfg�name�variant�purge� access_only�
extra_args�returnc C sn t D ]Z}|| ||||d�}||jkr|s0| S ||jkrT|j| | |||d� S t|d��qt|d��dS )a1 Returns a UAEntitlement object based on the provided name.
:param cfg: UAConfig instance
:param name: The name of the entitlement to return
:param variant: The variant name to be used
:param purge: If purge operation is enabled
:param access_only: If entitlement should be set with access only
:param extra_args: Extra parameters to create the entitlement
:raise EntitlementNotFoundError: If no entitlement with the given name is
found, or if the entitlement exists but no variant with the specified
name is found.
)r r � called_namer r )r r r r )Zentitlement_nameN)�ENTITLEMENT_CLASSES� valid_namesZvariantsr )r r r r r r � entitlement�ent� r% �@/usr/lib/python3/dist-packages/uaclient/entitlements/__init__.py�entitlement_factory' s( �
�
r' )r � all_namesr c sF t }|r0g }|D ]}|�|� d�j� qt|�S t� fdd�|D ��S )z�Return a list of valid services.
:param cfg: UAConfig instance
:param all_names: if we should return all the names for a service instead
of just the presentation_name
�r c s g | ]}|� d �j �qS )r) )Zpresentation_name)�.0�entitlement_clsr) r% r& �
<listcomp>d s �z"valid_services.<locals>.<listcomp>)r! �extendr" �sorted)r r( Z entitlements�namesr+ r% r) r& �valid_servicesT s
��r0 )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�r$ )�valid_ents_orderedr% r& � sort_order_with_nonexistent_lasts s zIorder_entitlements_for_enabling.<locals>.sort_order_with_nonexistent_lastc s � | �S r2 r% r6 )r8 r% r& �<lambda>y � z1order_entitlements_for_enabling.<locals>.<lambda>)�key)�entitlements_enable_orderr. )r r1 r% )r8 r7 r& � order_entitlements_for_enablingk s r= c @ s e Zd Ze� Ze� ZdS )� SortOrderN)�__name__�
__module__� __qualname__�object�REQUIRED_SERVICES�DEPENDENT_SERVICESr% r% r% r& r> | s r> )r r c C s t | tjd�S )zR
Return the entitlements disable order based on dependent services logic.
�r �
sort_order)�_sort_entitlementsr> rD r) r% r% r& �entitlements_disable_order� s rH c C s t | tjd�S )zP
Return the entitlements enable order based on required services logic.
rE )rG r> rC r) r% r% r& r< � s r< �r �ent_clsrF �visited�orderc C s� |j |krd S || �}|tjkr2dd� |jD �}n
t|j�}|D ] }|j |kr@t| ||||d� q@|�t|j �� d|t|j �< d S )Nc S s g | ]
}|j �qS r% )r# )r* �er% r% r& r, � s z,_sort_entitlements_visit.<locals>.<listcomp>rI T) r r> rC Zrequired_services�listZdependent_services�_sort_entitlements_visit�append�str)r rJ rF rK rL r$ Zcls_listZcls_dependencyr% r% r& rO � s"
�rO )r rF r c C s( g }i }t D ]}t| ||||d� q |S )NrI )r! rO )r rF rL rK rJ r% r% r& rG � s �rG )r/ r c C sD g }| D ] }|t |dd�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( )r0 rP r. �set)r/ r Zentitlements_found�ent_name�entitlements_not_foundr% r% r& �get_valid_entitlement_names� s rU c C sJ t |d�}d�|�}d�tjd| d dddd��}tjd d�| �|d
�S ) z{
Constructs the MESSAGE_INVALID_SERVICE_OP_FAILURE message
based on the attempted services and valid services.
r) �, �
zTry �.�P F)�widthZbreak_long_wordsZbreak_on_hyphens�enable)Z operationZinvalid_service� service_msg)r0 �join�textwrapZwrapr ZInvalidServiceOpError)rT r Zvalid_services_namesr" r\ r% r% r& �*create_enable_entitlements_not_found_error� s
���r_ c C s� t t�}t| �D ]�}t| |�}t|t�s*q|�� \}}|tjkr~|j }|j
pNd}|D ](}|j � ||�} || �
|||d�� qT|�� D ]T\}
} t| �dkr�| d d } | d d }
tj| jd�tdd � | D ���| |
d
��q�qd S ) Nr% )�from�apt_url�suite� r ra rb rV c s s | ]}t |d �V qdS )r` N)rQ )r* r$ r% r% r& � <genexpr>
s z>check_entitlement_apt_directives_are_unique.<locals>.<genexpr>)Zurlr/ ra rb T)r rN r0 r' �
isinstancer �applicability_statusr Z
APPLICABLEra �
apt_suitesZrepo_policy_check_tmpl�formatrP �itemsr5 r Z%EntitlementsAPTDirectivesAreNotUniqueZ contract_urlr] r. )r Zentitlement_directivesrS r$ rf �_ra rg rb Zentitlement_directiveZdef_pathZ
ent_directiveZ ent_apt_urlZ ent_suiter% r% r& �+check_entitlement_apt_directives_are_unique� sB
��� ��
rk )r rS c C s2 zt | ||d�jW S tjk
r, | Y S X d S )N)r )r' �titler �UbuntuProError)r rS r r% r% r& � get_title s rn )r FFN)F)r )>�enumr^ � collectionsr �typingr r r r Zuaclientr Zuaclient.configr Zuaclient.entitlementsr Zuaclient.entitlements.anboxr
Zuaclient.entitlements.baser Zuaclient.entitlements.ccr Zuaclient.entitlements.cisr
Z(uaclient.entitlements.entitlement_statusr Zuaclient.entitlements.esmr r Z uaclient.entitlements.landscaper Z uaclient.entitlements.livepatchr Z uaclient.entitlements.realtimer Zuaclient.entitlements.repor Zuaclient.entitlements.rosr r Zuaclient.exceptionsr ZFIPSEntitlementZFIPSUpdatesEntitlementZFIPSPreviewEntitlementr! rQ �boolr' r0 r= �unique�Enumr> rH r< rO rG rU rm r_ rk rn r% r% r% r& �<module> s� � �
� - �
� � � .