Your IP : 216.73.217.13
U
�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(e)eeee*dd �d d �Z+e,d!�d"d#�Z-e,e,d$�d%d&�Z.e&e,d'�d(d)�Z/e,e&d*�d+d,�Z0e,e&d*�d-d.�Z1dS )/z9ansible 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_pathZ
cc_ansibleZAnsiblez 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�titleZ frequencyZdistrosZactivate_by_schema_keysZ descriptionZexamples�meta�ansible_configc @ sz e Zd Zed�dd�Zee d�dd�Zed�dd�Z d d
� Z
e d �d d
�Z dd� Z
ejdd� �Zejed�dd��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|�}|rB|�d�}t�|�S d S )Nr z ([\d\.]+)) �do_asr �
splitlines�pop�re�search�groupr Zfrom_str)r" �stdout�_Z
first_lineZmatches�versionr# r# r$ � get_versionN s
zAnsiblePull.get_versionc G s | � | j|��\}}|S �N)r( r )r" �argsr. r/ 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 s( | j s| j|f|�S | jj|| j f|�S r2 )r r r r( �r" r9 �kwargsr# r# r$ r( _ s zAnsiblePull.do_asc K s t |fd| ji|��S )Nr )r r r: r# r# r$ r d s zAnsiblePull.subpc C s d S r2 r# r7 r# r# r$ r5 g s zAnsiblePull.is_installed��pkg_namec C s d S r2 r# �r" r= r# r# r$ �installk s zAnsiblePull.installN)�__name__�
__module__� __qualname__r r% r r r1 �strr4 r8 �listr( r �abc�abstractmethodr5 r? r# r# r# r$ r B s
r c sD e Zd Zeee d�� fdd� Zed�dd�Zed�dd �Z � Z
S )
�AnsiblePullPip)r �userc sf t � �|� || _| �tjddg�\}}|� d�}| j�d�}|rXd�||g�| jd< n
|| jd< d S )Nz-cz('import site; print(site.getuserbase())'z/bin/�PATH�:) �superr% r r( �sys�
executabler r! �join)r" r rH � user_baser/ Z ansible_pathZold_path�� __class__r# r$ r% q s
�
zAnsiblePullPip.__init__r< c C sz | � � svz ddl}W n$ tk
r8 | j�| jj� Y nX tjdddg}| jrX|� d� | �
|d�� | �
||f�� dS ) zishould cloud-init grow an interface for non-distro package
managers? this seems reusable
r N�-m�pipr? z--user� --upgrade)rT rS ) r5 rS � ImportErrorr �install_packagesZpip_package_namerL rM r �appendr( )r" r= rS �cmdr# r# r$ r? � s
zAnsiblePullPip.installr& c C s | � tjdddg�\}}d|kS )NrR rS rD r )r( rL rM )r" r. r/ r# r# r$ r5 � s zAnsiblePullPip.is_installed) r@ rA rB r r rC r% r? �boolr5 �
__classcell__r# r# rP r$ rG p s rG c @ s( e Zd Zed�dd�Zed�dd�ZdS )�AnsiblePullDistror< c C s | � � s| j�|� d S r2 )r5 r rV r> r# r# r$ r? � s zAnsiblePullDistro.installr& c C s t td��S )Nr )rY r r7 r# r# r$ r5 � s z AnsiblePullDistro.is_installedN)r@ rA rB rC r? rY r5 r# r# r# r$ r[ � s r[ )r �cfg�cloud�logr3 r' c C s� |� di �}|� d�}|� d�}|� d�}|� d�} |� d�}
|� dd�} |r�t|� |j} |d krpt| |�}
nt| �}
|
�| � |
�� |� d
d�}|r�||
jt< | r�t | |
� |
r�t
|
t |
�� |r�t ||
� d S ) Nr r �install_method�setup_controllerZgalaxyr4 � package_name� rS r )
r! �validate_configr rG r[ r? r8 r � CFG_OVERRIDE�ansible_galaxy�run_ansible_pullr �ansible_controller)r r\ r] r^ r3 Z ansible_cfgZ ansible_userr_ r` Z
galaxy_cfgZpull_cfgra r r r r# r# r$ �handle� s0
rh �r\ c C s� d}|D ]"}t | |�std|� d| � ���q| �d�r^dD ]"}t | |�s:td|� d| � ���q:| �d�}|r�t|�d�|�d�g�s�td |� ���| d
}|d kr�td ��d S )
N)r_ ra zMissing required key 'z' from r4 )zpull/urlzpull/playbook_namer` � repositories� run_ansiblezMissing required key from r_ )rS r z Invalid install method {install})r r6 r! �any)r\ Z
required_keys�keyZcontroller_cfgr? r# r# r$ rc � s&
��rc )r\ r' c C s dd� | � � D �S )zremove boolean false valuesc S s& i | ] \}}|d k r|� dd�|�qS )Fr/ �-)�replace��.0rm �valuer# r# r$ �
<dictcomp>� s �
z filter_args.<locals>.<dictcomp>)�itemsri r# r# r$ � filter_args� s �ru )r4 r\ c C s� |� d�}| �� }|s"t�d� n<|tddd�k r^|�d�r^td|j� d|j� d|j � d ���| j
d
d � t |�� � D �|f�� }|r�t
j�|� � d S ) N�
playbook_namez Cannot parse ansible version� � r ZdiffzAnsible version �.z%doesn't support --diff flag, exiting.c S s2 g | ]*\}}|d k r$d|� d|� �nd|� ��qS )T�--�=r# rp r# r# r$ �
<listcomp>� s �z$run_ansible_pull.<locals>.<listcomp>)r* r1 �LOG�warningr r! r6 �major�minorZpatchr4 ru rt rL r. �write)r4 r\ rv �vr. r# r# r$ rf � s"
�
��rf )r\ r c C s4 | � dg �}|s t�d| � |D ]}|�|� q d S )N�actionszInvalid config: %s)r! r} r~ r( )r\ r r� r9 r# r# r$ re s
re c C s� | � dg �D ] }|�dd|d |d g� q | � dg �D ]D}|�d�}|�d�}d |fd
d � t|��� D ��}|j||d � q8d S )
Nrj ZgitZclone�source�pathrk � playbook_dirrv zansible-playbookc S s g | ]\}}d |� d|� ��qS )rz r{ r# rp r# r# r$ r| s z&ansible_controller.<locals>.<listcomp>)�cwd)r! r( r* ru rt )r\ r Z
repositoryr3 r� rv r9 r# r# r$ rg s �
��rg )2�__doc__rE r r+ rL �copyr Zloggingr r �textwrapr �typingr Zcloudinit.cloudr Zcloudinit.configr Zcloudinit.config.schemar r
Zcloudinit.distrosr r Zcloudinit.settingsr
Zcloudinit.subpr r Zcloudinit.utilr r r �__annotations__r@ r} rd �ABCr rG r[ rC rD rh �dictrc ru rf re rg r# r# r# r$ �<module> sd �
� ���*.%
� '