Google
Edit File: helpers.cpython-39.pyc
a '�Dg�@ � @ 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 �5/usr/lib/python3.9/site-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 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 ? s zFileSemaphores.lockc C sN t |�}| �||�}zt�|� W n& ttfyH t�td|� Y dS 0 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 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 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 || jvr^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 |�|||��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) rE r r! r7 �debugr r � isinstance�dict) r r+ Zfunctor�argsr1 r2 rB Zlk�resultsr r r �run� s z Runners.run)NF)r r r r rE rK 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 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) rN Zget_config_objrG rH �appendr0 r r6 r7 )r Zd_cfgsZds_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 r9 �environrT 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 ]r}| j �|�}|r2tj�|�r2z|�t�|�� W q2 t y� t � d|� Y q2 t y� t� t d|� Y q20 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)rM rQ rT � get_ipath_curr9 r< �isfiler rW �PermissionErrorr7 rF r0 r6 )r Zi_cfgsZcc_pathsZcc_pZcc_fnr r r �_get_instance_configs� s, � � z"ConfigMerger._get_instance_configs)�returnc 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) rO rT r rW r0 r6 r7 �extendrX r_ rU rP Z mergemanydict)r �cfgsZc_fnr r r � _read_cfg� s � zConfigMerger._read_cfgc C s | j d u r| �� | _ | j S r )rR rc r r r r �cfg s zConfigMerger.cfg)NNNNT) r r r r rU rX r_ rH rc �propertyrd r r r r rL � s � rL 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 �rg �r Z content_typer r r ri s z ContentHandlers.is_registeredFTc C sj t � }|�� D ](}|r"|�|� q| �|�s|�|� q|D ]}|| j|< q<|rf|| jvrf| j�|� |S r )�setZ list_types�addri rg rh rT )r �modrh Z overwrite�types�tr r r �register s zContentHandlers.registerc C s | j | S r ro rp r r r rl ( s z ContentHandlers._get_handlerc C s t | j�� �S r )�listrg �itemsr r r r rx + s zContentHandlers.itemsN)FT) r r r r rk rn ri rv rl rx r r r r rf s rf 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� rY r� r� � instance_data�instance_data_sensitive�combined_cloud_configZnetwork_configZ instance_idZmanual_clean_markerZobj_pklr� rB ZuserdataZ userdata_rawZ vendordataZ vendordata2Zvendordata2_rawZvendordata_rawrZ r[ Zvendor_scriptsr� r� r� )rb �getr| r} r ZDEFAULT_RUN_DIRr~ r9 r< r> � instance_linkZ boot_finishedZseed_dirZ template_tpl�lookupsrS )r r{ �dsZ template_dirr r r r 2 sT ����!zPaths.__init__)�ci_pkl_versionr` c C sr t | d�s t| j| jd�j| _d| jvr2d| jd< d| jvrFd| jd< d| jvrZd| jd< d | jvrnd | jd < d S ) z(Perform deserialization fixes for Paths.r~ )r{ r� r� r� r� r� r� r� r� N)�hasattrry rb rS 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 u r d S t|��tjd�}tj�| jd|�}| j � |�}|rbtj�||�}|S )Nr) Z instances) rS 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 u r |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 ry / s 5 ry 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)Z has_section�lowerZ add_sectionr rq r� r r r rq � 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� rq 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? rL rf ZCloudInitPickleMixinry r� r r r r �<module> s( :-j#