/
lib
/
python3.9
/
site-packages
/
cloudinit
/
config
/
__pycache__
/
Upload File
HOME
a '�Dg�* � @ 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.9/site-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 sb d}t D ]<}z"| j�||� d�� |}W qFW q tjyB Y q0 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 }|durd|g}|dur(|d|g7 }|r6|dg7 }t j|dd�j}tj| �� dd��B}tj�|d �}t j ||d d � t j |g| d d �W d � S 1 s�0 Y 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 sR d|vrt �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 u �r�tD ]h}tt ��H |�r.||ggn|g}|j � |� |}W d � �qzW d � n1 �sl0 Y �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 |v �rrt�|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�r6| � d(t � � �} | � d)|�!� �} | �"� } |�#|| | � �q�t�$|jd*|j � t� |j|�%� � �q�d |v �r�t� |j&t'j(|d d d+�� | �r�t)|d,� | �r>t �d-� |d.g} d/|v �r&|d/ } t*| t+t,f��r�| �-| � n6t*| t.��r | �-| �/� � nt �d0t0| �� | �-t1� n | �-t1� t2j2| d d1� | �rNt)|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_attributesZconfc S s g | ] }|� � �qS r )�lstrip)�.0�ir r r! � <listcomp>� � 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 ������ � �� 6�� � � �� � � � � � 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 � �� $