File: //usr/lib/python3/dist-packages/cloudinit/config/__pycache__/cc_ansible.cpython-310.pyc
o
�Adx! � @ s� U d Z ddlZddlZddlZddlZddlmZ ddlmZm Z ddl
m Z ddl m
Z
ddlmZ ddlmZ dd lmZmZ dd
lmZmZ dd lmZ dd lmZm Z dd
l m Z m Z dddeegdge d�e d�e d�gd�Z ee!d< ee �Z e e"�Z#dZ$G dd� dej%�Z&G dd� de&�Z'G d d � d e&�Z(d e)d ed ed!ed"e*d#df d$d%�Z+d e,fd&d'�Z-d e,d#e,fd(d)�Z.d*e&d e,fd+d,�Z/d e,de&fd-d.�Z0d e,de&fd/d0�Z1dS )1z9ansible enables running on first boot either ansible-pull� N)�deepcopy)�Logger� getLogger)�dedent)�Optional)�Cloud)�Config)�
MetaSchema� get_meta_doc)� ALL_DISTROS�Distro)� PER_INSTANCE)�subp�which)�Version�get_cfg_by_path�
cc_ansible�Ansiblez Configure ansible for instance�ansiblea This module provides ``ansible`` integration for
augmenting cloud-init's configuration of the local
node.
This module installs ansible during boot and
then uses ``ansible-pull`` to run the playbook
repository at the remote URL.
z� ansible:
install_method: distro
pull:
url: "https://github.com/holmanb/vmboot.git"
playbook_name: ubuntu.yml
z� ansible:
package_name: ansible-core
install_method: pip
pull:
url: "https://github.com/holmanb/vmboot.git"
playbook_name: ubuntu.yml
)�id�name�title� frequency�distros�activate_by_schema_keys� description�examples�meta�ansible_configc @ sz e Zd Zdefdd�Zdee fdd�Zdefdd�Z d d
� Z
d e fd d
�Z dd� Z
ejdd� �Zejdefdd��ZdS )� AnsiblePull�distroc C s> dg| _ ddg| _|| _tj| _d | _| j�dd�| jd< d S )Nz ansible-pullz --version�HOMEz/root)�cmd_pull� cmd_versionr �os�environ�env�run_user�get)�selfr � r* �=/usr/lib/python3/dist-packages/cloudinit/config/cc_ansible.py�__init__C s
zAnsiblePull.__init__�returnc C sF | � | j�\}}|�� �d�}t�d|�}|r!|�d�}t�|�S d S )Nr z ([\d\.]+)) �do_asr# �
splitlines�pop�re�search�groupr �from_str)r) �stdout�_�
first_line�matches�versionr* r* r+ � get_versionN s
zAnsiblePull.get_versionc G s | � g | j�|��\}}|S �N)r. r" )r) �argsr5 r6 r* r* r+ �pullW s zAnsiblePull.pullc C s | � � std��d S )Nz!command: ansible is not installed)� is_installed�
ValueError�r) r* r* r+ �
check_deps[ s �zAnsiblePull.check_deps�commandc K s0 | j s | j|fi |��S | jj|| j fi |��S r; )r' r r r. �r) rB �kwargsr* r* r+ r. _ s zAnsiblePull.do_asc K s t |fd| ji|��S )Nr& )r r& rC r* r* r+ r d s zAnsiblePull.subpc C � d S r; r* r@ r* r* r+ r> g � zAnsiblePull.is_installed�pkg_namec C rE r; r* �r) rG r* r* r+ �installk rF zAnsiblePull.installN)�__name__�
__module__� __qualname__r r, r r r: �strr= rA �listr. r �abc�abstractmethodr> rI r* r* r* r+ r B s
r c sF e Zd Zdedee f� fdd� Zdefdd�Zdefd d
�Z � Z
S ) �AnsiblePullPipr �userc sh t � �|� || _| �tjddg�\}}|� d�}| j�d�}|r-d�||g�| jd< d S || jd< d S )Nz-cz('import site; print(site.getuserbase())'z/bin/�PATH�:) �superr, r' r. �sys�
executabler& r( �join)r) r rR � user_baser6 � ansible_path�old_path�� __class__r* r+ r, q s
�
zAnsiblePullPip.__init__rG c C s� | � � sAzddl}W n ty | j�| jj� Y nw tjdddg}| jr+|� d� | �
g |�d�d�� | �
g |�|�� dS dS )zishould cloud-init grow an interface for non-distro package
managers? this seems reusable
r N�-m�piprI z--userz --upgrade) r> r_ � ImportErrorr �install_packages�pip_package_namerV rW r' �appendr. )r) rG r_ �cmdr* r* r+ rI � s �
�zAnsiblePullPip.installr- c C s | � tjdddg�\}}d|v S )Nr^ r_ rN r )r. rV rW )r) r5 r6 r* r* r+ r> � s zAnsiblePullPip.is_installed) rJ rK rL r r rM r, rI �boolr> �
__classcell__r* r* r\ r+ rQ p s rQ c @ s( e Zd Zdefdd�Zdefdd�ZdS )�AnsiblePullDistrorG c C s | � � s | j�|� d S d S r; )r> r ra rH r* r* r+ rI � s �zAnsiblePullDistro.installr- c C s t td��S )Nr )re r r@ r* r* r+ r> � s z AnsiblePullDistro.is_installedN)rJ rK rL rM rI re r> r* r* r* r+ rg � s rg r �cfg�cloud�logr< r- c C s� |� di �}|� d�}|� d�}|� d�}|� d�} |� d�}
|� dd�} |rkt|� |j} |d kr8t| |�}
nt| �}
|
�| � |
�� |� d
d�}|rR||
jt< | rYt | |
� |
rbt
|
t |
�� |rmt ||
� d S d S d S ) Nr r' �install_method�setup_controller�galaxyr= � package_name� r_ r )
r( �validate_configr rQ rg rI rA r& � CFG_OVERRIDE�ansible_galaxy�run_ansible_pullr �ansible_controller)r rh ri rj r< � ansible_cfg� ansible_userrk rl �
galaxy_cfg�pull_cfgrn r r r r* r* r+ �handle� s4
�ry c C s� d}|D ]}t | |�std|� d| � ���q| �d�r/dD ]}t | |�s.td|� d| � ���q | �d�}|rIt|�d�|�d�g�sItd |� ���| d
}|d vrUtd ��d S )
N)rk rn zMissing required key 'z' from r= )zpull/urlzpull/playbook_namerl � repositories� run_ansiblezMissing required key from rk )r_ r z Invalid install method {install})r r? r( �any)rh �
required_keys�key�controller_cfgrI r* r* r+ rp � s,
�
�
���rp c C s dd� | � � D �S )zremove boolean false valuesc S s& i | ]\}}|d ur|� dd�|�qS )Fr6 �-)�replace��.0r~ �valuer* r* r+ �
<dictcomp>� s � �z filter_args.<locals>.<dictcomp>)�items)rh r* r* r+ � filter_args� s �r� r= c C s� |� d�}| �� }|st�d� n |tddd�k r/|�d�r/td|j� d|j� d|j � d ���| j
g d
d � t |�� � D ��|�R � }|rMt
j�|� � d S d S ) N�
playbook_namez Cannot parse ansible version� � r �diffzAnsible version �.z%doesn't support --diff flag, exiting.c S s2 g | ]\}}|d urd|� d|� �nd|� ��qS )T�--�=r* r� r* r* r+ �
<listcomp>� s ��z$run_ansible_pull.<locals>.<listcomp>)r0 r: �LOG�warningr r( r? �major�minor�patchr= r� r� rV r5 �write)r= rh r� �vr5 r* r* r+ rs � s&
�
����rs c C s4 | � dg �}|st�d| � |D ]}|�|� qd S )N�actionszInvalid config: %s)r( r� r� r. )rh r r� rB r* r* r+ rr s �rr c C s� | � dg �D ]}|�dd|d |d g� q| � dg �D ]"}|�d�}|�d�}d |gd
d � t|��� D ��}|j||d � q d S )
Nrz �git�clone�source�pathr{ � playbook_dirr� zansible-playbookc S s g | ] \}}d |� d|� ��qS )r� r� r* r� r* r* r+ r� s z&ansible_controller.<locals>.<listcomp>)�cwd)r( r. r0 r� r� )rh r �
repositoryr< r� r� rB r* r* r+ rt s �
���rt )2�__doc__rO r$ r1 rV �copyr �loggingr r �textwrapr �typingr �cloudinit.cloudr �cloudinit.configr �cloudinit.config.schemar r
�cloudinit.distrosr r �cloudinit.settingsr
�cloudinit.subpr r �cloudinit.utilr r r �__annotations__rJ r� rq �ABCr rQ rg rM rN ry �dictrp r� rs rr rt r* r* r* r+ �<module> sr �
� ���*.% �����
�'