File: //proc/self/root/lib/python3/dist-packages/uaclient/entitlements/__pycache__/esm.cpython-38.pyc
U
��Jh� � @ s� d dl Z d dlmZmZmZ d dlmZmZmZ d dl m
Z
m Z m Z d dl
mZ d dlmZ d dlmZ d dlmZ d d lmZ G d
d � d ej�ZG d d
� d
e�ZG dd� de�ZdS )� N)�Tuple�Type�Union)�api�messages�system)� APT_KEYS_DIR�DEB822_REPO_FILE_CONTENT� KEYRINGS_DIR)�ESM_APT_ROOTDIR)�repo)�
UAEntitlement)�CanDisableFailure)�set_filename_extensionc sl e Zd ZejjZdZee e
e df d�dd��Z e
jed�� fdd� Zd d�d
d �Zd d�d d
�Z� ZS )�ESMBaseEntitlementT.)�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_services��progressr c s2 ddl m} t� �|�}|r.|| j� | �� |S �Nr )�update_motd_messages)� uaclient.timer.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 �� j}| j�d�st�| jtd�d � }ttj �
t | j �d�}ttj �
t | j �d�}tj �
|�sttj �
|�rxd S dj|d�}dj||d�}t �|tj||t| jdd �� d S )
Nzesm-�sources�listz$https://esm.ubuntu.com/{name}/ubuntu)�namez0{series}-{name}-security {series}-{name}-updates)�seriesr% � )Zurl�suitesZ keyrings_dir� keyring_fileZdeb_src)r Zget_release_infor& r% �
startswith�AssertionError�lenr �os�path�normpathr � repo_file�exists�formatZ
write_filer r
�
repo_key_file)r r& Zesm_nameZsources_repo_filenameZlist_repo_filenameZesm_urlr( r r r �setup_local_esm_repo# sD
����� ���z'ESMBaseEntitlement.setup_local_esm_repoc C sV t j�tt | j �}t�|� t j�t| j �}t�t |d�� t�t |d�� d S )Nr# r$ )
r- r. r/ r r r3 r Zensure_file_absentr0 r )r r) Z
repo_filenamer r r r L s �
���z)ESMBaseEntitlement.disable_local_esm_repo)�__name__�
__module__� __qualname__r ZurlsZ
ESM_HOME_PAGEZ help_doc_urlZcheck_updates_pocket�propertyr r r
r r �ProgressWrapper�boolr r4 r �
__classcell__r r r! r r
s )r c sT e Zd ZdZdZejZejZ ej
Z dZ e
jeeedef f d�� fdd� Z� ZS )�ESMAppsEntitlementZ
UbuntuESMAppszesm-appszubuntu-pro-esm-apps.gpgNr c sB ddl m} t� �|�\}}|r:|| j� t�� r:| �� ||fS r )r r r �disabler r Zis_current_series_ltsr4 �r r r Zdisable_performedZfailr! r r r= f s
zESMAppsEntitlement.disable)r5 r6 r7 �originr% r ZESM_APPS_TITLE�titleZESM_APPS_DESCRIPTION� descriptionZESM_APPS_HELP_TEXT� help_textr3 r r9 r r: r r r= r; r r r! r r< ^ s �r< c sT e Zd ZdZdZejZejZ ej
Z dZ e
jeeedef f d�� fdd� Z� ZS )�ESMInfraEntitlementz esm-infraZ UbuntuESMzubuntu-pro-esm-infra.gpgNr c sB ddl m} t� �|�\}}|r:|| j� t�� r:| �� ||fS r )r r r r= r r Z is_current_series_active_esmr4 r> r! r r r= { s
zESMInfraEntitlement.disable)r5 r6 r7 r% r? r ZESM_INFRA_TITLEr@ ZESM_INFRA_DESCRIPTIONrA ZESM_INFRA_HELP_TEXTrB r3 r r9 r r: r r r= r; r r r! r rC s s �rC )r- �typingr r r Zuaclientr r r Z uaclient.aptr r r
Zuaclient.defaultsr Zuaclient.entitlementsr Zuaclient.entitlements.baser
Z(uaclient.entitlements.entitlement_statusr Z
uaclient.utilr ZRepoEntitlementr r< rC r r r r �<module> s Q