3
\�me�" �
@ s� U d Z ddlZddlZddlZddlZddlZddlZddlmZ ddl m
Z
ddl m Z ddl
mZ ddlmZ ddlmZmZ dd lmZmZ dd
lmZ dd lmZmZ dd l m Z m Z d
ddeegdge
d�e
d�e
d�gd�Z e ee �Z ej e!�Z"dZ#G dd� dej$�Z%G dd� de%�Z&G dd� de%�Z'e(eee)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*�Z/e+e%d(�d+d,�Z0dS )-z9ansible enables running on first boot either ansible-pull� N)�deepcopy)�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:
package_name: ansible-core
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_keys� descriptionZexamples�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 | _| jjdd�| jd< d S )Nz ansible-pullz --version�HOMEz/root)�cmd_pull� cmd_versionr �os�environ�env�run_user�get)�selfr � r! � /usr/lib/python3.6/cc_ansible.py�__init__E s
zAnsiblePull.__init__)�returnc C sF | j | j�\}}|j� jd�}tjd|�}|rB|jd�}tj|�S d S )Nr z ([\d\.]+)) �do_asr �
splitlines�pop�re�search�groupr Zfrom_str)r �stdout�_Z
first_lineZmatches�versionr! r! r" � get_versionP s
zAnsiblePull.get_versionc G s | j | j|��\}}|S )N)r% r )r �argsr+ r, r! r! r" �pullY s zAnsiblePull.pullc C s | j � 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 )N)r r r r% )r r4 �kwargsr! r! r" r% a s zAnsiblePull.do_asc K s t |fd| ji|��S )Nr )r r )r r4 r5 r! r! r" r f s zAnsiblePull.subpc C s d S )Nr! )r r! r! r" r1 i s zAnsiblePull.is_installed)�pkg_namec C s d S )Nr! )r r6 r! r! r" �installm s zAnsiblePull.installN)�__name__�
__module__� __qualname__r
r# r r r. �strr0 r3 �listr% r �abc�abstractmethodr1 r7 r! r! r! r" r D s r c sD e Zd Zeee d�� fdd� Zed�dd�Zed�dd �Z � Z
S )
�AnsiblePullPip)r �userc sf t � j|� || _| jtjddg�\}}|� d�}| jjd�}|rXdj||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 r@ � user_baser, Z ansible_pathZold_path)� __class__r! r" r# s s
zAnsiblePullPip.__init__)r6 c C s� | j � s�y ddl}W n& tk
r: | jj| jjg� Y nX tjdddg}tj j
tj j t j
d�d��rp|jd� | jr�|jd � | j|d �� | j||f�� dS ) zishould cloud-init grow an interface for non-distro package
managers? this seems reusable
r Nz-m�pipr7 �stdlibzEXTERNALLY-MANAGEDz--break-system-packagesz--user� --upgrade)rK rI )r1 rI � ImportErrorr �install_packagesZpip_package_namerD rE r �path�existsrF � sysconfig�get_path�appendr r% )r r6 rI �cmdr! r! r" r7 � s"
zAnsiblePullPip.install)r$ c C s | j tjdddg�\}}d|kS )Nz-mrI r< r )r% rD rE )r r+ r, r! r! r" r1 � s zAnsiblePullPip.is_installed) r8 r9 r: r
r r; r# r7 �boolr1 �
__classcell__r! r! )rH r" r? r s r? c @ s( e Zd Zed�dd�Zed�dd�ZdS )�AnsiblePullDistro)r6 c C s | j � s| jj|g� d S )N)r1 r rM )r r6 r! r! r" r7 � s zAnsiblePullDistro.install)r$ c C s t td��S )Nr )rT r
)r r! r! r" r1 � s z AnsiblePullDistro.is_installedN)r8 r9 r: r; r7 rT r1 r! r! r! r" rV � s rV )r �cfg�cloudr/ r$ c C s� |j di �}|j d�}|j d�}|j d�}|j d�}|j d�} |j dd�}
|r�t|� |j} |d krpt| |�} nt| �} | j|
� | j� |j d
d�}
|
r�|
| jt< |r�t || � | r�t
| t | �� |r�t || � d S ) Nr r �install_method�setup_controllerZgalaxyr0 � package_name� rI r )
r �validate_configr r? rV r7 r3 r � CFG_OVERRIDE�ansible_galaxy�run_ansible_pullr �ansible_controller)r rW rX r/ Z ansible_cfgZ ansible_userrY rZ Z
galaxy_cfgZpull_cfgr[ r r r r! r! r" �handle� s0
rb )rW c C s� d}x*|D ]"}t | |�s
td|� d| � ���q
W | jd�rfx*dD ]"}t | |�s@td|� d| � ���q@W | jd�}|r�t|jd �|jd
�g�s�td |� ���| d }|dkr�td��d S )NrY r[ zMissing required key 'z' from r0 �pull/url�pull/playbook_namerZ � repositories� run_ansiblezMissing required key from rI r z Invalid install method {install})rY r[ )rc rd )rI r )r r2 r �any)rW Z
required_keys�keyZcontroller_cfgr7 r! r! r" r] � s$
r] )rW r$ c C s dd� | j � D �S )zremove boolean false valuesc S s& i | ] \}}|d k r||j dd��qS )Fr, �-)�replace)�.0rh �valuer! r! r" �
<dictcomp>� s z filter_args.<locals>.<dictcomp>)�items)rW r! r! r" � filter_args� s ro )r0 rW c C s� |j d�}| j� }|s"tjd� n<|tddd�k r^|jd�r^td|j� d|j� d|j � d ���| j
d
d � t |�j � D �|f�� }|r�t
jj|� � 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 )Tz--�=r! )rk rh rl r! r! r" �
<listcomp> s z$run_ansible_pull.<locals>.<listcomp>)r' r. �LOG�warningr r r2 �major�minorZpatchr0 ro rn rD r+ �write)r0 rW rp �vr+ r! r! r" r` � s
r` )rW r c C s8 | j dg �}|s tjd| � x|D ]}|j|� q"W d S )N�actionszInvalid config: %s)r rv rw r% )rW r r| r4 r! r! r" r_
s
r_ c C s� x.| j dg �D ] }|jdd|d |d g� qW xT| j dg �D ]D}|jd�}|jd�}d |fd
d � t|�j� D ��}|j||d � q>W d S )
Nre ZgitZclone�sourcerN rf � playbook_dirrp zansible-playbookc S s g | ]\}}d |� d|� ��qS )z--rt r! )rk rh rl r! r! r" ru ! s z&ansible_controller.<locals>.<listcomp>)�cwd)r r% r' ro rn )rW r Z
repositoryr/ r~ rp r4 r! r! r" ra s
ra )1�__doc__r= Zloggingr r( rD rP �copyr �textwrapr Ztypingr Zcloudinit.cloudr Zcloudinit.configr Zcloudinit.config.schemar r Zcloudinit.distrosr r
Zcloudinit.settingsr Zcloudinit.subpr r
Zcloudinit.utilr r �metaZ getLoggerr8 rv r^ �ABCr r? rV r; r< rb �dictr] ro r` r_ ra r! r! r! r" �<module> sP
.0 %