Your IP : 216.73.216.52
U
8�-dF � @ s� d dl Z d dlmZmZmZ d dlmZmZ d dlm Z m
Z
m Z d dl m
Z
d dlmZ d dlmZ d dlmZ G d d
� d
ej�ZG d d � d e�ZG d
d� de�ZdS )� N)�Tuple�Type�Union)�gpg�system)� APT_KEYS_DIR�ESM_REPO_FILE_CONTENT� KEYRINGS_DIR)�ESM_APT_ROOTDIR)�repo)�
UAEntitlement)�CanDisableFailurec sd e Zd ZdZeeee df d�dd��Zde e d�� fdd �
Z
d
d�d d �Z d
d�d
d�Z � Z
S )�ESMBaseEntitlementz https://ubuntu.com/security/esm.��returnc C s ddl m}m} ||fS )Nr )�ROSEntitlement�ROSUpdatesEntitlement)Zuaclient.entitlements.rosr r )�selfr r � r �;/usr/lib/python3/dist-packages/uaclient/entitlements/esm.py�dependent_services s z%ESMBaseEntitlement.dependent_servicesF)�silentr c s4 ddl m} t� j|d�}|r0|| j� | �� |S �Nr )�update_motd_messages)r )� uaclient.jobs.update_messagingr �super�_perform_enable�cfg�disable_local_esm_repo)r r r Zenable_performed�� __class__r r r s
z"ESMBaseEntitlement._perform_enableNc C s� t �� d }| j�d�s t�| jtd�d � }tj�t | j
j | jd� �}| j }tj�
|�r`d S t �|tj ||d�� tj�t|�}tj�t t | �}tjtj�|�dd� t�||� d S )N�serieszesm-��name)r# r! T)�exist_ok)r Zget_platform_infor# �
startswith�AssertionError�len�os�path�normpathr
�repo_list_file_tmpl�format�
repo_key_file�existsZ
write_filer �joinr r �makedirs�dirnamer Zexport_gpg_key)r r! Zesm_name�
repo_filename� keyring_fileZsource_keyring_fileZdestination_keyring_filer r r �setup_local_esm_repo! s& � �
�z'ESMBaseEntitlement.setup_local_esm_repoc C sJ t j�tt | j �}t j�t| jj| jd� �}t �
|� t �
|� d S )Nr" ) r( r) r* r
r r- r+ r, r# r Zensure_file_absent)r r3 r2 r r r r < s ��
z)ESMBaseEntitlement.disable_local_esm_repo)F)�__name__�
__module__� __qualname__Z help_doc_url�propertyr r r r �boolr r4 r �
__classcell__r r r r r s r c sH e Zd ZdZdZdZdZdZd ee e
de f f d�� fd d
�
Z � Z
S ) �ESMAppsEntitlementZ
UbuntuESMAppszesm-appszUbuntu Pro: ESM Appsz.Expanded Security Maintenance for Applicationsz ubuntu-advantage-esm-apps.gpgFNr c sD ddl m} t� j|d�\}}|r<|| j� t�� r<| �� ||fS r )r r r �disabler r Zis_current_series_ltsr4 �r r r Zdisable_performedZfailr r r r< N s
zESMAppsEntitlement.disable)F)r5 r6 r7 �originr# �title� descriptionr- r r9 r r
r< r: r r r r r; G s ��r; c sH e Zd ZdZdZdZdZdZd ee e
de f f d�� fd d
�
Z � Z
S ) �ESMInfraEntitlementz esm-infraZ UbuntuESMzUbuntu Pro: ESM Infraz0Expanded Security Maintenance for Infrastructurez%ubuntu-advantage-esm-infra-trusty.gpgFNr c sD ddl m} t� j|d�\}}|r<|| j� t�� r<| �� ||fS r )r r r r< r r Z is_current_series_active_esmr4 r= r r r r< b s
zESMInfraEntitlement.disable)F)r5 r6 r7 r# r> r? r@ r- r r9 r r
r< r: r r r r rA [ s ��rA )r( �typingr r r Zuaclientr r Z uaclient.aptr r r Zuaclient.defaultsr
Zuaclient.entitlementsr Zuaclient.entitlements.baser Z(uaclient.entitlements.entitlement_statusr
ZRepoEntitlementr r; rA r r r r �<module> s ;