Server : Apache/2.4.52 (Ubuntu) System : Linux webserver 6.8.0-49-generic #49~22.04.1-Ubuntu SMP PREEMPT_DYNAMIC Wed Nov 6 17:42:15 UTC 2 x86_64 User : www-data ( 33) PHP Version : 8.1.2-1ubuntu2.21 Disable Function : NONE Directory : /lib/python3/dist-packages/uaclient/entitlements/__pycache__/ |
o
���f� � @ s� d dl Z d dlmZmZmZmZmZ d dlmZm Z m
Z
m Z m Z d dl
mZ d dlmZmZ d dlmZmZmZ e �� Ze �e �e��ZG dd� dej �Z G d d
� d
e �Z G d d � d e �Z G d
d� de �Z G dd� de �Z!G dd� de �Z"dS )� N)�Any�Dict�Optional�Tuple�Type)�apt� event_logger�messages�system�util)�repo)�EntitlementWithMessage�
UAEntitlement)�MessagingOperations�MessagingOperationsDict�StaticAffordancec @ s� e Zd ZdZejZejZej Z
ej j Z
dZdZdZdZdZdefdd�Zdeeee f fd d
�Zed d � �Zede e d
f fdd��Z ede e 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 tjttjttjtiS �N)�GenericRealtime� variant_name�NvidiaTegraRealtime�RaspberryPiRealtime�IntelIotgRealtime�r r r r �
_get_variants& s
�z'RealtimeKernelEntitlement._get_variantsc C s | j rd S tS r! )� access_onlyr" r' r r r �default_variant. s z)RealtimeKernelEntitlement.default_variant.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_services4 s ����z/RealtimeKernelEntitlement.incompatible_servicesc C s t jdd� dffS )Nc S s t �� S r! )r
� is_containerr r r r �<lambda>N s z>RealtimeKernelEntitlement.static_affordances.<locals>.<lambda>F)r �#REALTIME_ERROR_INSTALL_ON_CONTAINERr' r r r �static_affordancesI s
��z,RealtimeKernelEntitlement.static_affordancesc C sD d }| j stjtjdd�fg}d }| js tjdtjifg}||d�S )NT)�msg�defaultr8 )�
pre_enable� pre_disable)r) r �prompt_for_confirmationr �REALTIME_PROMPT�purge�REALTIME_PRE_DISABLE_PROMPT)r r: r; r r r � messagingS 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�formatrA )r r r r r rD s s
���z)RealtimeKernelEntitlement.remove_packages)r N)$�__name__�
__module__� __qualname__�namer �REALTIME_TITLErA �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 r
r3 r r7 r r@ rD r r r r r s0
� 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_INCOMPATIBLErG rA )�.0rK �clsr' r r �
<listcomp>� s ����z9RealtimeVariant.incompatible_services.<locals>.<listcomp>)�tuple�other_variants�items�superr3 )r �incompatible_variants�� __class__r' r r3 s
�� z%RealtimeVariant.incompatible_services)rH rI rJ r[ r r
r3 �
__classcell__r r rh r r\ ~ s $r\ c @ �$ e Zd ZdZejZejZdZ dZ
dS )r" �genericTN) rH rI rJ r# r �REALTIME_GENERIC_TITLErA � REALTIME_GENERIC_DESCRIPTIONrN �
is_variant� check_packages_are_installedr r r r r" � � r" c @ rk )r$ z nvidia-tegraTN) rH rI rJ r# r �REALTIME_NVIDIA_TITLErA �REALTIME_NVIDIA_DESCRIPTIONrN ro rp r r r r r$ � rq r$ c @ s2 e Zd ZdZejZejZdZ dZ
de fdd�Z dS )r% �raspiTr c
C s\ d}zt �|��� �� }d|v pd|v W S ty- } z
t�d|� W Y d }~dS d }~ww )Nz/proc/device-tree/modelzraspberry pi 4zraspberry pi 5z)Error while detecting if raspberry pi: %rF)r
� load_file�strip�lower� Exception�LOG�info)r �proc_file_path�model�er r r �variant_auto_select� s ��z'RaspberryPiRealtime.variant_auto_selectN)
rH rI rJ r# r �REALTIME_RASPI_TITLErA �REALTIME_RASPI_DESCRIPTIONrN ro rp rY r~ r r r r r% � s 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�,)rA �vendor�supported_vendors) r
� get_cpu_info� vendor_id�getr �INAPPLICABLE_VENDOR_NAMErG rA �join)r r� r� r� r r r �verify_platform_checks� s
�z(IntelIotgRealtime.verify_platform_checksN)rH rI rJ r# r �REALTIME_INTEL_TITLErA �REALTIME_INTEL_DESCRIPTIONrN ro rp r rZ r r rY r � NamedMessager� r r r r r& � s
��r& )#�logging�typingr r r r r �uaclientr r r r
r �uaclient.entitlementsr �uaclient.entitlements.baser
r �uaclient.typesr r r �get_event_loggerr � getLogger� replace_top_level_logger_namerH ry �RepoEntitlementr r\ r" r$ r% r& r r r r �<module> s m