File: //usr/lib/python3/dist-packages/cloudinit/__pycache__/helpers.cpython-39.pyc
a
��`�: � @ s. d dl m Z d dlZd dlZd dlmZmZmZ d dlmZ d dl m
Z
m Z m Z m
Z
d dlmZ d dlmZ d dlmZ e�e�ZG d d
� d
e�ZG d d � d e�ZG d
d� de�ZG dd� de�Zdd� Z G dd� de�Z G dd� de�Z G dd� de�Z G dd� de�Z G dd � d e�Z!G d d � d e�Z"dS ) � )�timeN)�NoSectionError�
NoOptionError�RawConfigParser)�StringIO)� PER_INSTANCE�
PER_ALWAYS�PER_ONCE� CFG_ENV_NAME)�log)�
type_utils)�utilc @ s e Zd ZdS )� LockFailureN��__name__�
__module__� __qualname__� r r �3/usr/lib/python3/dist-packages/cloudinit/helpers.pyr s r c @ s e Zd ZdS )� DummyLockNr r r r r r s r c @ s< e Zd Zdd� Zejd
dd��Zdd� Zdd � Zd
d � Z d S )�DummySemaphoresc C s d S �Nr ��selfr r r �__init__% s zDummySemaphores.__init__Fc c s t � V d S r )r )r �_name�_freqZ_clear_on_failr r r �lock( s zDummySemaphores.lockc C s dS )NFr �r r r r r r �has_run, s zDummySemaphores.has_runc C s dS )NTr r r r r �clear/ s zDummySemaphores.clearc C s d S r r r r r r � clear_all2 s 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 s
|| _ d S r )�fn)r r% r r r r 7 s zFileLock.__init__c C s dt �| �| jf S )Nz<%s using file %r>)r �obj_namer% r r r r �__str__: s zFileLock.__str__N)r r r r r' r r r r r$ 6 s r$ c C s | � dd�S )N�-�_)�replace)�namer r r �canon_sem_name> s r, c @ sL e Zd Zdd� Zejddd��Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dS )�FileSemaphoresc C s
|| _ d S r )�sem_path)r r. r r r r C s zFileSemaphores.__init__Fc c sD t |�}z| �||�V W n$ ty> |r8| �||� � Y n0 d S r )r, �_acquire� Exceptionr )r r+ �freq�
clear_on_failr r r r F s zFileSemaphores.lockc C sN t |�}| �||�}zt�|� W n& ttfyH t�td|� Y dS 0 dS )Nz Failed deleting semaphore %sFT)r, � _get_pathr
�del_file�IOError�OSError�logexc�LOG)r r+ r1 �sem_filer r r r P s zFileSemaphores.clearc C s< zt �| j� W n& ttfy6 t �td| j� Y n0 d S )Nz&Failed deleting semaphore directory %s)r
�del_dirr. r5 r6 r7 r8 r r r r r! Z s �zFileSemaphores.clear_allc C sn | � ||�rd S | �||�}dt�� t� f }zt�||� W n& ttfyd t� t
d|� Y d S 0 t |�S )Nz%s: %s
z Failed writing semaphore file %s) r r3 �os�getpidr r
�
write_filer5 r6 r7 r8 r$ )r r+ r1 r9 �contentsr r r r/ a s zFileSemaphores._acquirec C sf |r |t krdS t|�}| �||�}tj�|�r4dS ||krbtj�| �||��rbt�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 r, r3 r; �path�existsr8 �warning)r r+ r1 �cnamer9 r r r r q s �zFileSemaphores.has_runc C s: | j }|r|tkr tj�||�S tj�|d||f �S d S )Nz%s.%s)r. r r; r? �join)r r+ r1 r. r r r r3 � s zFileSemaphores._get_pathN)F) r r r r r" r# r r r! r/ r r3 r r r r r- B s
r- c @ s& e Zd Zdd� Zdd� Zd dd�ZdS )
�Runnersc C s || _ i | _d S r )�paths�sems)r rE r r r r � s zRunners.__init__c C sh |t ks |sd S d }|tkr*| j�d�}n|tkr>| j�d�}|sFd S || jvr^t|�| j|< | j| S )N�sem)r r rE � get_ipathr � get_cpathrF r- )r r1 r. r r r �_get_sem� s
zRunners._get_semNFc C s� | � |�}|st� }|s g }|�||�r:t�d||� dS |�|||��h}|s\td| ��nFt�d||� t|t�r�|f i |��}n||� }d|fW d � S W d � n1 s�0 Y d S )Nz%s already ran (freq=%s))FNz Failed to acquire lock for %szRunning %s using lock (%s)T) rJ r r r8 �debugr r �
isinstance�dict) r r+ �functor�argsr1 r2 rG Zlk�resultsr r r �run� s
z Runners.run)NF)r r r r rJ rQ r r r r rD � s rD c @ sB e Zd Zddd�Zdd� Zdd� Zd d
� Zd d � Zed
d� �Z dS )� ConfigMergerNTc C s( || _ || _|| _|| _|| _d | _d S r )�_paths�_ds�_fns� _base_cfg�_include_vendor�_cfg)r rE �
datasource�additional_fns�base_cfg�include_vendorr r r r � s zConfigMerger.__init__c C sX g }| j rTz&| j �� }|r.t|t�r.|�|� W n" tyR t�td| j � Y n0 |S )Nz2Failed loading of datasource config object from %s) rT �get_config_objrL rM �appendr0 r
r7 r8 )r Zd_cfgs�ds_cfgr r r �_get_datasource_configs� s
�
z$ConfigMerger._get_datasource_configsc C sR g }t tjv rNtjt }z|�t�|�� W n tyL t�td|� Y n0 |S )Nz%Failed loading of env. config from %s) r
r; �environr^ r
� read_confr0 r7 r8 )r Ze_cfgsZe_fnr r r �_get_env_configs� s
�
z ConfigMerger._get_env_configsc C s� g }| j s|S dg}| jr$|�d� |D ]r}| j �|�}|r(tj�|�r(z|�t�|�� W q( t yz t
� d|� Y q( t y� t�
t
d|� Y q(0 q(|S )N� cloud_config�vendor_cloud_configz5Skipped loading cloud-config from %s due to non-root.z&Failed loading of cloud-config from %s)rS rW r^ �
get_ipath_curr; r? �isfiler
rb �PermissionErrorr8 rK r0 r7 )r Zi_cfgsZcc_pathsZcc_pZcc_fnr r r �_get_instance_configs� s*
� � z"ConfigMerger._get_instance_configsc C s� g }| j rL| j D ]:}z|�t�|�� W q tyH t�td|� Y q0 q|�| �� � |�| � � � |�| �
� � | j r�|�| j � t� |�S )Nz'Failed loading of configuration from %s)
rU r^ r
rb r0 r7 r8 �extendrc ri r` rV �
mergemanydict)r �cfgsZc_fnr r r � _read_cfg� s
� zConfigMerger._read_cfgc C s | j d u r| �� | _ | j S r )rX rm r r r r �cfg s
zConfigMerger.cfg)NNNNT)
r r r r r` rc ri rm �propertyrn r r r r rR � s �
rR 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 s zContentHandlers.__init__c C s
| � |�S r )�
is_registered)r �itemr r r � __contains__ s z ContentHandlers.__contains__c C s
| � |�S r )� _get_handler)r �keyr r r � __getitem__! s zContentHandlers.__getitem__c C s
|| j v S r �rq �r � content_typer r r rs $ s z ContentHandlers.is_registeredFTc C sj t � }|�� D ](}|r"|�|� q| �|�s|�|� q|D ]}|| j|< q<|rf|| jvrf| j�|� |S r )�set�
list_types�addrs rq rr r^ )r �modrr � overwrite�types�tr r r �register' s
zContentHandlers.registerc C s
| j | S r ry rz r r r rv 5 s z ContentHandlers._get_handlerc C s t | j�� �S r )�listrq �itemsr r r r r� 8 s zContentHandlers.itemsN)FT)
r r r r ru rx rs r� rv r� r r r r rp s
rp c @ sR e Zd Zddd�Zddd�Zddd�Zddd �Zdd
d �Zdd d
�Zddd�Z dS )�PathsNc C s� || _ |�dd�| _|�dd�| _tj�| jd�| _tj�| jd�| _|�d�| _ tj�| jd�| _
|�d d
�}tj�|d �| _ d d
ddddddddddddddd �| _ || _
d S ) N� cloud_dirz/var/lib/cloud�run_dirz/run/cloud-init�instancez
boot-finished� upstart_dir�seed�
templates_dirz/etc/cloud/templates/z%s.tmpl�handlers�scriptszscripts/vendorrG � boothooksz
user-data.txtzuser-data.txt.izobj.pklzcloud-config.txtzvendor-cloud-config.txt�datazvendor-data.txtzvendor-data.txt.iz .instance-idz manual-clean�warnings)r� r� �vendor_scriptsrG r� � userdata_raw�userdata�obj_pklrd re r� �vendordata_raw�
vendordata� instance_id�manual_clean_markerr� )rl �getr� r� r; r? rC �
instance_link�
boot_finished�upstart_conf_dZseed_dir� template_tpl�lookupsrY )r Z path_cfgs�dsZ template_dirr r r r = s6 �zPaths.__init__c C s | � | j|�S r )r3 r� �r r+ r r r rf _ s zPaths.get_ipath_curc C s | � | j|�S r )r3 r� r� r r r rI d s zPaths.get_cpathc C sf | j s
d S | j �� }|d u r d S t|��tjd�}tj�| jd|�}| j �
|�}|rbtj�||�}|S )Nr) � instances) rY �get_instance_id�strr* r; �sepr? rC r� r� r� )r r+ �iidZ
path_safe_iid�ipath�add_onr r r �
_get_ipathi s
zPaths._get_ipathc C s$ | � |�}|s t�d� d S |S d S )Nz?No per instance data available, is there an datasource/iid set?)r� r8 rA )r r+ r� r r r rH y s
zPaths.get_ipathc C s |d u r |S t j�|| j| �S r )r; r? rC r� )r �baser+ r r r r3 � s zPaths._get_pathc C s | � | j|�S r )r3 r� r� r r r � get_runpath� s zPaths.get_runpath)N)N)N)N)N)N)N)
r r r r rf rI r� rH r3 r� r r r r r� < 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 s@ | j }zt�| ||�}W n" ty* Y n ty: Y n0 |S 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 r )�
has_optionr �
remove_option�r r� r� r r r r� � s z$DefaultingConfigParser.remove_optionc C s | � ||�s| jS t�| ||�S r )r� � DEF_BOOLEANr �
getbooleanr� r r r r� � s z!DefaultingConfigParser.getbooleanc C s | � ||�s| jS t�| ||�S r )r� � DEF_FLOATr �getfloatr� r r r r� � s z DefaultingConfigParser.getfloatc C s | � ||�s| jS t�| ||�S r )r� �DEF_INTr �getintr� r r r r� � s z DefaultingConfigParser.getintc C s< d}t � }| �|� |�� |�� }|r8d�||dg�}|S )N� �
)r �write�flush�getvaluerC )r �headerr> Z outputstreamr r r � stringify� s
z DefaultingConfigParser.stringify)N)N)r r r r� r� r� r� r� r| r� r� r� r� r� r r r r r� � s
r� )#r r" r; � configparserr r r �ior �cloudinit.settingsr r r r
� cloudinitr �loggingr r
� getLoggerr r8 r0 r �objectr r r$ r, r- rD rR rp r� r� r r r r �<module> s(
N-[$X