File: //usr/lib/python3/dist-packages/uaclient/entitlements/__pycache__/__init__.cpython-310.pyc
o
nHJe �
@ 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 lmZ d d
lmZmZ d d lmZ d d lmZ d d
lm Z d dl m Z m Z d dl m!Z! d dl"m#Z# e
eeeee j$e j%e j&eee e e g
Z'd3de de(de(fdd�Z) d4de de*de*dee( fdd �Z+de d ee( dee( fd d �Z,e j-G d d!� d!e j.��Z/de dee( fd"d#�Z0de dee( fd$d%�Z1de d&ee d'e/d(ee(e*f d)ee( f
d*d+�Z2de d'e/dee( fd,d-�Z3d.ee( de fd/d0�Z4de de*dej5fd1d2�Z6dS )5� N)�Dict�List�Type)�
exceptions)�UAConfig)�fips)�AnboxEntitlement)�
UAEntitlement)�CommonCriteriaEntitlement)�CISEntitlement)�ESMAppsEntitlement�ESMInfraEntitlement)�LandscapeEntitlement)�LivepatchEntitlement)�RealtimeKernelEntitlement)�ROSEntitlement�ROSUpdatesEntitlement)�EntitlementNotFoundError)�is_config_value_true� �cfg�name�variantc C sV t D ]#}|| d�}||jv r%|s| S ||jv r |j| S t|d��qt|d��)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_name)�ENTITLEMENT_CLASSES� valid_names�variantsr )r r r � entitlement�ent� 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>L s ��z"valid_services.<locals>.<listcomp>r c s g | ]}|� d �j �qS )r )�presentation_namer' r r r! r) Z s
��)r r r �extendr �sorted)r r# r$ �allow_beta_cfg� entitlements�namesr r r r! �valid_services= s"
�
��r0 �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_lasti s
�zIorder_entitlements_for_enabling.<locals>.sort_order_with_nonexistent_lastc s � | �S r2 r r6 )r8 r r! �<lambda>o s z1order_entitlements_for_enabling.<locals>.<lambda>)�key)�entitlements_enable_orderr, )r r1 r )r8 r7 r! � order_entitlements_for_enablinga s r< c @ s e Zd Ze� Ze� ZdS )� SortOrderN)�__name__�
__module__� __qualname__�object�REQUIRED_SERVICES�DEPENDENT_SERVICESr r r r! r= r s
r= c C � t | tjd�S )zR
Return the entitlements disable order based on dependent services logic.
�r �
sort_order)�_sort_entitlementsr= rC r r r r! �entitlements_disable_orderx � rH c C rD )zP
Return the entitlements enable order based on required services logic.
rE )rG r= rB r r r r! r; rI r; �ent_clsrF �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 rJ rF rK rL T)r r= rB �required_services�dependent_services�_sort_entitlements_visit�append�str)r rJ rF rK rL �cls_list�cls_dependencyr r r! rP � s"
��rP c C s( g }i }t D ] }t| ||||d� q|S )NrM )r rP )r rF rL rK rJ r r r! rG � s �rG 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$ )r0 rQ r, �set)r/ r �entitlements_found�ent_name�entitlements_not_foundr r r! �get_valid_entitlement_names� s �
�rY c C sL 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 r# z, �
zTry �.�P F)�width�break_long_words�break_on_hyphens�enable)� operation�invalid_service� service_msg)r0 �join�textwrap�wrapr �InvalidServiceOpError)rX r r# �valid_services_namesr rc r r r! �*create_enable_entitlements_not_found_error� s
���ri )r )FF)7�enumre �typingr r r �uaclientr �uaclient.configr �uaclient.entitlementsr �uaclient.entitlements.anboxr �uaclient.entitlements.baser �uaclient.entitlements.ccr
�uaclient.entitlements.cisr �uaclient.entitlements.esmr r
� uaclient.entitlements.landscaper � uaclient.entitlements.livepatchr � uaclient.entitlements.realtimer �uaclient.entitlements.rosr r �uaclient.exceptionsr �
uaclient.utilr �FIPSEntitlement�FIPSUpdatesEntitlement�FIPSPreviewEntitlementr rR r" �boolr0 r< �unique�Enumr= rH r; rP rG rY �UbuntuProErrorri r r r r! �<module> s� �����
�$��
����
�
� ���