Your IP : 216.73.216.172
U
���h�" � @ sF 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 m Z d dl m Z d dl!m"Z" d dl#m$Z$ d dl%m&Z& d dl'm(Z(m)Z) d dl*m+Z+ eeeee ee ej,ej-ej.e e"e$e(e)gZ/d6e e0e0e1e1eee0 ed�dd�Z2d7e e1ee0 d�dd�Z3e ee0 ee0 d �d d �Z4e j5G d d � d e j6��Z7e ee0 d!�d"d#�Z8e ee0 d!�d$d%�Z9e ee e7ee0e1f ee0 d&�d'd(�Z:e e7ee0 d)�d*d+�Z;ee0 e d,�d-d.�Z<e e
j=d!�d/d0�Z>e e1d!�d1d2�Z?d8e e0d3�d4d5�Z@dS )9� N)� defaultdict)�Dict�List�Optional�Type)�
exceptions)�UAConfig)�fips)�AnboxEntitlement)�
UAEntitlement)�CommonCriteriaEntitlement)�CISEntitlement)�ApplicabilityStatus)�ESMAppsEntitlement�ESMInfraEntitlement)�ESMAppsLegacyEntitlement�ESMInfraLegacyEntitlement)�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>j s �z"valid_services.<locals>.<listcomp>)r# �extendr$ �sorted)r r* Z entitlements�namesr- r' r+ r( �valid_servicesZ s
��r2 )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_lasty s zIorder_entitlements_for_enabling.<locals>.sort_order_with_nonexistent_lastc s � | �S r4 r' r8 )r: r' r( �<lambda> � z1order_entitlements_for_enabling.<locals>.<lambda>)�key)�entitlements_enable_orderr0 )r r3 r' )r: r9 r( � order_entitlements_for_enablingq 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@ rF r+ r' r' r( �entitlements_disable_order� s rJ c C s t | tjd�S )zP
Return the entitlements enable order based on required services logic.
rG )rI r@ rE r+ r' r' r( r> � s r> �r �ent_clsrH �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>rK T) r r@ rE Zrequired_services�listZdependent_services�_sort_entitlements_visit�append�str)r rL rH rM rN r& Zcls_listZcls_dependencyr' r' r( rQ � s"
�rQ )r rH r! c C s( g }i }t D ]}t| ||||d� q |S )NrK )r# rQ )r rH rN rM rL r' r' r( rI � s �rI )r1 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* )r2 rR r0 �set)r1 r Zentitlements_found�ent_name�entitlements_not_foundr' r' r( �get_valid_entitlement_names� s rW 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)r2 �join�textwrapZwrapr ZInvalidServiceOpError)rV r Zvalid_services_namesr$ r^ r' r' r( �*create_enable_entitlements_not_found_error� s
���ra 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 rc rd rX c s s | ]}t |d �V qdS )rb N)rS )r, r& r' r' r( � <genexpr> s z>check_entitlement_apt_directives_are_unique.<locals>.<genexpr>)Zurlr1 rc rd T)r rP r2 r) �
isinstancer �applicability_statusr Z
APPLICABLErc �
apt_suitesZrepo_policy_check_tmpl�formatrR �itemsr7 r Z%EntitlementsAPTDirectivesAreNotUniqueZ contract_urlr_ r0 )r Zentitlement_directivesrU r& rh �_rc ri rd Zentitlement_directiveZdef_pathZ
ent_directiveZ ent_apt_urlZ ent_suiter' r' r( �+check_entitlement_apt_directives_are_unique� sB
��� ��
rm )r rU c C s2 zt | ||d�jW S tjk
r, | Y S X d S )N)r )r) �titler �UbuntuProError)r rU r r' r' r( � get_title s rp )r FFN)F)r )A�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.esm_legacyr r Z uaclient.entitlements.landscaper Z uaclient.entitlements.livepatchr Z uaclient.entitlements.realtimer Zuaclient.entitlements.repor Zuaclient.entitlements.rosr r Zuaclient.exceptionsr ZFIPSEntitlementZFIPSUpdatesEntitlementZFIPSPreviewEntitlementr# rS �boolr) r2 r? �unique�Enumr@ rJ r> rQ rI rW ro ra rm rp r' r' r' r( �<module> s� � �
� - �
� � � .