File: //lib/python3/dist-packages/cloudinit/__pycache__/helpers.cpython-310.pyc
o
�Ad�A � @ s d dl Z d dlZd dlmZmZmZ d dlmZ d dlmZ d dl m
Z d dl m Z m
Z
mZ d dlmZmZmZmZ e �e�ZG dd � d e�ZG d
d � d �ZG d d
� d
�ZG dd� d�Zdd� Z G dd� d�Z G dd� d�Z G dd� d�Z G dd� d�Z G dd� de j!�Z"G d d � d e�Z#dS ) � N)�
NoOptionError�NoSectionError�RawConfigParser)�StringIO)�time)�log)� persistence�
type_utils�util)� CFG_ENV_NAME�
PER_ALWAYS� PER_INSTANCE�PER_ONCEc @ � e Zd ZdS )� LockFailureN��__name__�
__module__� __qualname__� r r �3/usr/lib/python3/dist-packages/cloudinit/helpers.pyr � r c @ r )� DummyLockNr r r r r r r r c @ s< e Zd Zdd� Zejd
dd��Zdd� Zdd � Zd
d � Z d S )�DummySemaphoresc C � d S �Nr ��selfr r r �__init__! � zDummySemaphores.__init__Fc c s � t � V d S r )r )r �_name�_freq�_clear_on_failr r r �lock$ s � zDummySemaphores.lockc C � dS �NFr �r r r! r r r �has_run( r zDummySemaphores.has_runc C r$ )NTr r& r r r �clear+ r zDummySemaphores.clearc C r r r r r r r � clear_all. r zDummySemaphores.clear_allN�F)
r r r r �
contextlib�contextmanagerr# r' r( r) r r r r r s r c @ s e Zd Zdd� Zdd� ZdS )�FileLockc C �
|| _ d S r )�fn)r r/ r r r r 3 �
zFileLock.__init__c C s dt �| �| jf S )Nz<%s using file %r>)r �obj_namer/ r r r r �__str__6 s zFileLock.__str__N)r r r r r2 r r r r r- 2 s r- c C s | � dd�S )N�-�_)�replace)�namer r r �canon_sem_name: s r7 c @ sL e Zd Zdd� Zejddd��Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dS )�FileSemaphoresc C r. r )�sem_path)r r9 r r r r ? r0 zFileSemaphores.__init__Fc c s@ � t |�}z
| �||�V W d S ty |r | �||� � w r )r7 �_acquire� Exceptionr( )r r6 �freq�
clear_on_failr r r r# B s � �zFileSemaphores.lockc C sL t |�}| �||�}zt�|� W dS ttfy% t�td|� Y dS w )Nz Failed deleting semaphore %sFT)r7 � _get_pathr
�del_file�IOError�OSError�logexc�LOG)r r6 r< �sem_filer r r r( L s ��zFileSemaphores.clearc C s< z t �| j� W d S ttfy t �td| j� Y d S w )Nz&Failed deleting semaphore directory %s)r
�del_dirr9 r@ rA rB rC r r r r r) V s
��zFileSemaphores.clear_allc C sl | � ||�rd S | �||�}dt�� t� f }z t�||� W t |�S ttfy5 t� t
d|� Y d S w )Nz%s: %s
z Failed writing semaphore file %s) r' r> �os�getpidr r
�
write_filer@ rA rB rC r- )r r6 r< rD �contentsr r r r: ^ s ��zFileSemaphores._acquirec C sf |r|t krdS t|�}| �||�}tj�|�rdS ||kr1tj�| �||��r1t�d||� dS dS )NFTz�%s has run without canonicalized name [%s].
likely the migrator has not yet run. It will run next boot.
run manually with: cloud-init single --name=migrator)r r7 r> rF �path�existsrC �warning)r r6 r< �cnamerD r r r r' n s �zFileSemaphores.has_runc C s6 | j }|r |tkrtj�||�S tj�|d||f �S )Nz%s.%s)r9 r
rF rJ �join)r r6 r< r9 r r r r> � s zFileSemaphores._get_pathNr* ) r r r r r+ r, r# r( r) r: r' r> r r r r r8 > s
r8 c @ s& e Zd Zdd� Zdd� Zd dd�ZdS )
�Runnersc C s || _ i | _d S r )�paths�sems)r rP r r r r � �
zRunners.__init__c C sh |t ks|sd S d }|tkr| j�d�}n
|tkr | j�d�}|s#d S || jvr/t|�| j|< | j| S )N�sem)r r
rP � get_ipathr � get_cpathrQ r8 )r r< r9 r r r �_get_sem� s
zRunners._get_semNFc C s� | � |�}|s
t� }|sg }|�||�r t�d||� dS |�|||��,}|s-td| ��t�d||� t|t�rA|di |��}n||� }d|fW d � S 1 sSw Y d S )Nz%s already ran (freq=%s))FNz Failed to acquire lock for %szRunning %s using lock (%s)Tr ) rV r r' rC �debugr# r �
isinstance�dict) r r6 �functor�argsr< r= rS �lk�resultsr r r �run� s"
$�z Runners.runr% )r r r r rV r^ r r r r rO � s rO c @ sR e Zd Z ddd�Zdd� Zdd� Zd d
� Zd efd d
�Ze dd� �Z
dS )� ConfigMergerNTc C s( || _ || _|| _|| _|| _d | _d S r )�_paths�_ds�_fns� _base_cfg�_include_vendor�_cfg)r rP �
datasource�additional_fns�base_cfg�include_vendorr r r r � s
zConfigMerger.__init__c C sh g }| j r2z| j �� }|rt|t�r |�|� W |S W |S W |S ty1 t�td| j � Y |S w |S )Nz2Failed loading of datasource config object from %s) ra �get_config_objrX rY �appendr; r
rB rC )r �d_cfgs�ds_cfgr r r �_get_datasource_configs� s(
�� ���z$ConfigMerger._get_datasource_configsc C sV g }t tjv r)tjt }z |�t�|�� W |S ty( t�td|� Y |S w |S )Nz%Failed loading of env. config from %s) r rF �environrk r
� read_confr; rB rC )r �e_cfgs�e_fnr r r �_get_env_configs� s
��z ConfigMerger._get_env_configsc C s� g }| j s|S dg}| jr|�d� |�d� |D ]9}| j �|�}|rRtj�|�rRz
|�t�|�� W q t yB t
� d|� Y q t yQ t�
t
d|� Y qw q|S )N� cloud_config�vendor2_cloud_config�vendor_cloud_configz5Skipped loading cloud-config from %s due to non-root.z&Failed loading of cloud-config from %s)r` rd rk �
get_ipath_currF rJ �isfiler
rp �PermissionErrorrC rW r; rB )r �i_cfgs�cc_paths�cc_p�cc_fnr r r �_get_instance_configs� s0
� ���z"ConfigMerger._get_instance_configs�returnc C s� g }| j r%| j D ] }z
|�t�|�� W q ty$ t�td|� Y qw |�| �� � |�| � � � |�| �
� � | j rC|�| j � t� |�S )Nz'Failed loading of configuration from %s)
rb rk r
rp r; rB rC �extendrs r~ rn rc �
mergemanydict)r �cfgs�c_fnr r r � _read_cfg s
��
zConfigMerger._read_cfgc C s | j d u r
| �� | _ | j S r )re r� r r r r �cfg s
zConfigMerger.cfg)NNNNT) r r r r rn rs r~ rY r� �propertyr� r r r r r_ � s
�
r_ c @ sF e Zd Zdd� Zdd� Zdd� Zdd� Zdd d �Zd
d� Zdd� Z dS )�ContentHandlersc C s i | _ g | _d S r )�
registered� initializedr r r r r ( rR zContentHandlers.__init__c C �
| � |�S r )�
is_registered)r �itemr r r � __contains__, r0 z ContentHandlers.__contains__c C r� r )� _get_handler)r �keyr r r � __getitem__/ r0 zContentHandlers.__getitem__c C s
|| j v S r �r� �r � content_typer r r r� 2 r0 z ContentHandlers.is_registeredFTc C sj t � }|�� D ]}|r|�|� q| �|�s|�|� q|D ]}|| j|< q |r3|| jvr3| j�|� |S r )�set�
list_types�addr� r� r� rk )r �modr� � overwrite�types�tr r r �register5 s
� zContentHandlers.registerc C s
| j | S r r� r� r r r r� C r0 z ContentHandlers._get_handlerc C s t | j�� �S r )�listr� �itemsr r r r r� F � zContentHandlers.itemsN)FT)
r r r r r� r� r� r� r� r� r r r r r� ' s
r� c @ st e Zd ZdZddefdd�Zdeddfdd �Zdd
d �Zdd d
�Z ddd�Z
ddd�Z dde fdd�Z
ddd�ZdS )�Paths� N� path_cfgsc C s� || _ |�dd�| _|�dd�| _tj�| jd�| _tj�| jd�| _tj�| jd�| _ |�dd �}tj�|d
�| _
i d d �d d
�dd�dd�dd�dd�dd�dd�dd�dd�dd�d d �d d �d d!�d"d#�d$d%�d&d'�d(d)d*d+d,��| _ || _ d S )-N� cloud_dirz/var/lib/cloud�run_dirz/run/cloud-init�instancez
boot-finished�seed�
templates_dirz/etc/cloud/templates/z%s.tmpl� boothooksrt zcloud-config.txt�data�handlers�
instance_data�instance-data.json�instance_data_sensitive� instance-data-sensitive.json� instance_idz .instance-id�manual_clean_markerz manual-clean�obj_pklzobj.pkl�scriptsrS �userdatazuser-data.txt.i� userdata_rawz
user-data.txt�
vendordatazvendor-data.txt.i� vendordata2zvendor-data2.txt.i�vendordata2_rawzvendor-data2.txt�vendordata_rawzvendor-data.txtzvendor2-cloud-config.txtzvendor-cloud-config.txtzscripts/vendor�warnings)ru rv �vendor_scriptsr� )
r� �getr� r� rF rJ rN �
instance_link�
boot_finished�seed_dir� template_tpl�lookupsrf )r r� �ds� template_dirr r r r M sj ������� � �
����������
�
zPaths.__init__�ci_pkl_versionr c C sN t | d�st| j| jd�j| _d| jvrd| jd< d| jvr%d| jd< dS dS )z(Perform deserialization fixes for Paths.r� )r� r� r� r� r� r� N)�hasattrr� r� rf r� r� )r r� r r r � _unpickle{ s
��
���zPaths._unpicklec C � | � | j|�S r )r> r� �r r6 r r r rw � r� zPaths.get_ipath_curc C r� r )r> r� r� r r r rU � r� zPaths.get_cpathc C sf | j sd S | j �� }|d u rd S t|��tjd�}tj�| jd|�}| j �
|�}|r1tj�||�}|S )Nr4 � instances) rf �get_instance_id�strr5 rF �seprJ rN r� r� r� )r r6 �iid�
path_safe_iid�ipath�add_onr r r �
_get_ipath� s
zPaths._get_ipathc C s | � |�}|st�d� d S |S )Nz?No per instance data available, is there an datasource/iid set?)r� rC rL )r r6 r� r r r rT � s
�zPaths.get_ipath�basec C s |d u r|S t j�|| j| �S r )rF rJ rN r� )r r� r6 r r r r> � s zPaths._get_pathc C r� r )r> r� r� r r r � get_runpath� r� zPaths.get_runpathr )r r r �_ci_pkl_versionrY r �intr� rw rU r� rT r� r> r� r r r r r� J s .
r� c @ sX e Zd ZdZdZdZdZdd� Zddd�Zd d
� Z d d � Z
d
d� Z dd� Z ddd�Z
dS )�DefaultingConfigParserr g FNc C sB | j }z
t�| ||�}W |S ty Y |S ty Y |S w r )�DEF_BASEr r� r r �r �section�option�valuer r r r� � s � ��zDefaultingConfigParser.getc C s4 | � |�s|�� dkr| �|� t�| |||� d S )N�default)� has_section�lower� add_sectionr r� r� r r r r� � s
zDefaultingConfigParser.setc C s" | � ||�rt�| ||� d S d S r )�
has_optionr �
remove_option�r r� r� r r r r� � s �z$DefaultingConfigParser.remove_optionc C � | � ||�s | jS t�| ||�S r )r� � DEF_BOOLEANr �
getbooleanr� r r r r� � � z!DefaultingConfigParser.getbooleanc C r� r )r� � DEF_FLOATr �getfloatr� r r r r� � r� z DefaultingConfigParser.getfloatc C r� r )r� �DEF_INTr �getintr� r r r r� � r� z DefaultingConfigParser.getintc C s< d}t � }| �|� |�� |�� }|r d�||dg�}|S )N� �
)r �write�flush�getvaluerN )r �headerrI � outputstreamr r r � stringify� s
z DefaultingConfigParser.stringifyr )r r r r� r� r� r� r� r� r� r� r� r� r� r r r r r� � s
r� )$r+ rF � configparserr r r �ior r � cloudinitr �loggingr r r
�cloudinit.settingsr r r
r � getLoggerr rC r; r r r r- r7 r8 rO r_ r� �CloudInitPickleMixinr� r� r r r r �<module> s(
R-j#|