Your IP : 216.73.217.13
U
�Ad�- � @ s� d dl Z d dlmZ d dlmZmZmZmZ d dlm Z m
Z
d dlm Z d dlm
Z
mZ d dlmZ d dlmZ d d lmZ d d
lmZ d d lmZ e �e�Zd Z G d
d� de�Z dd� Z dd� Z e e e!d�dd�Z"G dd� d�Z#dS )� N)�
ModuleType)�Dict�List�
NamedTuple�Optional)�config�importer)�log)�
type_utils�util)� ALL_DISTROS)� ConfigMerger)�ReportEventStack)� FREQUENCIES)�InitZcc_c @ s2 e Zd ZU eed< eed< eed< ee ed< dS )�
ModuleDetails�module�name� frequency�run_argsN)�__name__�
__module__� __qualname__r �__annotations__�strr � r r �:/usr/lib/python3/dist-packages/cloudinit/config/modules.pyr s
r c C sX | � dd�}|�� �d�r.|dt|�d � }|�� }|s>d S |�t�sTdt|f }|S )N�-�_z.pyr � z%s%s)�replace�lower�endswith�len�strip�
startswith�
MOD_PREFIX)r Z
canon_namer r r �form_module_name& s
r' c C s� t | d�r d| jks d| jkr4td| � d|� d���| jd tkrdtd| � d|� d| jd � d���t | d �r�td| � d|� d
���d S ) N�metar �distroszModule 'z
' with name 'z4' MUST have a 'meta' attribute of type 'MetaSchema'.z' has an invalid frequency �.ZschemazX' has a JSON 'schema' attribute defined. Please define schema in cloud-init-schema,json.)�hasattrr( �
ValueErrorr )�modr r r r �validate_module2 s" ���� �
�r. )�module_details�cfg�returnc C s2 t | jj�di ��}|s dS |�|�� �s.dS dS )NZactivate_by_schema_keysTF)� frozensetr r( �get� intersection�keys)r/ r0 Z activate_by_schema_keys_keysr r r �
_is_activeH s �r6 c @ sx e Zd Zded�dd�Zeejd�dd��Ze e
d�dd �Z e e d�d
d �Z
e e d �d
d�Zddd�Zdd� ZdS )�ModulesN)�initc C s2 || _ || _d | _|d kr(tdddd�}|| _d S )Nzmodule-reporterz module-descF)r � descriptionZreporting_enabled)r8 � cfg_files� _cached_cfgr �reporter)�selfr8 r: r< r r r �__init__T s �zModules.__init__)r1 c C s< | j d kr0t| jj| jj| j| jjd�}|j| _ t�| j �S )N)�paths�
datasourceZadditional_fnsZbase_cfg) r; r
r8 r? r@ r: r0 �copy�deepcopy)r= Zmergerr r r r0 a s
�z Modules.cfgc C s` g }|| j kr|S | j �|�}|s&|S |D �].}|s6q*t|t�rT|�d|�� i� q*t|t�r�i }t|�dkr~|d �� |d< t|�dkr�|d �� |d< t|�dkr�|dd� |d< |r�|�|� q*t|t��rDi }d }d
|kr�|d
�� |d< d }d |k�r|d �� |d< d|k�r,|d �p&g |d< |�rZ|�rZ|�|� q*t d
|t
� |�f ��q*|S )aU Read the modules from the config file given the specified name.
Returns a list of module definitions. E.g.,
[
{
"mod": "bootcmd",
"freq": "always",
"args": "some_arg",
}
]
Note that in the default case, only "mod" will be set.
r- � r � �freqr N�argsFr Tr z3Failed to read '%s' item in config, unknown type %s) r0 r3 �
isinstancer �appendr$ �listr# �dict� TypeErrorr
�obj_name)r= r Z module_listZcfg_mods�item�contentsZvalidr r r �
_read_moduleso sV
��
��zModules._read_modulesc C s� g }|D ]�}|d }|� d�}|� d�p*g }t|�}|s:q|rX|tkrXt�d||� d}t�|dt�t �gdg�\}} |s�t�d|| � qt�
|d �}
t |
|� |dkr�|
j d
}|�
t|
|||d �� q|S ) z�Convert list of returned from _read_modules() into new format.
Invalid modules and arguments are ignored.
Also ensures that the module has the required meta fields.
r- rE rF z6Config specified module %s has an unknown frequency %sN� �handlez,Could not find module named %s (searched %s)r r )r r r r )r3 r' r �LOGZwarningr � find_moduler
rL r �
import_moduler. r( rH r ) r= �raw_mods� mostly_modsZraw_modZraw_namerE r �mod_nameZmod_locsZ looked_locsr- r r r �_fixup_modules� sP
� ��
��zModules._fixup_modules)rV c C s | j �� }g }g }|D ]�\}}}}z�t�d|||� || j|t|g} |�|� d|� �}
d|
|f } t|
| | jd�} | �8 |j|
|j | |d�\}
}|
r�d|
| _
n
d|
| _
W 5 Q R X W q t k
r� } z"t �
td||� |�||f� W 5 d }~X Y qX q||fS ) Nz(Running module %s (%s) with frequency %szconfig-z running %s with frequency %s)r r9 �parent)rE z%s ran successfullyz%s previously ranz Running module %s (%s) failed)r8 ZcloudifyrR �debugr0 rH r r< �runrQ �message� Exceptionr Zlogexc)r= rV ZccZfailuresZ which_ranr- r rE rF Z func_args�run_nameZdescZmyrepZranZ_r�er r r � _run_modules� sD
�
� �
"zModules._run_modulesc C s& |||d�}|g}| � |�}| �|�S )N)r- rF rE )rX r` )r= rW rF rE Z mod_to_berU rV r r r �
run_single s �
zModules.run_singlec C s | � |�}| �|�}| jjj}g }g }| j�dg �}g }g } |D ]�}
|
\} } }
}| dkrZq@| jd }t|
| j�s||� | � q@|r�|t
gkr�||kr�| |kr�|� | � q@|� | � | � | | |
|g� q@|r�t � dd�
|�� |r�t � dd�
|�|� |�r
t � dd�
|�� | �| �S ) z�Runs all modules in the given section.
section_name - One of the modules lists as defined in
/etc/cloud/cloud.cfg. One of:
- cloud_init_modules
- cloud_config_modules
- cloud_final_modules
Zunverified_modulesNr) z?Skipping modules '%s' because no applicable config is provided.�,zSkipping modules '%s' because they are not verified on distro '%s'. To run anyway, add them to 'unverified_modules' in config.z running unverified_modules: '%s'z, )rO rX r8 Zdistror r0 r3 r( r6 rH r rR �info�joinr` )r= Z section_namerU rV Z distro_nameZskippedZforcedZ
overriddenZinapplicable_modsZ active_modsr/ r- r Z_freqZ_argsZworked_distrosr r r � run_section sJ
��zModules.run_section)NN)NN)r r r r r> �propertyr ZConfigr0 r r rO r rX r` ra re r r r r r7 S s
;0)
r7 )$rA �typesr �typingr r r r Z cloudinitr r r Zloggingr
r Zcloudinit.distrosr Zcloudinit.helpersr
Zcloudinit.reporting.eventsr Zcloudinit.settingsr Zcloudinit.stagesr Z getLoggerr rR r& r r' r. rJ �boolr6 r7 r r r r �<module> s"