File: //lib/python3/dist-packages/uaclient/entitlements/__pycache__/realtime.cpython-38.pyc
U
��Jhn � @ 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��ZddiZ G d d
� d
ej �Z G d d � d e �Z G d
d� de �Z G dd� 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�StaticAffordanceZ GenuineIntelZintelc @ s� e Zd ZdZejZejZej Z
ej j Z
dZdZdZdZdZed�dd�Zeeee f d�d d
�Zed d � �Zee e d
f d�dd��Z ee e d
f d�dd��Z ee!d�dd��Z"dd�dd�Z#dS )�RealtimeKernelEntitlementzrealtime-kernelz ubuntu-pro-realtime-kernel.gpgTFZUbuntuRealtimeKernel��returnc C s( t jt| j�tdg�d�}t�|� |S )z%Check if system needs to be rebooted.zlinux-.*-realtime)Zinstalled_pkgsZinstalled_pkgs_regex)r
Z
should_reboot�set�packages�eventZ needs_reboot)�selfZreboot_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 r
d S tS r )� access_onlyr r" r r r �default_variant0 s z)RealtimeKernelEntitlement.default_variant.c C s> ddl m}m} ddlm} t|tj�t|tj�t|tj �fS )Nr )�FIPSEntitlement�FIPSUpdatesEntitlement)�LivepatchEntitlement)
Zuaclient.entitlements.fipsr& r' Z uaclient.entitlements.livepatchr( r
r ZREALTIME_FIPS_INCOMPATIBLEZ"REALTIME_FIPS_UPDATES_INCOMPATIBLEZ REALTIME_LIVEPATCH_INCOMPATIBLE)r r& r' r( r r r �incompatible_services6 s �� ��z/RealtimeKernelEntitlement.incompatible_servicesc C s t jdd� dffS )Nc S s t �� S r )r
Z is_containerr r r r �<lambda>P � z>RealtimeKernelEntitlement.static_affordances.<locals>.<lambda>F)r Z#REALTIME_ERROR_INSTALL_ON_CONTAINERr" r r r �static_affordancesK s
��z,RealtimeKernelEntitlement.static_affordancesc C sD d }| j s tjtjdd�fg}d }| js:tjdtjifg}||d�S )NT)�msg�defaultr- )�
pre_enable� pre_disable)r$ r �prompt_for_confirmationr ZREALTIME_PROMPT�purgeZREALTIME_PRE_DISABLE_PROMPT)r r/ r0 r r r � messagingU s& ���
���
�z#RealtimeKernelEntitlement.messagingNc C s< t | j��t t�� ��}|r8t�t|�tjj | j
d�� d S )N)�title) r r � intersectionr Z get_installed_packages_names�remove_packages�listr ZDISABLE_FAILED_TMPL�formatr4 )r r r r r r6 u s
��z)RealtimeKernelEntitlement.remove_packages)$�__name__�
__module__� __qualname__�namer ZREALTIME_TITLEr4 ZREALTIME_DESCRIPTION� descriptionZREALTIME_HELP_TEXTZ help_textZurlsZREALTIME_HOME_PAGEZ help_doc_urlZ
repo_key_fileZapt_noninteractiveZsupports_access_onlyZsupports_purge�origin�boolr r �strr r r# �propertyr% r r
r) r r, r r3 r6 r r r r r s,
� r c s. e Zd Zeeedf d�� fdd� �Z� ZS )�RealtimeVariant.r c s( t � fdd�� j�� D ��}t� j| S )Nc s* g | ]"\}}t |tjj� j|jd ���qS ))�service�variant)r
r Z REALTIME_VARIANT_INCOMPATIBLEr8 r4 )�.0r< �clsr" r r �
<listcomp>� s � ��z9RealtimeVariant.incompatible_services.<locals>.<listcomp>)�tupleZother_variants�items�superr) )r Zincompatible_variants�� __class__r" r r) � s
�� z%RealtimeVariant.incompatible_services)r9 r: r; rA r r
r) �
__classcell__r r rK r rB � s rB c s> e Zd ZdZejZejZdZ dZ
e e d�� fdd� �Z
� ZS )r �genericTr c sd t � j}t�� j}|dkr`|�d�p&g }tjj| j | j
d|p>dd�}|� t j
d|if� ||d< |S )NrN r/ �unknown)rD rC Z base_flavor�current_flavorr- )rJ r3 r
Zget_kernel_infoZflavor�getr Z#KERNEL_FLAVOR_CHANGE_WARNING_PROMPTr8 r r< �appendr r1 )r r3 rP r/ r- rK r r r3 � s"
���zGenericRealtime.messaging)r9 r: r; r r ZREALTIME_GENERIC_TITLEr4 Z REALTIME_GENERIC_DESCRIPTIONr= �
is_variant� check_packages_are_installedrA r r3 rM r r rK r r � s �r c @ s$ e Zd ZdZejZejZdZ dZ
dS )r z nvidia-tegraTN) r9 r: r; r r ZREALTIME_NVIDIA_TITLEr4 ZREALTIME_NVIDIA_DESCRIPTIONr= rS rT r r r r r � s
r c @ s2 e Zd ZdZejZejZdZ dZ
e d�dd�Z dS )r ZraspiTr c
C sb d}z$t �|��� �� }d|kp&d|kW S tk
r\ } zt�d|� W Y �dS d }~X Y nX d S )Nz/proc/device-tree/modelzraspberry pi 4zraspberry pi 5z)Error while detecting if raspberry pi: %rF)r
Z load_file�strip�lower� Exception�LOG�info)r Zproc_file_pathZmodel�er r r �variant_auto_select� s z'RaspberryPiRealtime.variant_auto_selectN)
r9 r: r; r r ZREALTIME_RASPI_TITLEr4 ZREALTIME_RASPI_DESCRIPTIONr= rS rT r? r[ r r r r r � s r c @ sJ e Zd ZdZejZejZdZ dZ
e e e
f eeeej f d�dd�ZdS )r! z
intel-iotgT)�platform_checksr c C sT t �� jp d}t�||�}|�dg �}||kr2dS dtjj| j|d� |�d�fS d S )N� �cpu_vendor_ids)TNF�,)r4 ZvendorZsupported_vendors)
r
Z get_cpu_infoZcpuinfo_vendor_id�CPU_VENDOR_MAPrQ r ZINAPPLICABLE_VENDOR_NAMEr8 r4 �join)r r\ Z vendor_idr^ r r r �verify_platform_checks� s �z(IntelIotgRealtime.verify_platform_checksN)r9 r: r; r r ZREALTIME_INTEL_TITLEr4 ZREALTIME_INTEL_DESCRIPTIONr= rS rT r r@ r r r? r Z NamedMessagerb r r r r r! � s
�r! )$Zlogging�typingr r r r r Zuaclientr r r r
r Zuaclient.entitlementsr Zuaclient.entitlements.baser
r Zuaclient.typesr r r Zget_event_loggerr Z getLoggerZ replace_top_level_logger_namer9 rX r` ZRepoEntitlementr rB r r r r! r r r r �<module> s m