File: //usr/lib/python3/dist-packages/uaclient/entitlements/__pycache__/__init__.cpython-310.pyc
o
��Jh@" � @ 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 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, d<de de-de-de.de.deee- defd d �Z/d=de d e.dee- fd d �Z0de d!ee- dee- fd"d#�Z1e j2G d$d%� d%e j3��Z4de dee- fd&d'�Z5de dee- fd(d)�Z6de d*ee d+e4d,ee-e.f d-ee- f
d.d/�Z7de d+e4dee- fd0d1�Z8d2ee- de fd3d4�Z9de de
j:fd5d6�Z;de de.fd7d8�Z<d>de d9e-fd:d;�Z=dS )?� 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 sj t D ]-}|| ||||d�}||jv r/|s| S ||jv r*|j| | |||d� S t|d��qt|d��)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 )�entitlement_name)�ENTITLEMENT_CLASSES� valid_names�variantsr )r r r r r r � entitlement�ent� r' �@/usr/lib/python3/dist-packages/uaclient/entitlements/__init__.py�entitlement_factory' s* �
�
�
r) � all_namesc sF t }|rg }|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+ )�presentation_name)�.0�entitlement_clsr+ r' r( �
<listcomp>d s
��z"valid_services.<locals>.<listcomp>)r" �extendr# �sorted)r r* � entitlements�namesr. r' r+ r( �valid_servicesT s
��r4 �entsc s( t | ���fdd�� t|� fdd�d�S )zR
A function to sort entitlments for enabling that preserves invalid names
c s( z� � | �W S ty t� � Y S w �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 r6 r' r: )r< r' r( �<lambda>y s z1order_entitlements_for_enabling.<locals>.<lambda>)�key)�entitlements_enable_orderr1 )r r5 r' )r< r; 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( rA | s
rA c C � t | tjd�S )zR
Return the entitlements disable order based on dependent services logic.
�r �
sort_order)�_sort_entitlementsrA rG r+ r' r' r( �entitlements_disable_order� � rL c C rH )zP
Return the entitlements enable order based on required services logic.
rI )rK rA rF r+ r' r' r( r? � rM r? �ent_clsrJ �visited�orderc C s� |j |v rd S || �}|tjkrdd� |jD �}nt|j�}|D ]}|j |vr0t| ||||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>�r rN rJ rO rP T) r rA rF �required_services�list�dependent_services�_sort_entitlements_visit�append�str)r rN rJ rO rP r& �cls_list�cls_dependencyr' r' r( rV � s$
��rV c C s( g }i }t D ] }t| ||||d� q|S )NrR )r" rV )r rJ rP rO rN r' r' r( rK � s �rK r3 c C sD g }| D ]}|t |dd�v r|�|� 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* )r4 rW r1 �set)r3 r �entitlements_found�ent_name�entitlements_not_foundr' r' r( �get_valid_entitlement_names� s
�r_ 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)�width�break_long_words�break_on_hyphens�enable)� operation�invalid_service� service_msg)r4 �join�textwrap�wrapr �InvalidServiceOpError)r^ r �valid_services_namesr# rj r' r' r( �*create_enable_entitlements_not_found_error� s
���rp c C s� t t�}t| �D ]f}t| |�}t|t�sq|�� \}}|tjkr?|j }|j
p'd}|D ]}|j � ||�} || �
|||d�� q*|�� D ]*\}
} t| �dkrm| d d } | d d }
tj| jd�tdd � | D ���| |
d
��qCqd S ) Nr' )�from�apt_url�suite� r rr rs r` c s s � | ] }t |d �V qdS )rq N)rX )r- r& r' r' r( � <genexpr>
s � z>check_entitlement_apt_directives_are_unique.<locals>.<genexpr>)�urlr3 rr rs T)r rT r4 r) �
isinstancer �applicability_statusr �
APPLICABLErr �
apt_suites�repo_policy_check_tmpl�formatrW �itemsr9 r �%EntitlementsAPTDirectivesAreNotUnique� contract_urlrk r1 )r �entitlement_directivesr] r& rx �_rr rz rs �entitlement_directive�def_path�
ent_directive� ent_apt_url� ent_suiter' r' r( �+check_entitlement_apt_directives_are_unique� sD
��� ����r� r] c C s, z t | ||d�jW S tjy | Y S w )N)r )r) �titler �UbuntuProError)r r] r r' r' r( � get_title s
�r� )r FFN)F)r )>�enumrl � collectionsr �typingr r r r �uaclientr �uaclient.configr �uaclient.entitlementsr �uaclient.entitlements.anboxr
�uaclient.entitlements.baser �uaclient.entitlements.ccr �uaclient.entitlements.cisr
�(uaclient.entitlements.entitlement_statusr �uaclient.entitlements.esmr r � uaclient.entitlements.landscaper � uaclient.entitlements.livepatchr � uaclient.entitlements.realtimer �uaclient.entitlements.repor �uaclient.entitlements.rosr r �uaclient.exceptionsr �FIPSEntitlement�FIPSUpdatesEntitlement�FIPSPreviewEntitlementr" rX �boolr) r4 r@ �unique�EnumrA rL r? rV rK r_ r� rp r� r� r' r' r' r( �<module> s� �������
�
� -��
����
�
� �
��
�.