File: //lib/python3/dist-packages/cloudinit/config/__pycache__/cc_ubuntu_pro.cpython-310.pyc
o
x[h�6 �
@ sV 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 de fdd�Z!de fdd�Z"d-defdd�Z#d-d d �Z$d efd d �Z%de d!e&fd"d#�Z'de fd$d%�Z(de fd&d'�Z)d(e*d)ed ed*ed!df
d+d,�Z+dS ).z1ubuntu_pro: Configure Ubuntu Pro support services� N)�Any�List)�urlparse)� performance�subp�util)�Cloud)�Config)�
MetaSchema)� PER_INSTANCEzubuntu-advantage�ubuntu_advantage�
cc_ubuntu_pro�ubuntu�
ubuntu_pro)�id�distros� frequency�activate_by_schema_keys�meta�REDACTEDzUUnable to determine if this is an Ubuntu Pro instance. Fallback to normal Pro attach.)�
http_proxy� https_proxy�global_apt_http_proxy�global_apt_https_proxy�ua_apt_http_proxy�ua_apt_https_proxy� pro_sectionc C s� d| vrd S | d }t |t�s dt|�j� �}t�|� t|��d|vr&d 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 ]:\}}|tvrt�d|� q|du r qzt|�}|jdvr.|�d|� �� W q tt fyB |�d|� �� Y qw |rOt 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�value�
parsed_urlr( r( r) � supplemental_schema_validationE s2
�
�����r= c C s� | d u rd S t | t�stdt| �j� d���t| � g }t| �� �D ]�\}}d }i }|d u r;t� d|� ddd|g}n4|� d�}t� d|� t
� d |�rW|� d
t
� |�� �}n|� d
|� �}ddd |g}d |d d
� |g i}z t
j
|fi |�� W q" t
jy� } zt|�} |d ur�| �|t�} |�|| f� W Y d }~q"d }~ww |r�|D ] \}
} t�d|
| � q�td�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�config�unsetz =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)r7 )�.0�param�_r( r( r) � <genexpr>� s � z!set_pro_config.<locals>.<genexpr>)r r r% r! r" r= r/ r0 r# �debug�re�search�escaper �ProcessExecutionError�str�replacer r4 r2 r7 r8 ) r+ �
enable_errorsr: r; �redacted_key_value� subp_kwargs�
config_cmd� key_value�e�err_msgrG r$ r( r( r) �set_pro_configl sR
��
�� ���rX c
C sH |du rg }n t |t�rt�d� |g}nt |t�s%t�dt|�j� g }|r.ddd| g}ndd| g}|dd� tg }t�dd � |�� z t
j
|d
d h|d � W n% t
j yv } zt|�� | t�}d
|� �}t
�t|� t|�|�d}~ww |s{dS g d�| }zt
j
|dd
dhd�\}} W n t
j y� } z
tdd� |� �|�d}~ww zt�|�}
W n tjy� } ztd|� ��|�d}~ww g } |
�dg �D ]} | d dkr�t�d| d � q�| �| � q�| �r"g }
| D ].} | �d�}|du�r|
�|� d|� d| d � �}n d | d � d| d � �}t
�t|� q�td 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> �attachz--no-auto-enablerD zAttaching to Ubuntu Pro. %s� r � )�rcsrC z Failure attaching Ubuntu Pro:
)r> �enablez --assume-yesz--format�jsonT� )�capturer\ z!Error while enabling service(s): rE zPro response was not json: r9 � message_codezservice-already-enabledz Service `%s` already enabled.�servicezFailure enabling `z`: �messagezFailure of type `r! z(Failure enabling Ubuntu Pro service(s): )r rO r# r2 �listr! r" r rJ r8 r rN rP r �logexcr% r^ �loads�JSONDecodeError�getr4 )�tokenr] �
attach_cmd� redacted_cmdrV �err' �cmd�
enable_stdoutrH � enable_resprQ �err�error_servicesrb r( r( r) �
configure_pro� s�
�
�
�� ��������
���rr �cloudc C sl t �d�rdS z| j�� W n ty t�td� � w z
| j�dg� W dS ty5 t�td� � w )z.Install ubuntu-advantage-tools if not present.r> NzPackage update failedzubuntu-advantage-toolsz(Failed to install ubuntu-advantage-tools) r �which�distro�update_package_sources� Exceptionr re r# �install_packages)rs r( r( r) �maybe_install_ua_tools� s
� �ry �returnc
C s� t | �di ��dd��}|rdS ddlm} ddlm} z!tjddd �� |� }W d � W |jS 1 s5w Y W |jS |y[ } zt� d
|� t�
t � W Y d }~dS d }~ww ) Nr r Fr )�UserFacingError)�should_auto_attachz6Checking if the instance can be attached to Ubuntu Pro�always��log_modez%Error during `should_auto_attach`: %s) r& rh �uaclient.api.exceptionsr{ �4uaclient.api.u.pro.attach.auto.should_auto_attach.v1r| r �Timedr# rJ r2 � ERROR_MSG_SHOULD_AUTO_ATTACH)r r r{ r| �result�exr( r( r) �_should_auto_attach s. � � � � �
��r� c C sZ | � d�}|sd}t�|� t|��| � d�}|r"t�dd�|�� t|| � d�d� d S )Nri z8`ubuntu_pro.token` required in non-Pro Ubuntu instances.� enable_betaz<Ignoring `ubuntu_pro.enable_beta` services in Pro attach: %srE r] )ri r] )rh r# r$ r% rJ r8 rr )r ri r' r� r( r( r) �_attach( s
�r� c
C s� ddl m}m} ddlm}m} | �d�}| �d�}|||d�}z!tjddd �� ||d
� W d � W d S 1 s:w Y W d S |y[ |d usP|d urXt �
d � Y d S Y d S |yv } zd |j � �} t � | � t
| �|�d }~ww )
Nr )�AlreadyAttachedErrorr{ )�FullAutoAttachOptions�full_auto_attachr] r� )r] r� 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� r� r{ �2uaclient.api.u.pro.attach.auto.full_auto_attach.v1r� r� rh r r� r# r2 r' r$ r% )
r r� r{ r� r� r] r� r� r� r' r( r( r) � _auto_attach7 s0
� &�
��
��r� �name�cfg�argsc C s2 d }t t�|��}|r)t|�dkrtdd�|� ��t�dd�|�� ||d }d|v r;|r7t�d|d � |d }|d u rGt�d| � d S t |t
�s]d t |�j � �}t�
|� t|��d
|v rld }t�
|� t|��t|� t|�d �� t|� t�d
� t|�r�t|� d S |�� dhks�t|� d S d S )Nr_ zLUnable to configure Ubuntu Pro. Multiple deprecated config keys provided: %srE 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 �commandszPDeprecated configuration "ubuntu-advantage: commands" provided. Expected "token"r? zETo discover more log info, please check /var/log/ubuntu-advantage.logr )rd �DEPRECATED_KEYS� intersection�lenr% r8 r# r2 rJ r r r! r" r$ ry rX rh r* r� r� �keysr� )r� r� rs r� r �
deprecatedr' r( r( r) �handleY sd ��� ��
��
�
� �r� )N),�__doc__r^ �loggingrK �typingr r � urllib.parser � cloudinitr r r �cloudinit.cloudr �cloudinit.configr �cloudinit.config.schemar
�cloudinit.settingsr rB r� rd r �__annotations__� getLoggerr" r# r r� r1 r r* r= rX rr ry r& r� r� r� rO r� r( r( r( r) �<module> s> �
�
'
,d ""