File: //usr/lib/python3/dist-packages/uaclient/entitlements/__pycache__/__init__.cpython-310.pyc
o
� ~c< � @ 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"de de#fdd�Z$ d/de de%de%dee# fdd�Z&de dee# dee# fdd�Z'e j(G d d � d e j)��Z*de dee# fd d �Z+de dee# fd d!�Z,de d"ee
d#e*d$ee#e%f d%ee# f
d&d'�Z-de d#e*dee# fd(d)�Z.d*ee# de fd+d,�Z/de de%dej0fd-d.�Z1dS )0� 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�jv r| S qt|��)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 )�ENTITLEMENT_CLASSES� valid_namesr )r r � entitlement� r �@/usr/lib/python3/dist-packages/uaclient/entitlements/__init__.py�entitlement_factory s
�r F�
allow_beta� all_names�returnc sl t � jd�}||O }t}|sdd� |D �}|r+g }|D ] }|�|� d�j� qt|�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 )�is_beta��.0r r r r �
<listcomp>A s ��z"valid_services.<locals>.<listcomp>r c s g | ]}|� d �j �qS )r )�presentation_namer r r r r" O s
��)r r r �extendr �sorted)r r r �allow_beta_cfg� entitlements�namesr r r r �valid_services2 s"
�
��r) �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��ent)�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/ )r2 r r �<lambda>d s z1order_entitlements_for_enabling.<locals>.<lambda>)�key)�entitlements_enable_orderr% )r r* r )r2 r1 r � order_entitlements_for_enablingV s r6 c @ s e Zd Ze� Ze� ZdS )� SortOrderN)�__name__�
__module__� __qualname__�object�REQUIRED_SERVICES�DEPENDENT_SERVICESr r r r r7 g s
r7 c C � t | tjd�S )zR
Return the entitlements disable order based on dependent services logic.
�r �
sort_order)�_sort_entitlementsr7 r= r r r r �entitlements_disable_orderm � rB c C r> )zP
Return the entitlements enable order based on required services logic.
r? )rA r7 r< r r r r r5 t rC r5 �ent_clsr@ �visited�orderc C sv |j |v rd S |tjkr|| �j}n|| �j}|D ]}|j |vr)t| ||||d� q|�t|j �� d|t|j �< d S )N�r rD r@ rE rF T)r r7 r< �required_services�dependent_services�_sort_entitlements_visit�append�str)r rD r@ rE rF �cls_list�cls_dependencyr r r rJ { s"
��rJ c C s( g }i }t D ] }t| ||||d� q|S )NrG )r rJ )r r@ rF rE rD r r r rA � s �rA r( c C sF g }| D ]}|t |ddd�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 r )r) rK r% �set)r( r �entitlements_found�ent_name�entitlements_not_foundr r r �get_valid_entitlement_names� s �
�rS 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)�width�break_long_words�break_on_hyphens�enable)� operation�invalid_service� service_msg)r) �join�textwrap�wrapr �INVALID_SERVICE_OP_FAILURE�format)rR r r �valid_services_namesr r] r r r �,create_enable_entitlements_not_found_message� s
���rd )FF)2�enumr_ �typingr r r �uaclientr �uaclient.configr �uaclient.entitlementsr �uaclient.entitlements.baser �uaclient.entitlements.ccr �uaclient.entitlements.cisr
�uaclient.entitlements.esmr r � uaclient.entitlements.livepatchr
� uaclient.entitlements.realtimer �uaclient.entitlements.rosr r �uaclient.exceptionsr �
uaclient.utilr �FIPSEntitlement�FIPSUpdatesEntitlementr rL r �boolr) r6 �unique�Enumr7 rB r5 rJ rA rS � NamedMessagerd r r r r �<module> s� �����
�$��
����
�
� ���