File: //usr/lib/python3/dist-packages/cloudinit/config/__pycache__/cc_ubuntu_pro.cpython-38.pyc
U
Ӈg�6 � @ sL U d Z ddlZddlZddlZddlmZmZ ddlmZ ddl m
Z
m Z m Z ddl
mZ ddlmZ ddlmZ dd lmZ ed
d g�Zd d
gedgee� d�Zeed< e�e�Z dZ dZ dZ e d�dd�Z!e d�dd�Z"d+ed�dd�Z#d,d d �Z$ed �d d �Z%e e&d!�d"d#�Z'e d�d$d%�Z(e d�d&d'�Z)e*eeedd(�d)d*�Z+dS )-z1ubuntu_pro: Configure Ubuntu Pro support services� N)�Any�List)�urlparse)� performance�subp�util)�Cloud)�Config)�
MetaSchema)� PER_INSTANCEzubuntu-advantageZubuntu_advantageZ
cc_ubuntu_proZubuntu�
ubuntu_pro)�idZdistrosZ frequencyZactivate_by_schema_keys�meta�REDACTEDzUUnable to determine if this is an Ubuntu Pro instance. Fallback to normal Pro attach.)Z
http_proxyZ https_proxyZglobal_apt_http_proxyZglobal_apt_https_proxyZua_apt_http_proxyZua_apt_https_proxy)� pro_sectionc C s� d| kr d S | d }t |t�s@dt|�j� �}t�|� t|��d|krLd S |d }t |t�s�dt|�j� �}t�|� t|��d S )N�featuresz.'ubuntu_pro.features' should be a dict, not a �disable_auto_attachzB'ubuntu_pro.features.disable_auto_attach' should be a bool, not a )�
isinstance�dict�type�__name__�LOG�error� RuntimeError�bool)r r �msgr � r �@/usr/lib/python3/dist-packages/cloudinit/config/cc_ubuntu_pro.py�validate_schema_features* s
�
�
r )�
pro_configc
C s� g }t | �� �D ]z\}}|tkr0t�d|� qn
|dkr:qz&t|�}|jdkr^|�d|� �� W q tt fk
r� |�d|� �� Y qX q|r�t d�
d� |����dS )a� Validate user-provided ua:config option values.
This function supplements flexible jsonschema validation with specific
value checks to aid in triage of invalid user-provided configuration.
Note: It does not log/raise config values as they could be urls containing
sensitive auth info.
@param pro_config: Dictionary of config value under 'ubuntu_pro'.
@raises: ValueError describing invalid values provided.
z4Not validating unknown ubuntu_pro.config.%s propertyN)ZhttpZhttpsz-Expected URL scheme http/https for ua:config:z Expected a URL for ua:config:z$Invalid ubuntu_pro configuration:
{}�
) �sorted�items�KNOWN_PRO_CONFIG_PROPSr �warningr Zscheme�append�AttributeError�
ValueError�format�join)r �errors�key�valueZ
parsed_urlr r r � supplemental_schema_validationE s,
�
��r- c C s� | d kr d S t | t�s,tdt| �j� d���t| � g }t| �� �D �] \}}d }i }|d krxt� d|� ddd|g}nh|� d�}t� d|� t
� d |�r�|� d
t
� |�� �}n|� d
|� �}ddd |g}d |d d
� |g i}zt
j
|f|� W qD t
jk
�rD } z0t|�} |d k �r&| �|t�} |�|| f� W 5 d }~X Y qDX qD|�r�|D ]\}
} t�d|
| � �qRtd�d�dd� |D �����d S )Nz+ubuntu_pro: config should be a dict, not a z%; skipping enabling config parameterszDisabling Pro config for %s�pro�configZunsetz =REDACTEDzEnabling Pro config %sz\s�=�set� logstring���z#Failure enabling/disabling "%s":
%sz3Failure enabling/disabling Ubuntu Pro config(s): {}�, c s s | ]\}}d � |�V qdS )z"{}"N)r( )�.0�param�_r r r � <genexpr>� s z!set_pro_config.<locals>.<genexpr>)r r r r r r- r! r" r �debug�re�search�escaper �ProcessExecutionError�str�replacer r% r$ r( r) ) r �
enable_errorsr+ r, Zredacted_key_valueZ subp_kwargsZ
config_cmd� key_value�e�err_msgr6 r r r r �set_pro_configl sH
�
" ��rD c
C sz |dkrg }n<t |t�r*t�d� |g}n t |t�sJt�dt|�j� g }|r\ddd| g}n
dd| g}|dd� tg }t�dd � |�� zt
j
|d
d h|d � W nT t
j k
r� } z4t|�� | t�}d
|� �}t
�t|� t|�|�W 5 d}~X Y nX |�sdS dddddg| }z t
j
|dd
dhd�\}} W n: t
j k
�rj } ztdd� |� �|�W 5 d}~X Y nX zt�|�}
W n6 tjk
�r� } ztd|� ��|�W 5 d}~X Y nX g } |
�dg �D ]2}|d dk�r�t�d|d � �q�| �|� �q�| �rvg } | D ]^}|�d �}
|
dk �r<| �|
� d |
� d |d � �}nd |d! � d |d � �}t
�t|� �qtd"d� | � ��dS )#z=Call ua command line client to attach and/or enable services.NzNubuntu_pro: enable should be a list, not a string; treating as a single enablezIubuntu_pro: enable should be a list, not a %s; skipping enabling servicesr. Zattachz--no-auto-enabler3 zAttaching to Ubuntu Pro. %s� r � )�rcsr2 z Failure attaching Ubuntu Pro:
�enablez --assume-yesz--format�jsonT� )ZcapturerG z!Error while enabling service(s): r4 zPro response was not json: r* Z message_codezservice-already-enabledz Service `%s` already enabled.�servicezFailure enabling `z`: �messagezFailure of type `r z(Failure enabling Ubuntu Pro service(s): )r r> r r$ �listr r r r9 r) r r= r? r �logexcr rI �loadsZJSONDecodeError�getr% )�tokenrH Z
attach_cmdZ redacted_cmdrB �errr �cmdZ
enable_stdoutr7 Z enable_respr@ Zerror_servicesrK r r r �
configure_pro� s�
�
�
����
��rT ��cloudc C sz t �d�rdS z| j�� W n" tk
r> t�td� � Y nX z| j�dg� W n" tk
rt t�td� � Y nX dS )z.Install ubuntu-advantage-tools if not present.r. NzPackage update failedzubuntu-advantage-toolsz(Failed to install ubuntu-advantage-tools) r ZwhichZdistroZupdate_package_sources� Exceptionr rN r Zinstall_packagesrU r r r �maybe_install_ua_tools� s
rX )r �returnc
C s� t | �di ��dd��}|r dS ddlm} ddlm} z$tjddd �� |� }W 5 Q R X W n> |k
r� } z t� d
|� t�
t � W Y �dS d }~X Y nX |jS ) Nr r Fr )�UserFacingError)�should_auto_attachz6Checking if the instance can be attached to Ubuntu Pro�always�Zlog_modez%Error during `should_auto_attach`: %s) r rP �uaclient.api.exceptionsrZ Z4uaclient.api.u.pro.attach.auto.should_auto_attach.v1r[ r �Timedr r9 r$ � ERROR_MSG_SHOULD_AUTO_ATTACH)r r rZ r[ �result�exr r r �_should_auto_attach s$ � �
rc c C sZ | � d�}|s$d}t�|� t|��| � d�}|rDt�dd�|�� t|| � d�d� d S )NrQ z8`ubuntu_pro.token` required in non-Pro Ubuntu instances.� enable_betaz<Ignoring `ubuntu_pro.enable_beta` services in Pro attach: %sr4 rH )rQ rH )rP r r r r9 r) rT )r rQ r rd r r r �_attach( s
�re c
C s� ddl m}m} ddlm}m} | �d�}| �d�}|||d�}z(tjddd �� ||d
� W 5 Q R X W nn |k
r� |d k s�|d k r�t �
d � Y nB |k
r� } z$d |j � �} t � | � t
| �|�W 5 d }~X Y nX d S )
Nr )�AlreadyAttachedErrorrZ )�FullAutoAttachOptions�full_auto_attachrH rd )rH rd zAttaching to Ubuntu Pror\ r] )�optionsz�The instance is already attached to Pro. Leaving enabled services untouched. Ignoring config directives ubuntu_pro: enable and enable_betaz!Error during `full_auto_attach`: )r^ rf rZ Z2uaclient.api.u.pro.attach.auto.full_auto_attach.v1rg rh rP r r_ r r$ r r r )
r rf rZ rg rh rH rd ri rb r r r r � _auto_attach7 s(
��
rj )�name�cfgrV �argsrY c C s0 d }t t�|��}|rRt|�dkr4tdd�|� ��t�dd�|�� ||d }d|krv|rnt�d|d � |d }|d kr�t�d| � d S t |t
�s�d t |�j � �}t�
|� t|��d
|kr�d }t�
|� t|��t|� t|�d �� t|� t�d
� t|��rt|� n|�� dhk�s,t|� d S )NrJ zLUnable to configure Ubuntu Pro. Multiple deprecated config keys provided: %sr4 z^Deprecated configuration key(s) provided: %s. Expected "ubuntu_pro"; will attempt to continue.r r z;Ignoring deprecated key %s and preferring ubuntu_pro configz=Skipping module named %s, no 'ubuntu_pro' configuration foundz%'ubuntu_pro' should be a dict, not a ZcommandszPDeprecated configuration "ubuntu-advantage: commands" provided. Expected "token"r/ zETo discover more log info, please check /var/log/ubuntu-advantage.logr )rM �DEPRECATED_KEYS� intersection�lenr r) r r$ r9 r r r r r rX rD rP r rc rj �keysre )rk rl rV rm r Z
deprecatedr r r r �handleY s^ ��� ��
�
�
�
rr )N)N),�__doc__rI Zloggingr: �typingr r Z urllib.parser Z cloudinitr r r Zcloudinit.cloudr Zcloudinit.configr Zcloudinit.config.schemar
Zcloudinit.settingsr r1 rn rM r �__annotations__Z getLoggerr r r r` r# r r r- rD rT rX r rc re rj r> rr r r r r �<module> s< �
�
',
d "