Your IP : 216.73.216.52
U
8�-dL< � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlmZmZm Z m
Z
m Z m Z d dl
mZmZmZmZmZmZmZ d dlmZ d dlmZ e�� ZG dd� dej�Z dS ) � N)�exists)�Any�Dict�List�Optional�Tuple�Union)�apt�contract� event_logger�
exceptions�messages�system�util)�base)�ApplicationStatusc s$ e Zd ZdZdZdZdZeee e
df d�dd��Z ee e
d�dd ��Z
ed�d
d �Zeeje
d�d d
���Zd#eed�dd�Zd$dd�Zeeeej f d�dd�Zdd� Zd%e e
e f e e
e f eed�� fdd�
Z d&ee e
eedd�d d �Z d'edd�d d �Z d(eed �d!d"�Z!� Z"S ))�RepoEntitlementz*/etc/apt/sources.list.d/ubuntu-{name}.listz$/etc/apt/preferences.d/ubuntu-{name}NF)�returnc C s d S �N� ��selfr r �</usr/lib/python3/dist-packages/uaclient/entitlements/repo.py�repo_pin_priority# s z!RepoEntitlement.repo_pin_priorityc C sJ g }| j jj�| ji ��di �}|rF|�di �}t�|�dg ��}|}|S )z debs to install on enablement� entitlement�
directives�additionalPackages)�cfg�machine_token_file� entitlements�get�name�copy)r �packagesr r Zadditional_packagesr r r r# ' s
� �
�zRepoEntitlement.packagesc C s t jt| j�d�}t�|� |S )z%Check if system needs to be rebooted.)Zinstalled_pkgs)r Z
should_reboot�setr# �eventZ needs_reboot)r Zreboot_requiredr r r �_check_for_reboot: s
�
z!RepoEntitlement._check_for_rebootc C s d S r r r r r r �
repo_key_fileB s z RepoEntitlement.repo_key_file)�silentr c C s� | j |d� | jrb| jrbt| j�dkr6dd�| j� nd}t�d�|�� t�t j
j| j d�� n*| � � t�t j
j| j d�� | jdd � d
S ) z�Enable specific entitlement.
@return: True on success, False otherwise.
@raises: UserFacingError on failure to install suggested packages
�r( r z: � � z Skipping installing packages{}��title�install)Z operationT)�setup_apt_configZsupports_access_onlyZ access_only�lenr# �joinr% �info�formatr
ZACCESS_ENABLED_TMPLr- �install_packagesZ ENABLED_TMPLZ_check_for_reboot_msg)r r( Z packages_strr r r �_perform_enableG s �� z RepoEntitlement._perform_enablec C s" t | d�r| �� | j|d� dS )N�remove_packagesr) T)�hasattrr6 �remove_apt_config)r r( r r r �_perform_disable] s
z RepoEntitlement._perform_disablec C s� | j jj�| ji �}|�di ��di �}|�d�}|sLtjtjj | j
d�fS t j tj
jd�}t�d� |�|�}|r�tjtjj | j
d�fS tjtjj | j
d�fS )Nr r �aptURLr, )� error_msgz {}/ubuntu)r r r r r! r �DISABLEDr
ZNO_APT_URL_FOR_SERVICEr3 r- r Zget_apt_cache_policyZAPT_POLICY_FAILED�msg�re�searchZENABLEDZSERVICE_IS_ACTIVEZSERVICE_NOT_CONFIGURED)r Zentitlement_cfgr �repo_urlZpolicy�matchr r r �application_statusc s0
� �
����z"RepoEntitlement.application_statusc C sP | j j| jd�}tdd� t�|��� �d�D ��r6dS |s>dS t|t�|�k�S )z�Check if apt url delta should be applied.
:param apt_url: string containing the apt url to be used.
:return: False if apt url is already found on the source file.
True otherwise.
�r! c s s | ]}|� d �V qdS )�#N)�
startswith)�.0�liner r r � <genexpr>� s �z<RepoEntitlement._check_apt_url_is_applied.<locals>.<genexpr>�
FT) �repo_list_file_tmplr3 r! �allr Z load_file�strip�split�bool)r Zapt_urlZapt_filer r r �_check_apt_url_is_applied� s �z)RepoEntitlement._check_apt_url_is_applied)� orig_access�deltas� allow_enabler c s
t � �|||�rdS |�di �}|�di �}|�d�}|�d�}| j�d�}|r^|r^| �� } n | �� \} }
| tjkrxdS | � |�s�t
� d| j � |�di �} | �di ��d�} | r�| j
j| j d �}
t�|
| � | �� | �� |�rt
� d
�d �|��� | j|d � dS )
a1 Process any contract access deltas for this entitlement.
:param orig_access: Dictionary containing the original
resourceEntitlement access details.
:param deltas: Dictionary which contains only the changed access keys
and values.
:param allow_enable: Boolean set True if allowed to perform the enable
operation. When False, a message will be logged to inform the user
about the recommended enabled service.
:return: True when delta operations are processed; False when noop.
Tr r r: r z status-cacheFz5Updating '%s' apt sources list on changed directives.rC z-Installing packages on changed directives: {}�, )� package_list)�super�process_contract_deltasr r Z
read_cacheZ"_check_application_status_on_cacherB r r< rO �loggingr2 r! rJ r3 r �remove_auth_apt_repor8 r/ r1 r4 )r rP rQ rR Zdelta_entitlementZdelta_directivesZ
delta_apt_urlZdelta_packagesZ status_cacherB �_Zorig_entitlementZold_url�
repo_filename�� __class__r r rV � s@
� �� z'RepoEntitlement.process_contract_deltasT)rT �cleanup_on_failure�verboser c C s� |s
| j }|sdS | j�dg �}t�|�s.dS |rFt�dj| jd�� | j r`ddi}ddd g}ni }g }z(t
j j| jd�}t j
|||j|d
� W n$ tjk
r� |r�| �� � Y nX dS ) a) Install contract recommended packages for the entitlement.
:param package_list: Optional package list to use instead of
self.packages.
:param cleanup_on_failure: Cleanup apt files if apt install fails.
:param verbose: If true, print messages to stdout
NZ pre_installzInstalling {title} packagesr, ZDEBIAN_FRONTENDZnoninteractivez--allow-downgradesz$-o Dpkg::Options::="--force-confdef"z$-o Dpkg::Options::="--force-confold")r# � apt_optionsr; �env)r# Z messagingr r Zhandle_message_operationsr% r2 r3 r- �apt_noninteractiver
�ENABLED_FAILEDr �run_apt_install_commandr= r �UserFacingErrorr8 )r rT r] r^ Zmsg_opsr` r_ r= r r r r4 � s:
��
z RepoEntitlement.install_packagesc C s� d}d}d}| j js | j jrNt�d| j jtj�}t�d| j jtj�}tjj }n@| j j
s^| j j r�t�d| j j
tj�}t�d| j j tj�}tjj }tj
|||d� | jj| jd�}| j jj�| j�}|d �di �}|d �di �}|�d �} | �sJ| j jd
}
|�d ��s2t�| j �} | �|
| j�} | �r2| �d �} | �sJ|
} t�d | j� |�d
�}
|
�slt� d�| j���|�d�}|�s�t� | j��|�d�}|�s�t� d�| j| j j ���| j �r| j �s�t� d�| j t!j"j| jd�j#���| j$j| jd�}t�%||| j | j � g }t&tj'��s |�(d� t&tj)��s6|�(d� |�r�|�sXt*�+d�d�,|��� ztj-|d� W n" tj k
�r� | �.� � Y nX t�/||| || j0� |�s�t*�+t!j1� z t�2� W n& tj k
�r� | j.dd� � Y nX dS )z�Setup apt config based on the resourceToken and directives.
Also sets up apt proxy if necessary.
:raise UserFacingError: on failure to setup any aspect of this apt
configuration
NZhttpZhttps)�
http_proxy� https_proxyZ proxy_scoperC r r � obligationsZ
resourceTokenZ machineTokenZenableByDefaultzWNo resourceToken present in contract for service %s. Using machine token as credentials�aptKeyz6Ubuntu Pro server provided no aptKey directive for {}.r: Zsuitesz%Empty {} apt suites directive from {}z:Cannot setup apt pin. Empty apt repo origin value '{}'.
{}r, zapt-transport-httpszca-certificatesz Installing prerequisites: {}rS )r# F)�run_apt_update)3r Zglobal_apt_http_proxyZglobal_apt_https_proxyr Zvalidate_proxyZ PROXY_VALIDATION_APT_HTTP_URLZ PROXY_VALIDATION_APT_HTTPS_URLr Z
AptProxyScopeZGLOBALZua_apt_http_proxyZua_apt_https_proxyZUACLIENTZsetup_apt_proxyrJ r3 r! r r r �
machine_tokenr
ZUAContractClientZ request_resource_machine_accessrW Zwarningr- r rd �MissingAptURLDirectiveZ contract_urlr �originr
rb r= �repo_pref_file_tmplZadd_ppa_pinningr ZAPT_METHOD_HTTPS_FILE�appendZCA_CERTIFICATES_FILEr% r2 r1 rc r8 Zadd_auth_apt_repor' �APT_UPDATING_LISTS�run_apt_update_command)r r( re rf ZscoperZ Z resource_cfgr rg �tokenrj ZclientZmachine_accessrh r@ Z repo_suites�repo_pref_fileZprerequisite_pkgsr r r r/
s� ��
�� �
�
�
��
�����
�� � z RepoEntitlement.setup_apt_config)ri r( c C s� t �� d }| jj| jd�}| jjj| j �di �}|�di �}|�d�}|sZt �
| j��t � ||| j
� t �||� | jr�| jj| jd�}t �|� |r�|s�t�tj� t �� dS )z�Remove any repository apt configuration files.
:param run_apt_update: If after removing the apt update
command after removing the apt files.
�seriesrC r r r: N)r Zget_platform_inforJ r3 r! r r r r r rk r rX r' Zremove_apt_list_filesr rm Zensure_file_absentr% r2 r
ro rp ) r ri r( rs rZ r Zaccess_directivesr@ rr r r r r8 � s& �
z!RepoEntitlement.remove_apt_config)F)F)F)NTT)F)TF)#�__name__�
__module__� __qualname__rJ rm rl ra �propertyr �int�strr r r# rN r& �abc�abstractmethodr' r5 r9 r r r r
Z NamedMessagerB rO r r rV r4 r/ r8 �
__classcell__r r r[ r r sR
� �
�? �
� 3 � �r ) rz r" rW r> �os.pathr �typingr r r r r r Zuaclientr r
r r r
r r Zuaclient.entitlementsr Z(uaclient.entitlements.entitlement_statusr Zget_event_loggerr% Z
UAEntitlementr r r r r �<module> s $