Current File : //usr/lib/python3.9/site-packages/cloudinit/config/__pycache__/cc_ubuntu_pro.cpython-39.pyc
a
'�Dg�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| vr d S | d }t |t�s@dt|�j� �}t�|� t|��d|vrLd 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 �B/usr/lib/python3.9/site-packages/cloudinit/config/cc_ubuntu_pro.py�validate_schema_features* s(
��
��
r )�
pro_configc
C s� g }t | �� �D ]x\}}|tvr0t�d|� qn
|du r:qz&t|�}|jdvr^|�d|� �� W q tt fy� |�d|� �� Y q0 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)�http�httpsz-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 �scheme�append�AttributeError�
ValueError�format�join)r �errors�key�valueZ
parsed_urlr r r � supplemental_schema_validationE s,
�
��r0 c C s� | d u r d S t | t�s,tdt| �j� d���t| � g }t| �� �D �]\}}d }i }|d u rxt� d|� ddd|g}nh|� d�}t� d|� t
� d |�r�|� d
t
� |�� �}n|� d
|� �}ddd |g}d |d d
� |g i}zt
j
|fi |�� W qD t
j�yJ } z8t|�} |d u�r(| �|t�} |�|| f� W Y d }~qDd }~0 0 qD|�r�|D ]\}
} t�d|
| � �qXtd�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>� � z!set_pro_config.<locals>.<genexpr>)r r r r r r0 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_msgr9 r r r r �set_pro_configl sL
��
& ��rH c
C sz |du rg }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 nV t
j y� } z<t|�� | t�}d
|� �}t
�t|� t|�|�W Y d}~n
d}~0 0 |�sdS g d�| }z t
j
|dd
dhd�\}} W n< t
j �yh } z tdd� |� �|�W Y d}~n
d}~0 0 zt�|�}
W n8 tj�y� } z td|� ��|�W Y d}~n
d}~0 0 g } |
�dg �D ]2}|d dk�r�t�d|d � �q�| �|� �q�| �rvg } | D ]^}|�d�}
|
du�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 servicesr1 Zattachz--no-auto-enabler6 zAttaching to Ubuntu Pro. %s� r � )�rcsr5 z Failure attaching Ubuntu Pro:
)r1 �enablez --assume-yesz--format�jsonT� )ZcapturerK z!Error while enabling service(s): r7 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 rB r r& �listr r r r= r, r rA rC r �logexcr rM �loadsZJSONDecodeError�getr( )�tokenrL Z
attach_cmdZ redacted_cmdrF �errr �cmdZ
enable_stdoutr: Z enable_resprD Zerror_servicesrO r r r �
configure_pro� s�
�
�
����
��rX ��cloudc C sv t �d�rdS z| j�� W n ty< t�td� � Y n0 z| j�dg� W n typ t�td� � Y n0 dS )z.Install ubuntu-advantage-tools if not present.r1 NzPackage update failedzubuntu-advantage-toolsz(Failed to install ubuntu-advantage-tools) r �whichZdistroZupdate_package_sources� Exceptionr rR r Zinstall_packagesrY r r r �maybe_install_ua_tools� s
r] )r �returnc
C s� t | �di ��dd��}|r dS ddlm} ddlm} z8tjddd �� |� }W d � n1 sd0 Y W n< |y� } z$t� d
|� t�
t � W Y d }~dS d }~0 0 |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 rT �uaclient.api.exceptionsr_ Z4uaclient.api.u.pro.attach.auto.should_auto_attach.v1r` r �Timedr r= r& � ERROR_MSG_SHOULD_AUTO_ATTACH)r r r_ r` �result�exr r r �_should_auto_attach s$ � �(
rh c C sZ | � d�}|s$d}t�|� t|��| � d�}|rDt�dd�|�� t|| � d�d� d S )NrU z8`ubuntu_pro.token` required in non-Pro Ubuntu instances.� enable_betaz<Ignoring `ubuntu_pro.enable_beta` services in Pro attach: %sr7 rL )rU rL )rT r r r r= r, rX )r rU r ri r r r �_attach( s
�rj c
C s� ddl m}m} ddlm}m} | �d�}| �d�}|||d�}z<tjddd �� ||d
� W d � n1 sp0 Y W nn |y� |d us�|d ur�t �
d � Y nD |y� } z,d |j � �} t � | � t
| �|�W Y d }~n
d }~0 0 d S )
Nr )�AlreadyAttachedErrorr_ )�FullAutoAttachOptions�full_auto_attachrL ri )rL ri zAttaching to Ubuntu Prora rb )�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`: )rc rk r_ Z2uaclient.api.u.pro.attach.auto.full_auto_attach.v1rl rm rT r rd r r& r r r )
r rk r_ rl rm rL ri rn rg r r r r � _auto_attach7 s(
�, �
ro )�name�cfgrZ �argsr^ c C s0 d }t t�|��}|rRt|�dkr4tdd�|� ��t�dd�|�� ||d }d|v rv|rnt�d|d � |d }|d u r�t�d| � d S t |t
�s�d t |�j � �}t�
|� t|��d
|v r�d }t�
|� t|��t|� t|�d �� t|� t�d
� t|��rt|� n|�� dhk�s,t|� d S )NrN zLUnable to configure Ubuntu Pro. Multiple deprecated config keys provided: %sr7 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"r2 zETo discover more log info, please check /var/log/ubuntu-advantage.logr )rQ �DEPRECATED_KEYS� intersection�lenr r, r r& r= r r r r r r] rH rT r rh ro �keysrj )rp rq rZ rr r Z
deprecatedr r r r �handleY sb ��� ��
��
�
�
rw )N)N),�__doc__rM Zloggingr> �typingr r � urllib.parser Z cloudinitr r r Zcloudinit.cloudr Zcloudinit.configr Zcloudinit.config.schemar
Zcloudinit.settingsr r4 rs rQ r �__annotations__Z getLoggerr r r re r% r r r0 rH rX r] r rh rj ro rB rw r r r r �<module> s< �
�
',
d "
Mini Shell