← Back
Editing: realtime.cpython-38.pyc
U ��gn � @ 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
Save File
Cancel