File: //usr/lib/python3/dist-packages/uaclient/entitlements/__pycache__/realtime.cpython-310.pyc
o
nHJe� � @ s� d dl mZmZmZmZmZ d dlmZmZm Z m
Z
m Z d dl m
Z
d dlmZmZ d dlmZmZmZ e�� ZG dd� de
j�ZG dd � d e�ZG d
d � d e�ZG d d
� d
e�ZG dd� de�Z dS )� )�Any�Dict�Optional�Tuple�Type)�apt� event_logger�messages�system�util)�repo)�IncompatibleService�
UAEntitlement)�MessagingOperations�MessagingOperationsDict�StaticAffordancec @ s� e Zd ZdZejZejZej Z
ej j Z
dZdZdZdZdZdefdd�Zdeeee f fd d
�Zedee d f fd d
��Z edee d f fdd��Z ede fdd��Z!ddd�Z"dS )�RealtimeKernelEntitlementzrealtime-kernelz ubuntu-pro-realtime-kernel.gpgTF�UbuntuRealtimeKernel�returnc C s( t jt| j�tdg�d�}t�|� |S )z%Check if system needs to be rebooted.zlinux-.*-realtime)�installed_pkgs�installed_pkgs_regex)r
�
should_reboot�set�packages�event� needs_reboot)�self�reboot_required� r �@/usr/lib/python3/dist-packages/uaclient/entitlements/realtime.py�_check_for_reboot s �
z+RealtimeKernelEntitlement._check_for_rebootc C s t jt tjttjtiS �N)�GenericRealtime� variant_name�NvidiaTegraRealtime�IntelIotgRealtime�r r r r �
_get_variants$ s �z'RealtimeKernelEntitlement._get_variants.c C s> ddl m}m} ddlm} t|tj�t|tj�t|tj �fS )Nr )�FIPSEntitlement�FIPSUpdatesEntitlement)�LivepatchEntitlement)
�uaclient.entitlements.fipsr( r) � uaclient.entitlements.livepatchr* r
r �REALTIME_FIPS_INCOMPATIBLE�"REALTIME_FIPS_UPDATES_INCOMPATIBLE� REALTIME_LIVEPATCH_INCOMPATIBLE)r r( r) r* r r r �incompatible_services+ s ����z/RealtimeKernelEntitlement.incompatible_servicesc C s t jdd� dffS )Nc S s t �� S r! )r
� is_containerr r r r �<lambda>E s z>RealtimeKernelEntitlement.static_affordances.<locals>.<lambda>F)r �#REALTIME_ERROR_INSTALL_ON_CONTAINERr& r r r �static_affordances@ s
��z,RealtimeKernelEntitlement.static_affordancesc C sL d }| j stjtj| jdd�fg}d }| js!tjtj| jd�fg}||d�S )NT)�msg�
assume_yes�default)r5 r6 )�
pre_enable� pre_disable)� access_onlyr �prompt_for_confirmationr �REALTIME_PROMPTr6 �purge�REALTIME_PRE_DISABLE_PROMPT)r r8 r9 r r r � messagingJ s( ��� ��� �z#RealtimeKernelEntitlement.messagingNc C s@ t | j��t t�� ��}|r t�t|�tjj | j
d�� d S d S )N)�title) r r � intersectionr � get_installed_packages_names�remove_packages�listr �DISABLE_FAILED_TMPL�formatr@ )r r r r r rC l s
���z)RealtimeKernelEntitlement.remove_packages)r N)#�__name__�
__module__� __qualname__�namer �REALTIME_TITLEr@ �REALTIME_DESCRIPTION� description�REALTIME_HELP_TEXT� help_text�urls�REALTIME_HOME_PAGE� help_doc_url�
repo_key_file�apt_noninteractive�supports_access_only�supports_purge�origin�boolr r �strr r r' �propertyr r
r0 r r4 r r? rC r r r r r s, �!r c s. e Zd Zedeedf f� fdd� �Z� ZS )�RealtimeVariantr .c s( t � fdd�� j�� D ��}t� j| S )Nc s* g | ]\}}t |tjj� j|jd ���qS ))�service�variant)r
r � REALTIME_VARIANT_INCOMPATIBLErF r@ )�.0rJ �clsr& r r �
<listcomp>{ s ����z9RealtimeVariant.incompatible_services.<locals>.<listcomp>)�tuple�other_variants�items�superr0 )r �incompatible_variants�� __class__r& r r0 x s
�� z%RealtimeVariant.incompatible_services)rG rH rI rZ r r
r0 �
__classcell__r r rg r r[ w s $r[ c @ �$ e Zd ZdZejZejZdZ dZ
dS )r" �genericTN) rG rH rI r# r �REALTIME_GENERIC_TITLEr@ � REALTIME_GENERIC_DESCRIPTIONrM �
is_variant� check_packages_are_installedr r r r r" � � r" c @ rj )r$ z nvidia-tegraTN) rG rH rI r# r �REALTIME_NVIDIA_TITLEr@ �REALTIME_NVIDIA_DESCRIPTIONrM rn ro r r r r r$ � rp r$ c @ sL e Zd ZdZejZejZdZ dZ
de e e
f deeeej f fdd�ZdS )r% z
intel-iotgT�platform_checksr c C s@ t �� j}|�dg �}||v rdS dtjj| j|d�|�d�fS )N�cpu_vendor_ids)TNF�,)r@ �vendor�supported_vendors) r
� get_cpu_info� vendor_id�getr �INAPPLICABLE_VENDOR_NAMErF r@ �join)r rs ry rt r r r �verify_platform_checks� s
�z(IntelIotgRealtime.verify_platform_checksN)rG rH rI r# r �REALTIME_INTEL_TITLEr@ �REALTIME_INTEL_DESCRIPTIONrM rn ro r rY r r rX r � NamedMessager} r r r r r% � s
��r% N) �typingr r r r r �uaclientr r r r
r �uaclient.entitlementsr �uaclient.entitlements.baser
r �uaclient.typesr r r �get_event_loggerr �RepoEntitlementr r[ r" r$ r% r r r r �<module> s h