← Back
Editing: 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 "
Save File
Cancel