File: //lib/python3/dist-packages/uaclient/entitlements/__pycache__/esm.cpython-310.pyc
o
S��cv � @ 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 s^ e Zd ZdZedeee df fdd��Zdde de f� fdd �
Z
dd d �Z dd
d�Z � Z
S )�ESMBaseEntitlementz https://ubuntu.com/security/esm�return.c C s ddl m}m} ||fS )Nr )�ROSEntitlement�ROSUpdatesEntitlement)�uaclient.entitlements.rosr r )�selfr r � r �;/usr/lib/python3/dist-packages/uaclient/entitlements/esm.py�dependent_services s z%ESMBaseEntitlement.dependent_servicesF�silentc s4 ddl m} t� j|d�}|r|| j� | �� |S �Nr )� update_apt_and_motd_messages)r )� uaclient.jobs.update_messagingr �super�_perform_enable�cfg�disable_local_esm_repo)r r r �enable_performed�� __class__r r r s
z"ESMBaseEntitlement._perform_enableNc C s� t �� d }| j�d�sJ �| jtd�d � }tj�t| j j
| jd� �}| j }tj� |�r0d S t �
|tj
||d�� tj�t|�}tj�tt | �}tjtj�|�dd� t�||� d S )N�serieszesm-��name)r$ r" T)�exist_ok)r �get_platform_infor$ �
startswith�len�os�path�normpathr
�repo_list_file_tmpl�format�
repo_key_file�exists�
write_filer �joinr r �makedirs�dirnamer �export_gpg_key)r r" �esm_name�
repo_filename� keyring_file�source_keyring_file�destination_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 �ensure_file_absent)r r7 r6 r r r r < s ��
z)ESMBaseEntitlement.disable_local_esm_repo�F)r N)�__name__�
__module__� __qualname__� help_doc_url�propertyr r r r �boolr r: r �
__classcell__r r r r r s
r c �J e Zd ZdZdZdZdZdZ d dee e
de f f f� fd d
�
Z � Z
S ) �ESMAppsEntitlement�
UbuntuESMAppszesm-appszUbuntu Pro: ESM Appsz.Expanded Security Maintenance for Applicationsz ubuntu-advantage-esm-apps.gpgFr Nc �D ddl m} t� j|d�\}}|r || j� t�� r | �� ||fS r )r r r �disabler r �is_current_series_ltsr: �r r r �disable_performed�failr r r rH N �
zESMAppsEntitlement.disabler< )r= r> r? �originr$ �title� descriptionr. r rB r r
rH rC r r r r rE G � ��rE c rD ) �ESMInfraEntitlementz esm-infra� UbuntuESMzUbuntu Pro: ESM Infraz0Expanded Security Maintenance for Infrastructurez%ubuntu-advantage-esm-infra-trusty.gpgFr Nc rG r )r r r rH r r � is_current_series_active_esmr: rJ r r r rH b rM zESMInfraEntitlement.disabler< )r= r> r? r$ rN rO rP r. r rB r r
rH rC r r r r rR [ rQ rR )r) �typingr r r �uaclientr r � uaclient.aptr r r �uaclient.defaultsr
�uaclient.entitlementsr �uaclient.entitlements.baser �(uaclient.entitlements.entitlement_statusr
�RepoEntitlementr rE rR r r r r �<module> s ;