CasperSecurity
o
쑛h�) �
@ s� U d Z ddlZddlZddlZddlZddlZddlZddlmZ ddl m
Z
m Z ddl m
Z
mZ ddlmZ ddlmZ ddlmZ dd lmZmZ dd
lmZ dd lmZ d eegd
gd�Z ee d< e� e �Z dZ!G dd� dej"�Z#G dd� de#�Z$G dd� de#�Z%de&dedede'ddf
d d �Z(de)fd d �Z*de)de)fd d!�Z+d"e#de)fd#d$�Z,de)d
e#fd%d&�Z-de)d
e#fd'd(�Z.dS ))z9ansible enables running on first boot either ansible-pull� N)�deepcopy)�List�Optional)� lifecycle�subp)�Cloud)�Config)�
MetaSchema)� ALL_DISTROS�Distro)� PER_INSTANCE)�get_cfg_by_path�
cc_ansible�ansible)�id� frequency�distros�activate_by_schema_keys�meta�ansible_configc @ s| e Zd Zdefdd�Zdeej fdd�Zde fdd�Z
d d
� Z d e fd d
�Z
dd� Zejdd� �Zejde fdd��ZdS )� AnsiblePull�distroc C s< dg| _ ddg| _|| _i | _d | _tj�dd�| jd< d S )Nz ansible-pullz --version�HOMEz/root)�cmd_pull� cmd_versionr �env�run_user�os�environ�get)�selfr � r! �=/usr/lib/python3/dist-packages/cloudinit/config/cc_ansible.py�__init__ s
zAnsiblePull.__init__�returnc C sH | � | j�\}}|�� �d�}t�d|�}|r"|�d�}tj� |�S d S )Nr z ([\d\.]+))
�do_asr �
splitlines�pop�re�search�groupr �Version�from_str)r �stdout�_�
first_line�matches�versionr! r! r" � get_version+ s
zAnsiblePull.get_versionc G s | � g | j�|��\}}|S �N)r% r )r �argsr- r. r! r! r" �pull4 s zAnsiblePull.pullc C s | � � std��d S )Nz!command: ansible is not installed)� is_installed�
ValueError�r r! r! r" �
check_deps8 s �zAnsiblePull.check_deps�commandc K s0 | j s | j|fi |��S | jj|| j fi |��S r3 )r r r r% �r r: �kwargsr! r! r" r% < s zAnsiblePull.do_asc K s t j |fd| ji|��S )N�
update_env)r r r; r! r! r" r A s zAnsiblePull.subpc C � d S r3 r! r8 r! r! r" r6 D � zAnsiblePull.is_installed�pkg_namec C r> r3 r! �r r@ r! r! r" �installH r? zAnsiblePull.installN)�__name__�
__module__� __qualname__r r# r r r+ r2 �strr5 r9 �listr% r �abc�abstractmethodr6 rB r! r! r! r" r s
r c sd e Zd Zdedee f� fdd� Zdd� Zdd� Zd efd
d �Z d e
fd
d�Z de fdd�Z
� ZS )�AnsiblePullPipr �userc s t � �|� || _| �� d S r3 )�superr# r � add_pip_install_site_to_path�r r rK �� __class__r! r" r# N s zAnsiblePullPip.__init__c C s` | j r.| �tjddg�\}}|� d�}| j�d�}|r'd�||g�| jd< d S || jd< d S d S )Nz-cz&import site; print(site.getuserbase())z/bin/�PATH�:)r r% �sys�
executabler r �join)r � user_baser. � ansible_path�old_pathr! r! r" rM S s ��
�z+AnsiblePullPip.add_pip_install_site_to_pathc C s6 zdd l }W d S ty | j�| jjg� Y d S w )Nr )�pip� ImportErrorr �install_packages�pip_package_name)r rY r! r! r" �bootstrap_pip_if_requiredd s
�z(AnsiblePullPip.bootstrap_pip_if_requiredr@ c C s� | � � | �� sFtjdddg}tj�tj�t� d�d��r"|�
d� | j r*|�
d� | � |� t
�d|� | �g |�|�� t
�d |� d
S d
S ) zishould cloud-init grow an interface for non-distro package
managers? this seems reusable
�-mrY rB �stdlibzEXTERNALLY-MANAGEDz--break-system-packages�--userzInstalling the %s packagezInstalled the %s packageN)r] r6 rS rT r �path�existsrU � sysconfig�get_path�appendr � _AnsiblePullPip__upgrade_pip�LOG�infor% )r r@ �cmdr! r! r" rB j s( �
��
�zAnsiblePullPip.installri c
C sh t �d� z
| �g |�d�d�� W n tjy, } z
t �d|� W Y d}~dS d}~ww t �d� dS )zOTry to upgrade pip then not raise an exception
if the pip upgrade failsz
Upgrading pipz --upgraderY zZFailed at upgrading pip. This is usually not criticalso the script will skip this step.
%sNz Upgraded pip)rg rh r% r �ProcessExecutionError�warning)r ri �er! r! r" �
__upgrade_pip� s
��� z AnsiblePullPip.__upgrade_pipr$ c C s4 t jdddg}| jr|�d� | �|�\}}d|v S )Nr^ rY rG r` r )rS rT r re r% )r ri r- r. r! r! r" r6 � s
zAnsiblePullPip.is_installed)rC rD rE r r rF r# rM r] rB rG rf �boolr6 �
__classcell__r! r! rO r" rJ M s rJ c sF e Zd Zdedee f� fdd� Zdefdd�Zdefd d
�Z � Z
S ) �AnsiblePullDistror rK c s t � �|� || _d S r3 )rL r# r rN rO r! r" r# � s
zAnsiblePullDistro.__init__r@ c C s | � � s
| j�|g� d S d S r3 )r6 r r[ rA r! r! r" rB � s �zAnsiblePullDistro.installr$ c C s t t�d��S )Nr )rn r �whichr8 r! r! r" r6 � s z AnsiblePullDistro.is_installed) rC rD rE r r rF r# rB rn r6 ro r! r! rO r" rp � s rp �name�cfg�cloudr4 r$ c C s� |� di �}|� d�}|� d�}|� d�}|� d�}|� d�} |� dd�}
|ryt|� |j} |d kr8t| |�} nt| |�} | �|
� | �� |� d
d�}
|
rS|
| jt< |rZt || � | rpt
| t �rd| g} | D ] }t | t
|�� qf|r{t|| � d S d S d S ) Nr r �install_method�setup_controller�galaxyr5 � package_name� rY r )r �validate_configr rJ rp rB r9 r � CFG_OVERRIDE�ansible_galaxy�
isinstance�dict�run_ansible_pullr �ansible_controller)rr rs rt r4 � ansible_cfg� ansible_userru rv �
galaxy_cfg�pull_cfgrx r r r r! r! r" �handle� s:
�r� c C s6 d}|D ]}t | |�std|� d| � ���q| �d�}|rst|t�r&|g}n t|t�s2td|� ���|D ]>}t|t�sBtd|� ���t |d�sNtd|� ���t |d �}t |d
�}t||g�setd |� ���t||g�rrtd |� ���q4| �d
�}|r�t|�d�|�d�g�s�td|� ���| d }|dvr�td��d S )N)ru rx zMissing required key 'z' from r5 zLInvalid value ansible.pull. Expected either dict of list of dicts but found z7Invalid value of ansible.pull. Expected dict but found �urlz Missing required key 'url' from �
playbook_name�playbook_namesz+Missing required key 'playbook_names' from zzKey 'ansible.pull.playbook_name' and 'ansible.pull.playbook_names' are mutually exclusive. Please use 'playbook_names' in rv � repositories� run_ansiblezMissing required key from ru )rY r z Invalid install method {install})r
r7 r r} r~ rG �any�all) rs �
required_keys�keyr� �p_cfg� has_playbook�
has_playbooks�controller_cfgrB r! r! r" rz � sb
�
��
��
� ���
���rz c C s dd� | � � D �S )zremove boolean false valuesc S s& i | ]\}}|d ur|� dd�|�qS )Fr. �-)�replace��.0r� �valuer! r! r" �
<dictcomp>
s
�z filter_args.<locals>.<dictcomp>)�items)rs r! r! r" � filter_args s �r� r5 c C s |� dd �}|� dd �}|r|g}| �� }|s t�d� n |t�ddd�k r<|�d�r<td|j� d |j � d |j
� d
���d d � t |�� � D �}|ri|t�dd
d�kri| j
g |�|�R � }|rgtj�|� � d S d S |D ]}| j
g |�|�R � }|r�tj�|� � qkd S )Nr� r� z 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>� )r' r2 rg rk r r+ r r7 �major�minor�patchr� r� r5 rS r- �write)r5 rs r� r� �v� pull_argsr- �playbookr! r! r" r s4
�
����r c C s4 | � dg �}|st�d| � |D ]}|�|� qd S )N�actionszInvalid config: %s)r rg rk r% )rs r r� r: r! r! r" r| 5 s �r| c C s� | � dg �D ]}|�dd|d |d g� q| � dg �D ]"}|�d�}|�d�}d |gd
d � t|��� D ��}|j||d � q d S )
Nr� �git�clone�sourcera r� � playbook_dirr� zansible-playbookc S s g | ] \}}d |� d|� ��qS )r� r� r! r� r! r! r" r� I s z&ansible_controller.<locals>.<listcomp>)�cwd)r r% r' r� r� )rs r �
repositoryr4 r� r� r: r! r! r" r� > s �
���r� )/�__doc__rH �loggingr r( rS rc �copyr �typingr r � cloudinitr r �cloudinit.cloudr �cloudinit.configr �cloudinit.config.schemar �cloudinit.distrosr
r �cloudinit.settingsr �cloudinit.utilr
r �__annotations__� getLoggerrC rg r{ �ABCr rJ rp rF rG r� r~ rz r� r r| r� r! r! r! r" �<module> s@ �
.T
(5 !