Your IP : 216.73.216.52
U
8�-d� � @ sz d dl mZmZ d dlmZmZmZmZmZ d dl m
Z
d dl m Z d dl
mZmZmZ e�� ZdZG dd� de
j�Zd S )
� )�Optional�Tuple)�apt� event_logger�messages�system�util)�repo)�IncompatibleService)�MessagingOperations�MessagingOperationsDict�StaticAffordancez"https://ubuntu.com/realtime-kernelc @ s� e Zd ZdZdZdZeZdZdZ dZ
e d�dd�Z e
eed f d�d
d ��Ze
eed f d�d d
��Ze
ed�dd��Zdd�dd�ZdS )�RealtimeKernelEntitlementzrealtime-kernelzReal-time kernelz0Ubuntu kernel with PREEMPT_RT patches integratedz$ubuntu-advantage-realtime-kernel.gpgT)�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_reboot.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_services# s �� ��z/RealtimeKernelEntitlement.incompatible_servicesc C s t jdd� dffS )Nc S s t �� S )N)r Z is_containerr r r r �<lambda>= � z>RealtimeKernelEntitlement.static_affordances.<locals>.<lambda>F)r Z#REALTIME_ERROR_INSTALL_ON_CONTAINER)r r r r �static_affordances8 s
��z,RealtimeKernelEntitlement.static_affordancesc C s> d }| j s"tjtj| jdd�fg}|tjtj| jd�fgd�S )NT)�msg�
assume_yes�default)r r )�
pre_enableZ pre_disable)Z access_onlyr Zprompt_for_confirmationr ZREALTIME_PROMPTr ZREALTIME_PRE_DISABLE_PROMPT)r r! r r r � messagingB 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�formatr# )r r r r r r% _ s
��z)RealtimeKernelEntitlement.remove_packages)�__name__�
__module__� __qualname__�namer# Z description�REALTIME_KERNEL_DOCS_URLZ help_doc_urlZ
repo_key_fileZapt_noninteractiveZsupports_access_only�boolr �propertyr r
r r
r r r" r% r r r r r s � r N)�typingr r Zuaclientr r r r r Zuaclient.entitlementsr Zuaclient.entitlements.baser
Zuaclient.typesr r r
Zget_event_loggerr r, ZRepoEntitlementr r r r r �<module> s