Your IP : 216.73.216.172
U
Ӈg�* � @ s8 U d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z m
Z
ddl Z ddl m
Z
mZmZmZmZ ddlmZ ddlmZ dd lmZ dd
lmZmZmZ dd l m Z d Z d
gZ dZ dege dgd�Z!ee"d< e�#e$�Z%G dd� d�Z&ee'd�dd�Z(dd� Z)e dddfed�d d �Z*e'eee+dd �d d �Z,dS )!z+Puppet: Install, configure and start puppet� N)�suppress)�StringIO)�List�Union)�helpers�subp�
temp_utils�
url_helper�util)�Cloud)�Config)�
MetaSchema)� ALL_DISTROS�Distro�PackageInstallerError)� PER_INSTANCEzKhttps://raw.githubusercontent.com/puppetlabs/install-puppet/main/install.shz--test)� puppet-agent�puppetZ cc_puppetr )�idZdistrosZ frequencyZactivate_by_schema_keys�metac @ s e Zd Zdd� ZdS )�PuppetConstantsc C s8 || _ || _tj�|d�| _tj�| jd�| _|| _d S )NZcertszca.pem)� conf_path�ssl_dir�os�path�join� ssl_cert_dir�
ssl_cert_path�csr_attributes_path)�selfZpuppet_conf_fileZpuppet_ssl_dirr � r �</usr/lib/python3/dist-packages/cloudinit/config/cc_puppet.py�__init__* s
zPuppetConstants.__init__N)�__name__�
__module__� __qualname__r" r r r r! r ) s r )�cloud�actionc C sd d}t D ]>}z"| j�||� d�� |}W qHW q tjk
rD Y qX q|s`t�d|d�t �� dS )z8Attempts to perform action on one of the puppet services� z.servicez0Could not '%s' any of the following services: %s�, N)�PUPPET_PACKAGE_NAMES�distroZmanage_servicer ZProcessExecutionError�LOG�warningr )r& r' Zservice_managed� puppet_namer r r! �_manage_puppet_services7 s
�r/ c C s t � | dd|g�\}}|�� S )z�Get the config value for a given setting using `puppet config print`
:param puppet_bin: path to puppet binary
:param setting: setting to query
�config�print)r �rstrip)�
puppet_binZsetting�out�_r r r! �get_config_valueI s r6 T)r+ c
C s� g }|dk rd|g}|dk r(|d|g7 }|r6|dg7 }t j|dd�j}tj| �� dd��@}tj�|d �}t j
||d
d � t j |g| d d
�W 5 Q R � S Q R X dS )a� Install puppet-agent from the puppetlabs repositories using the one-shot
shell script
:param distro: Instance of Distro
:param url: URL from where to download the install script
:param version: version to install, blank defaults to latest
:param collection: collection to install, blank defaults to latest
:param cleanup: whether to purge the puppetlabs repo after installation
Nz-vz-cz --cleanup� )�urlZretriesT)�dirZ needs_exezpuppet-installi� )�modeF�Zcapture) r Zreadurl�contentsr ZtempdirZget_tmp_exec_pathr r r r
�
write_filer ) r+ r8 �version�
collection�cleanup�argsZcontentZtmpdZtmpfr r r! �install_puppet_aioR s
�rB )�name�cfgr& rA �returnc
C s8 d|krt �d| � d S |d }t�|dd�}t�|dd �}t�|dd �}t�|dd�}t�|d d�} tj|d
d d �}
tj|d
dd �} tj|dtd �} |dkr�d}
d}d}n d}
d}d }t�|d|�}|s�|r�t �d� n�|�r�t �d|r�|nd|� |dk�r�|d k�rztD ]P}tt ��< |�r,||ggn|g}|j
� |� |}W 5 Q R � �q`W 5 Q R X �q|�s�t �dd� t�� n"|�r�||ggn|g}|j
� |� n.|dk�r�t
|j
| ||| � nt �d|� d }
t�|dt|d��}t�|d t|d ��}t�|d t|d ��}t|||�}d |k�rXt�|j�}t�� }d!d"� |�� D �}d#� |�}|jt|�|jd$� |d �� D ]�\}}|d%k�r�t�|jd&� t�|j|
d� t�|j� t�|j|
d� t� |j |� t�|j |
d� nT|�� D ]J\} } | d'k�r | � d(t � � �} | � d)|�!� �} | �"� } |�#|| | � �q�t�$|jd*|j � t� |j|�%� � �qnd |k�r~t� |j&t'j(|d d d+�� | �r�t)|d,� |
�r$t �d-� |d.g} d/|k�r |d/ } t*| t+t,f��r�| �-| � n6t*| t.��r�| �-| �/� � nt �d0t0| �� | �-t1� n
| �-t1� t2j2| d d1� | �r4t)|d2� d S )3Nr z9Skipping module named %s, no 'puppet' configuration found�installTr> r? � install_typeZpackagesr@ �execF)�defaultZ
start_service�aio_install_urlZaio�rootz/opt/puppetlabs/bin/puppetr � package_namez@Puppet install set to false but version supplied, doing nothing.z'Attempting to install puppet %s from %sZlatestz+No installable puppet package in any of: %sr) z Unknown puppet install type '%s'� conf_filer0 r Zssldirr Zcsr_attributes�confc S s g | ] }|� � �qS r )�lstrip)�.0�ir r r! �
<listcomp>� s zhandle.<locals>.<listcomp>�
)�sourceZca_certi� Zcertnamez%fz%iz%s.old)Zdefault_flow_style�enablezRunning puppet-agentZagentZ exec_argszOUnknown type %s provided for puppet 'exec_args' expected list, tuple, or stringr; �start)3r, �debugr
Zget_cfg_option_boolZget_cfg_option_str�AIO_INSTALL_URLr- r* r r r+ Zinstall_packagesr rB r6 r Zload_text_filer r ZDefaultingConfigParser�
splitlinesZ read_filer �itemsZ
ensure_dirr Z chownbynamer r= r �replace�socketZgetfqdnZget_instance_id�lower�set�renameZ stringifyr �yaml�dumpr/ �
isinstance�list�tuple�extend�str�split�type�PUPPET_AGENT_DEFAULT_ARGSr ) rC rD r& rA Z
puppet_cfgrF r> r? rG r@ �runZ
start_puppetdrJ Z puppet_userr3 Zpuppet_packagerL r. Z
to_installrM r r Z p_constantsr< Z
puppet_configZ
cleaned_linesZcleaned_contentsZcfg_name�o�v�cmdZcmd_argsr r r! �handlev s. � � � � ��
�
�� ��
� � ��
�
�
�
�
�
rn )-�__doc__Zloggingr r\ �
contextlibr �ior �typingr r r` Z cloudinitr r r r r
Zcloudinit.cloudr Zcloudinit.configr Zcloudinit.config.schemar
Zcloudinit.distrosr r r Zcloudinit.settingsr rX ri r* r �__annotations__Z getLoggerr# r, r rf r/ r6 rB rc rn r r r r! �<module> sB �
�� $