File: //lib/python3/dist-packages/cloudinit/config/__pycache__/cc_ansible.cpython-310.pyc
o
x[h� �
@ s� U d Z ddlZddlZddlZddlZddlZddlZddlmZ ddl m
Z
ddl m Z 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)�Optional)� lifecycle�signal_handler�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 sD t �� � tj|fd| ji|��W d � S 1 sw Y d S )N�
update_env)r �
suspend_crashr r r; r! r! r" r A s
$�zAnsiblePull.subpc C � d S r3 r! r8 r! r! r" r6 E � zAnsiblePull.is_installed�pkg_namec C r? r3 r! �r rA r! r! r" �installI r@ zAnsiblePull.installN)�__name__�
__module__� __qualname__r r# r r r+ r2 �strr5 r9 �listr% r �abc�abstractmethodr6 rC r! r! r! r" r s
r c sV e Zd Zdedee f� fdd� Zdd� Zdd� Zd efd
d �Z d e
fd
d�Z � Z S )�AnsiblePullPipr �userc s t � �|� || _| �� d S r3 )�superr# r � add_pip_install_site_to_path�r r rL �� __class__r! r" r# O 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" rN T 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 rZ r! r! r" �bootstrap_pip_if_requirede s
�z(AnsiblePullPip.bootstrap_pip_if_requiredrA c C s� | � � | �� s@tjdddg}tj�tj�t� d�d��r"|�
d� | j r*|�
d� | � g |�d�d�� | � g |�|�� d S d S )
zishould cloud-init grow an interface for non-distro package
managers? this seems reusable
�-mrZ rC �stdlibzEXTERNALLY-MANAGEDz--break-system-packages�--userz --upgradeN)
r^ r6 rT rU r �path�existsrV � sysconfig�get_path�appendr r% )r rA �cmdr! r! r" rC k s$ �
��
�zAnsiblePullPip.installr$ c C s4 t jdddg}| jr|�d� | �|�\}}d|v S )Nr_ rZ rH ra r )rT rU r rf r% )r rg r- r. r! r! r" r6 � s
zAnsiblePullPip.is_installed)
rD rE rF r r rG r# rN r^ rC �boolr6 �
__classcell__r! r! rP r" rK N s rK c sF e Zd Zdedee f� fdd� Zdefdd�Zdefd d
�Z � Z
S ) �AnsiblePullDistror rL c s t � �|� || _d S r3 )rM r# r rO rP r! r" r# � s
zAnsiblePullDistro.__init__rA c C s | � � s
| j�|g� d S d S r3 )r6 r r\ rB r! r! r" rC � s �zAnsiblePullDistro.installr$ c C s t t�d��S )Nr )rh r �whichr8 r! r! r" r6 � s z AnsiblePullDistro.is_installed) rD rE rF r r rG r# rC rh r6 ri r! r! rP r" rj � s rj �name�cfg�cloudr4 r$ c C s� |� di �}|� d�}|� d�}|� d�}|� d�}|� d�} |� dd�}
|rlt|� |j} |d kr8t| |�} nt| |�} | �|
� | �� |� d
d�}
|
rS|
| jt< |rZt || � | rct
| t | �� |rnt || � d S d S d S ) Nr r �install_method�setup_controller�galaxyr5 � package_name� rZ r )
r �validate_configr rK rj rC r9 r � CFG_OVERRIDE�ansible_galaxy�run_ansible_pullr �ansible_controller)rl rm rn r4 � ansible_cfg� ansible_userro rp �
galaxy_cfg�pull_cfgrr r r r r! r! r" �handle� s4
�r} 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)ro rr zMissing required key 'z' from r5 )zpull/urlzpull/playbook_namerp � repositories� run_ansiblezMissing required key from ro )rZ r z Invalid install method {install})r
r7 r �any)rm �
required_keys�key�controller_cfgrC r! r! r" rt � s,
�
�
���rt 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)rm r! r! r" � filter_args� s �r� r5 c C s� |� d�}| �� }|st�d� n |t�ddd�k r0|�d�r0td|j� d|j � d|j
� d ���| j g d
d � t |��
� D ��|�R � }|rNtj�|� � 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>)r' r2 �LOG�warningr r+ r r7 �major�minor�patchr5 r� r� rT r- �write)r5 rm r� �vr- r! r! r" rw � s&
�
����rw c C s4 | � dg �}|st�d| � |D ]}|�|� qd S )N�actionszInvalid config: %s)r r� r� r% )rm r r� r: r! r! r" rv � s �rv 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�sourcerb r � 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% r' r� r� )rm r �
repositoryr4 r� r� r: r! r! r" rx s �
���rx )/�__doc__rI �loggingr r( rT rd �copyr �typingr � cloudinitr r r �cloudinit.cloudr �cloudinit.configr �cloudinit.config.schemar �cloudinit.distrosr
r �cloudinit.settingsr �cloudinit.utilr
r �__annotations__� getLoggerrD r� ru �ABCr rK rj rG rH r} �dictrt r� rw rv rx r! r! r! r" �<module> s@ �
/A
%