Current File : //snap/core20/2582/lib/python3/dist-packages/cloudinit/__pycache__/helpers.cpython-38.pyc
U
Ӈg�@ � @ s d dl Z d dlZd dlZd dlmZmZmZ d dlmZ d dl m Z d dl
m Z m Z m
Z
mZ d dlmZmZmZmZ e�e�ZG dd� de�ZG d d
� d
�ZG d d � d �ZG d
d� 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 dd � d e�Z#dS ) � N)�
NoOptionError�NoSectionError�RawConfigParser)�StringIO)�time)� persistence�settings�
type_utils�util)� CFG_ENV_NAME�
PER_ALWAYS� PER_INSTANCE�PER_ONCEc @ 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 @ s4 e Zd Zdd� Zejd dd��Zdd� Zdd � Zd
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.clearN)F) r r r r �
contextlib�contextmanagerr 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 0 s zFileLock.__init__c C s dt �| �| jf S )Nz<%s using file %r>)r Zobj_namer& r r r r �__str__3 s zFileLock.__str__N)r r r r r' r r r r r% / s r% c C s | � dd�S )N�-�_)�replace)�namer r r �canon_sem_name7 s r, c @ sD e Zd Zdd� Zejddd��Zdd� Zdd � Zd
d � Z d d
� Z
dS )�FileSemaphoresc C s
|| _ d S r )�sem_path)r r. r r r r < s zFileSemaphores.__init__Fc c sF t |�}z| �||�V W n& tk
r@ |r:| �||� � Y nX d S r )r, �_acquire� Exceptionr" )r r+ �freq�
clear_on_failr r r r ? s zFileSemaphores.lockc C sP t |�}| �||�}zt�|� W n( ttfk
rJ t�td|� Y dS X dS )Nz Failed deleting semaphore %sFT)r, � _get_pathr
Zdel_file�IOError�OSError�logexc�LOG)r r+ r1 �sem_filer r r r" I s zFileSemaphores.clearc C sp | � ||�rd S | �||�}dt�� t� f }zt�||� W n( ttfk
rf t� t
d|� Y d S X t |�S )Nz%s: %s
z Failed writing semaphore file %s) r! r3 �os�getpidr r
Z
write_filer4 r5 r6 r7 r% )r r+ r1 r8 �contentsr r r r/ S s zFileSemaphores._acquirec C s0 |r |t krdS t|�}| �||�}tj�|�S r )r r, r3 r9 �path�exists)r r+ r1 Zcnamer8 r r r r! c s
zFileSemaphores.has_runc C s: | j }|r|tkr tj�||�S tj�|d||f �S d S )Nz%s.%s)r. r
r9 r< �join)r r+ r1 r. r r r r3 m s zFileSemaphores._get_pathN)F) r r r r r# r$ r r" r/ r! r3 r r r r r- ; s
r- c @ s& e Zd Zdd� Zdd� Zd dd�ZdS )
�Runnersc C s || _ i | _d S r )�paths�sems)r r@ r r r r v s zRunners.__init__c C sh |t ks |sd S d }|tkr*| j�d�}n|tkr>| j�d�}|sFd S || jkr^t|�| j|< | j| S )N�sem)r r
r@ � get_ipathr � get_cpathrA r- )r r1 r. r r r �_get_semz s
zRunners._get_semNFc
C s� | � |�}|st� }|s g }|�||�r:t�d||� dS |�|||��X}|s\td| ��n@t�d||� t|t�r�|f |�}n||� }d|fW 5 Q R � S W 5 Q R X d S )Nz%s already ran (freq=%s))FNz Failed to acquire lock for %szRunning %s using lock (%s)T) rE r r! r7 �debugr r �
isinstance�dict) r r+ Zfunctor�argsr1 r2 rB ZlkZresultsr r r �run� s
z Runners.run)NF)r r r r rE rJ r r r r r? u s r? c @ sH e Zd Zddd�Zdd� Zdd� Zd d
� Zed �d d
�Ze dd� �Z
dS )� ConfigMergerNTc C s( || _ || _|| _|| _|| _d | _d S r )�_paths�_ds�_fns� _base_cfg�_include_vendor�_cfg)r r@ �
datasourceZadditional_fnsZbase_cfgZinclude_vendorr r r r � s zConfigMerger.__init__c C sZ g }| j rVz&| j �� }|r.t|t�r.|�|� W n$ tk
rT t�td| j � Y nX |S )Nz2Failed loading of datasource config object from %s) rM Zget_config_objrG rH �appendr0 r
r6 r7 )r Zd_cfgsZds_cfgr r r �_get_datasource_configs� s
�
z$ConfigMerger._get_datasource_configsc C sT g }t tjkrPtjt }z|�t�|�� W n" tk
rN t�td|� Y nX |S )Nz%Failed loading of env. config from %s) r r9 �environrS r
� read_confr0 r6 r7 )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� |D ]v}| j �|�}|r2tj�|�r2z|�t�|�� W q2 t k
r� t
� d|� Y q2 t k
r� t�
t
d|� Y q2X q2|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)rL rP rS �
get_ipath_curr9 r< �isfiler
rV �PermissionErrorr7 rF r0 r6 )r Zi_cfgsZcc_pathsZcc_pZcc_fnr r r �_get_instance_configs� s0
� � z"ConfigMerger._get_instance_configs)�returnc C s� g }| j rN| j D ]<}z|�t�|�� W q tk
rJ t�td|� Y qX q|�| �� � |�| � � � |�| �
� � | j r�|�| j � t� |�S )Nz'Failed loading of configuration from %s)
rN rS r
rV r0 r6 r7 �extendrW r^ rT rO Z
mergemanydict)r �cfgsZc_fnr r r � _read_cfg� s"
� zConfigMerger._read_cfgc C s | j d kr| �� | _ | j S r )rQ rb r r r r �cfg s
zConfigMerger.cfg)NNNNT) r r r r rT rW r^ rH rb �propertyrc r r r r rK � s �
rK 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 kS r �rf �r Z content_typer r r rh s z ContentHandlers.is_registeredFTc C sj t � }|�� D ](}|r"|�|� q| �|�s|�|� q|D ]}|| j|< q<|rf|| jkrf| j�|� |S r )�setZ
list_types�addrh rf rg rS )r �modrg Z overwrite�types�tr r r �register s
zContentHandlers.registerc C s
| j | S r rn ro r r r rk ( s z ContentHandlers._get_handlerc C s t | j�� �S r )�listrf �itemsr r r r rw + s zContentHandlers.itemsN)FT)
r r r r rj rm rh ru rk rw r r r r re s
re c @ sr e Zd ZdZded�dd�Zedd�dd�Zdd d
�Zdd d �Z dd
d�Z
ddd�Z de d�dd�Z
d dd�ZdS ) �Paths� N)� path_cfgsc C s� || _ |�dd�| _|�dd�| _|�dtj�| _tj� | jd�| _
tj� | j
d�| _ tj� | jd�| _ |�d d
�}tj� |d �| _
d d
ddddddddddddddd d d d d d!d"d#d$d%�| _|| _d S )&N� cloud_dirz/var/lib/cloud�docs_dirz/usr/share/doc/cloud-init/�run_dir�instancez
boot-finishedZseedZ
templates_dirz/etc/cloud/templates/z%s.tmpl� boothookszcloud-config.txt�data�handlers�instance-data.json� instance-data-sensitive.json�combined-cloud-config.jsonznetwork-config.jsonz .instance-idz manual-cleanzobj.pkl�scriptsrB zuser-data.txt.iz
user-data.txtzvendor-data.txt.izvendor-data2.txt.izvendor-data2.txtzvendor-data.txtzvendor2-cloud-config.txtzvendor-cloud-config.txtzscripts/vendor�warnings�hotplug.enabled�
.skip-network)r rX r� r� �
instance_data�instance_data_sensitive�combined_cloud_configZnetwork_configZ instance_idZmanual_clean_markerZobj_pklr� rB ZuserdataZ userdata_rawZ
vendordataZ vendordata2Zvendordata2_rawZvendordata_rawrY rZ Zvendor_scriptsr� r� r� )ra �getr{ r| r ZDEFAULT_RUN_DIRr} r9 r< r> �
instance_linkZ
boot_finishedZseed_dirZ template_tpl�lookupsrR )r rz �dsZ template_dirr r r r 2 sZ � � ��!zPaths.__init__)�ci_pkl_versionr_ c C sr t | d�s t| j| jd�j| _d| jkr2d| jd< d| jkrFd| jd< d| jkrZd| jd< d | jkrnd | jd < d
S ) z(Perform deserialization fixes for Paths.r} )rz r� r� r� r� r� r� r� r� N)�hasattrrx ra rR r} r� )r r� r r r � _unpickleg s
�
�
�
zPaths._unpicklec C s | � | j|�S r )r3 r� �r r+ r r r r[ � s zPaths.get_ipath_curc C s | � | j|�S r )r3 r{ r� r r r rD � s zPaths.get_cpathc C sf | j s
d S | j �� }|d kr d S t|��tjd�}tj�| jd|�}| j �
|�}|rbtj�||�}|S )Nr) Z instances) rR Zget_instance_id�strr* r9 �sepr< r> r{ r� r� )r r+ ZiidZ
path_safe_iid�ipathZadd_onr r r �
_get_ipath� 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� r7 Zwarning)r r+ r� r r r rC � s
�zPaths.get_ipath)�basec C s |d kr |S t j�|| j| �S r )r9 r< r> r� )r r� r+ 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 Z_ci_pkl_versionrH r �intr� r[ rD r� rC r� r3 r� r r r r rx / s 5
rx c @ sX e Zd ZdZdZdZdZdd� Zddd�Zd d
� Z d d � Z
d
d� Z dd� Z ddd�Z
dS )�DefaultingConfigParserr g FNc C sD | j }zt�| ||�}W n& tk
r, Y n tk
r> Y nX |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)Z has_section�lowerZ add_sectionr rp r� r r r rp � 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�getvaluer> )r �headerr; Z outputstreamr r r � stringify� s
z DefaultingConfigParser.stringify)N)N)r r r r� r� r� r� r� rp r� r� r� r� r� r r r r r� � s
r� )$r# Zloggingr9 Z configparserr r r �ior r Z cloudinitr r r r
Zcloudinit.settingsr r r
r Z getLoggerr r7 r0 r r r r% r, r- r? rK re ZCloudInitPickleMixinrx r� r r r r �<module> s(
:-j#
Mini Shell