File: //snap/core18/current/usr/lib/python3/dist-packages/cloudinit/__pycache__/helpers.cpython-36.pyc
3
�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 je�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 @ 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 )� DummyLockN)r r r 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 )N)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 r r �clear+ s zDummySemaphores.clearc C s d S )Nr )r r r r � clear_all. 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 )N)�fn)r r" r r r r 3 s zFileLock.__init__c C s dt j| �| jf S )Nz<%s using file %r>)r Zobj_namer" )r r r r �__str__6 s zFileLock.__str__N)r r r r r# r r r r r! 2 s r! c C s | j 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 )N)�sem_path)r r* r r r r ? s zFileSemaphores.__init__Fc c sF t |�}y| j||�V W n& tk
r@ |r:| j||� � Y nX d S )N)r( �_acquire� Exceptionr )r r' �freq�
clear_on_failr r r r B s zFileSemaphores.lockc C sN t |�}| j||�}ytj|� W n& ttfk
rH tjtd|� dS X dS )Nz Failed deleting semaphore %sFT)r( � _get_pathr
Zdel_file�IOError�OSError�logexc�LOG)r r' r- �sem_filer r r r L s zFileSemaphores.clearc C s> yt j| j� W n( ttfk
r8 t jtd| j� Y nX d S )Nz&Failed deleting semaphore directory %s)r
Zdel_dirr* r0 r1 r2 r3 )r r r r r V s
zFileSemaphores.clear_allc C sn | j ||�rd S | j||�}dtj� t� f }ytj||� W n& ttfk
rd tj t
d|� d S X t |�S )Nz%s: %s
z Failed writing semaphore file %s) r r/ �os�getpidr r
Z
write_filer0 r1 r2 r3 r! )r r' r- r4 �contentsr r r r+ ^ s zFileSemaphores._acquirec C sh | s|t krdS t|�}| j||�}tjj|�r6dS ||krdtjj| j||��rdtjd||� 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( r/ r5 �path�existsr3 �warning)r r' r- Zcnamer4 r r r r n s zFileSemaphores.has_runc C s< | j }| s|tkr"tjj||�S tjj|d||f �S d S )Nz%s.%s)r* r
r5 r8 �join)r r' r- r* r r r r/ � s zFileSemaphores._get_pathN)F) r r r r r r r r r r+ r r/ r r r r r) > s
r) c @ s& e Zd Zdd� Zdd� Zd dd�ZdS )
�Runnersc C s || _ i | _d S )N)�paths�sems)r r= r r r r � s zRunners.__init__c C sj |t ks| rd S d }|tkr,| jjd�}n|tkr@| jjd�}|sHd S || jkr`t|�| j|< | j| S )N�sem)r r
r= � get_ipathr � get_cpathr> r) )r r- r* r r r �_get_sem� s
zRunners._get_semNFc C s� | j |�}|st� }|s g }|j||�r:tjd||� dS |j|||��L}|s\td| ��n4tjd||� t|t�r�|f |�}n||� }d|fS W d Q R X d S )Nz%s already ran (freq=%s)Fz Failed to acquire lock for %szRunning %s using lock (%s)T)FN) rB r r r3 �debugr r �
isinstance�dict) r r' Zfunctor�argsr- r. r? ZlkZresultsr r r �run� s
z Runners.run)NF)r r r r rB rG r r r r r< � s r< c @ sH e Zd Zddd�Zdd� Zdd� Zd d
� Zed �d d
�Ze dd� �Z
dS )� ConfigMergerNTc C s( || _ || _|| _|| _|| _d | _d S )N)�_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 rVy&| j j� }|r.t|t�r.|j|� W n$ tk
rT tjtd| j � Y nX |S )Nz2Failed loading of datasource config object from %s) rJ Zget_config_objrD rE �appendr, r
r2 r3 )r Zd_cfgsZds_cfgr r r �_get_datasource_configs� s
z$ConfigMerger._get_datasource_configsc C sT g }t tjkrPtjt }y|jtj|�� W n" tk
rN tjtd|� Y nX |S )Nz%Failed loading of env. config from %s) r r5 �environrP r
� read_confr, r2 r3 )r Ze_cfgsZe_fnr r r �_get_env_configs� s
z ConfigMerger._get_env_configsc C s� g }| j s|S dg}| jr.|jd� |jd� x~|D ]v}| j j|�}|r4tjj|�r4y|jtj|�� W q4 t k
r� t
j d|� Y q4 t k
r� tj
t
d|� Y q4X q4W |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)rI rM rP �
get_ipath_curr5 r8 �isfiler
rS �PermissionErrorr3 rC r, r2 )r Zi_cfgsZcc_pathsZcc_pZcc_fnr r r �_get_instance_configs� s(
z"ConfigMerger._get_instance_configs)�returnc C s� g }| j rRxF| j D ]<}y|jtj|�� W q tk
rL tjtd|� Y qX qW |j| j� � |j| j � � |j| j
� � | j r�|j| j � tj |�S )Nz'Failed loading of configuration from %s)
rK rP r
rS r, r2 r3 �extendrT r[ rQ rL Z
mergemanydict)r �cfgsZc_fnr r r � _read_cfg s zConfigMerger._read_cfgc C s | j d kr| j� | _ | j S )N)rN r_ )r r r r �cfg s
zConfigMerger.cfg)NNNNT) r r r r rQ rT r[ rE r_ �propertyr` r r r r rH � s
rH c @ sF e Zd Zdd� Zdd� Zdd� Zdd� Zdd d �Zd
d� Zdd� Z dS )�ContentHandlersc C s i | _ g | _d S )N)�
registered� initialized)r r r r r ( s zContentHandlers.__init__c C s
| j |�S )N)�
is_registered)r �itemr r r � __contains__, s z ContentHandlers.__contains__c C s
| j |�S )N)� _get_handler)r �keyr r r � __getitem__/ s zContentHandlers.__getitem__c C s
|| j kS )N)rc )r � content_typer r r re 2 s z ContentHandlers.is_registeredFTc C sr t � }x4|j� D ](}|r$|j|� q| j|�s|j|� qW x|D ]}|| j|< qBW |rn|| jkrn| jj|� |S )N)�setZ
list_types�addre rc rd rP )r �modrd Z overwrite�types�tr r r �register5 s
zContentHandlers.registerc C s
| j | S )N)rc )r rk r r r rh C s z ContentHandlers._get_handlerc C s t | jj� �S )N)�listrc �items)r r r r rs F s zContentHandlers.itemsN)FT)
r r r r rg rj re rq rh rs r r r r rb ' s
rb 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� || _ |jdd�| _|jdd�| _tjj| jd�| _tjj| jd�| _tjj| jd�| _ |jdd �}tjj|d
�| _
d d d
ddddddddddddddd d d d d �| _ || _ d S )!N� cloud_dirz/var/lib/cloud�run_dirz/run/cloud-init�instancez
boot-finishedZseedZ
templates_dirz/etc/cloud/templates/z%s.tmpl� boothookszcloud-config.txt�data�handlerszinstance-data.jsonz instance-data-sensitive.jsonz .instance-idz manual-cleanzobj.pkl�scriptsr? 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)rz rU r{ r| �
instance_data�instance_data_sensitiveZ instance_idZmanual_clean_markerZobj_pklr} r? ZuserdataZ userdata_rawZ
vendordataZ vendordata2Zvendordata2_rawZvendordata_rawrV rW Zvendor_scriptsr~ )
r^ �getrw rx r5 r8 r; �
instance_linkZ
boot_finishedZseed_dirZ template_tpl�lookupsrO )r rv �dsZ template_dirr r r r M s@
zPaths.__init__)�ci_pkl_versionr\ c C sJ t | d�s t| j| jd�j| _d| jkr2d| jd< d| jkrFd| jd< dS )z(Perform deserialization fixes for Paths.rx )rv r� r zinstance-data.jsonr� z instance-data-sensitive.jsonN)�hasattrrt r^ rO rx r� )r r� r r r � _unpickle{ s
zPaths._unpicklec C s | j | j|�S )N)r/ r� )r r' r r r rX � s zPaths.get_ipath_curc C s | j | j|�S )N)r/ rw )r r' r r r rA � s zPaths.get_cpathc C sf | j s
d S | j j� }|d kr d S t|�jtjd�}tjj| jd|�}| j j
|�}|rbtjj||�}|S )Nr% Z instances) rO Zget_instance_id�strr& r5 �sepr8 r; rw r� r� )r r' ZiidZ
path_safe_iid�ipathZadd_onr r r �
_get_ipath� s
zPaths._get_ipathc C s$ | j |�}|s tjd� d S |S d S )Nz?No per instance data available, is there an datasource/iid set?)r� r3 r: )r r' r� r r r r@ � s
zPaths.get_ipath)�basec C s |d kr |S t jj|| j| �S )N)r5 r8 r; r� )r r� r' r r r r/ � s zPaths._get_pathc C s | j | j|�S )N)r/ rx )r r' r r r � get_runpath� s zPaths.get_runpath)N)N)N)N)N)N)N)r r r Z_ci_pkl_versionrE r �intr� rX rA r� r@ r� r/ r� r r r r rt J s .
rt 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 }ytj| ||�}W n& tk
r, Y n tk
r> Y nX |S )N)�DEF_BASEr r� r r )r �section�option�valuer r r r� � s zDefaultingConfigParser.getc C s6 | j |� r"|j� dkr"| j|� tj| |||� d S )N�default)Z has_section�lowerZ add_sectionr rl )r r� r� r� r r r rl � s
zDefaultingConfigParser.setc C s | j ||�rtj| ||� d S )N)�
has_optionr �
remove_option)r r� r� r r r r� � s z$DefaultingConfigParser.remove_optionc C s | j ||�s| jS tj| ||�S )N)r� � DEF_BOOLEANr �
getboolean)r r� r� r r r r� � s z!DefaultingConfigParser.getbooleanc C s | j ||�s| jS tj| ||�S )N)r� � DEF_FLOATr �getfloat)r r� r� r r r r� � s z DefaultingConfigParser.getfloatc C s | j ||�s| jS tj| ||�S )N)r� �DEF_INTr �getint)r r� r� r r r r� � s z DefaultingConfigParser.getintc C s< d}t � }| j|� |j� |j� }|r8dj||dg�}|S )N� �
)r �write�flush�getvaluer; )r �headerr7 Z outputstreamr r r � stringify� s
z DefaultingConfigParser.stringify)N)N)r r r r� r� r� r� r� rl r� r� r� r� r� r r r r r� � s
r� )$r r5 Z configparserr r r �ior r Z cloudinitr Zloggingr r r
Zcloudinit.settingsr r r
r Z getLoggerr r3 r, r r r r! r( r) r< rH rb ZCloudInitPickleMixinrt r� r r r r �<module> s&
R-j#|