File: //lib/python3.6/site-packages/cloudinit/config/__pycache__/modules.cpython-36.pyc
3
�q�h / � @ s� d dl Z d dlmZ d dlmZ d dlmZmZmZm Z d dl
m Z m Z d dl
m
Z d dl
mZmZ d dlmZ d d lmZ d d
lmZ d d lmZ d d lmZ eje �Z d
Z G dd� de�Z dd� Z dd� Z!e e"e#d�dd�Z$G dd� d�Z%dS )� N)� signature)�
ModuleType)�Dict�List�
NamedTuple�Optional)�config�importer)�log)�
type_utils�util)� ALL_DISTROS)� ConfigMerger)�ReportEventStack)� FREQUENCIES)�InitZcc_c @ s" e Zd ZU eeeee dS )�
ModuleDetailsN)
�__name__�
__module__� __qualname__r �module�str�name� frequencyr �run_args� r r � /usr/lib/python3.6/modules.pyr s
r c C sX | j dd�}|j� jd�r.|dt|�d � }|j� }|s>d S |jt�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� s d| jks d| jkr6td| � d|� d���| jd tkrftd| � 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_module3 s
r. )�module_details�cfg�returnc C s2 t | jjjdi ��}|s dS |j|j� �s.dS dS )NZactivate_by_schema_keysTF)� frozensetr r( �get� intersection�keys)r/ r0 Z activate_by_schema_keys_keysr r r �
_is_activeI 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__U s zModules.__init__)r1 c C s< | j d kr0t| jj| jj| j| jjd�}|j| _ tj| j �S )N)�paths�
datasourceZadditional_fnsZbase_cfg) r; r r8 r? r@ r: r0 �copy�deepcopy)r= Zmergerr r r r0 b s
z Modules.cfgc C sh g }|| j kr|S | j j|�}|s&|S �x:|D �]0}|s:q.t|t�rX|jd|j� i� q.t|t�r�i }t|�dkr�|d j� |d< t|�dkr�|d j� |d< t|�dkr�|dd� |d< |r�|j|� q.t|t��rJi }d }d
|k� r�|d
j� |d< d }d |k�r|d j� |d< d|k�r2|d �p,g |d< |�r`|�r`|j|� q.t d
|t
j |�f ��q.W |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_modulesp sL
zModules._read_modulesc C s� g }x�|D ]�}|d }|j d�}|j d�p,g }t|�}|s<q
|rZ|tkrZtjd||� d}tj|dtjt �gdg�\}} |s�tjd|| � q
tj
|d �}
t |
|� |dkr�|
j d
}|j
t|
|||d �� q
W |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� sB
zModules._fixup_modules)rV c C sF | j j� }g }g }�x(|D �] \}}}}y�tjd|||� |j|� d|� �} d| |f }
t| |
| jd�} || j||d�} | �p t|j �}
|
j
}t |�dkr�t j
ddd � | jd
ti� |j| |j | |d �\}}|r�d | | _n
d
| | _W d Q R X W q tk
�r8 } z$t jtd||� |j||f� W Y d d }~X qX qW ||fS )Nz(Running module %s (%s) with frequency %szconfig-z running %s with frequency %s)r r9 �parent)r r0 ZcloudrF � z%Config modules with a `log` parameterz23.2)Z
deprecatedZdeprecated_versionr
)rE z%s ran successfullyz%s previously ranz Running module %s (%s) failed)r8 ZcloudifyrR �debugrH r r< r0 r rQ Z
parametersr# r Z deprecate�updateZrun�message� ExceptionZlogexc)r= rV ZccZfailuresZ which_ranr- r rE rF Zrun_nameZdescZmyrepZ func_argsZfunc_signatureZ func_paramsZranZ_r�er r r � _run_modules� sB
$zModules._run_modulesc C s& |||d�}|g}| j |�}| j|�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 | j |�}| j|�}| jjj}g }g }| jjdg �}g }g } x�|D ]�}
|
\} } }
}| dkr\qB| jd }t|
| j�s~|j | � qB|r�|t
gkr�||kr�| |kr�|j | � qB|j | � | j | | |
|g� qBW |r�t j ddj
|�� |r�t j ddj
|�|� |�rt j ddj
|�� | j| �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 sF
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 T s
;02
r7 )&rA �inspectr �typesr Ztypingr 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$