| Server IP : 172.67.131.151 / Your IP : 104.23.197.56 Web Server : Apache System : Linux keen-cori.18-142-40-148.plesk.page 5.15.0-1084-aws #91~20.04.1-Ubuntu SMP Fri May 2 06:59:36 UTC 2025 x86_64 User : simottodesign.com_2tntp341vs7 ( 10011) PHP Version : 8.3.31 Disable Function : opcache_get_status MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : ON | Sudo : ON | Pkexec : ON Directory : /usr/lib/python3/dist-packages/uaclient/entitlements/__pycache__/ |
Upload File : |
U
2�d� � @ 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%eee e e g Z&d/e e'e'd�dd�Z(d0e e)e)ee' d�dd�Z*e ee' ee' d�dd�Z+e j,G d d � d e j-��Z.e ee' d �d d �Z/e ee' d �d!d"�Z0e ee e.ee'e)f ee' d#�d$d%�Z1e e.ee' d&�d'd(�Z2ee' e d)�d*d+�Z3e e)ej4d,�d-d.�Z5dS )1� N)�Dict�List�Type)�messages)�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}|| d�}||jkr|s(| S ||jkr@|j| S t|��qt|��dS )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 N)�ENTITLEMENT_CLASSES� valid_namesZvariantsr )r r r � entitlement�ent� r �@/usr/lib/python3/dist-packages/uaclient/entitlements/__init__.py�entitlement_factory$ s
r F)r �
allow_beta� all_names�returnc sl t � jd�}||O }t}|s*dd� |D �}|rVg }|D ]}|�|� d�j� q6t|�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 )Zis_beta��.0r r r r �
<listcomp>K s �z"valid_services.<locals>.<listcomp>r c s g | ]}|� d �j �qS )r )Zpresentation_namer$ r r r r& Y s �)r r r �extendr �sorted)r r! r" Zallow_beta_cfgZ entitlements�namesr r r r �valid_services< s"
�
��r* )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_lasth s zIorder_entitlements_for_enabling.<locals>.sort_order_with_nonexistent_lastc s � | �S r, r r0 )r2 r r �<lambda>n � z1order_entitlements_for_enabling.<locals>.<lambda>)�key)�entitlements_enable_orderr( )r r+ r )r2 r1 r � order_entitlements_for_enabling` s r7 c @ s e Zd Ze� Ze� ZdS )� SortOrderN)�__name__�
__module__� __qualname__�object�REQUIRED_SERVICES�DEPENDENT_SERVICESr r r r r8 q s r8 )r r# c C s t | tjd�S )zR
Return the entitlements disable order based on dependent services logic.
�r �
sort_order)�_sort_entitlementsr8 r> r r r r �entitlements_disable_orderw s rB c C s t | tjd�S )zP
Return the entitlements enable order based on required services logic.
r? )rA r8 r= r r r r r6 ~ s r6 �r �ent_clsr@ �visited�orderc C sv |j |krd S |tjkr$|| �j}n
|| �j}|D ] }|j |kr2t| ||||d� q2|�t|j �� d|t|j �< d S )NrC T)r r8 r= Zrequired_servicesZdependent_services�_sort_entitlements_visit�append�str)r rD r@ rE rF Zcls_listZcls_dependencyr r r rG � s
�rG )r r@ r# c C s( g }i }t D ]}t| ||||d� q |S )NrC )r rG )r r@ rF rE rD r r r rA � s �rA )r) r c C sF g }| D ] }|t |ddd�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! r" )r* rH r( �set)r) r Zentitlements_foundZent_name�entitlements_not_foundr r r �get_valid_entitlement_names� s � rL )r r! r# 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)�widthZbreak_long_wordsZbreak_on_hyphens�enable)Z operationZinvalid_service� service_msg)r* �join�textwrapZwrapr ZINVALID_SERVICE_OP_FAILURE�format)rK r r! Zvalid_services_namesr rR r r r �,create_enable_entitlements_not_found_message� s
���rV )r )FF)6�enumrT �typingr r r Zuaclientr Zuaclient.configr Zuaclient.entitlementsr Zuaclient.entitlements.anboxr Zuaclient.entitlements.baser Zuaclient.entitlements.ccr
Zuaclient.entitlements.cisr Zuaclient.entitlements.esmr r
Z uaclient.entitlements.landscaper Z uaclient.entitlements.livepatchr Z uaclient.entitlements.realtimer Zuaclient.entitlements.rosr r Zuaclient.exceptionsr Z
uaclient.utilr ZFIPSEntitlementZFIPSUpdatesEntitlementr rI r �boolr* r7 �unique�Enumr8 rB r6 rG rA rL Z NamedMessagerV r r r r �<module> st � � � % �
� �