File: //snap/core20/2669/lib/python3/dist-packages/cloudinit/config/__pycache__/modules.cpython-38.pyc
U
Ӈg�4 � @ s� d dl Z d dlZd dlmZ d dlmZ d dlmZmZm Z m
Z
d dl m Z m
Z
mZmZmZmZ d dlmZ d dlmZ d dlmZ d d lmZ d d
lmZ e� e �Z d Z d d
gZ ddiZ!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� lifecycle� performance�
type_utils�util)� ALL_DISTROS)� ConfigMerger)�ReportEventStack)� FREQUENCIES)�InitZcc_Z cc_migratorZcc_rightscale_userdataZcc_ubuntu_advantageZ
cc_ubuntu_proc @ 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 2 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_name9 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_moduleE s" ���� �
�r0 )�module_details�cfg�returnc C s2 t | jj�di ��}|s dS |�|�� �s.dS dS )NZactivate_by_schema_keysTF)� frozensetr r* �get� intersection�keys)r1 r2 Z activate_by_schema_keys_keysr r r �
_is_active[ s �r8 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)r: � cfg_files� _cached_cfgr �reporter)�selfr: r< r> r r r �__init__g s �zModules.__init__)r3 c C s< | j d kr0t| jj| jj| j| jjd�}|j| _ t�| j �S )N)�paths�
datasourceZadditional_fnsZbase_cfg) r= r r: rA rB r< r2 �copy�deepcopy)r? Zmergerr r r r2 t 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) r2 r5 �
isinstancer �appendr& �listr% �dict� TypeErrorr �obj_name)r? r Z module_listZcfg_mods�item�contentsZvalidr r r �
_read_modules� sV
��
��zModules._read_modulesc C s. g }|D �] }|d }|� d�}|� d�p,g }t|�}|s<q|rf|tkrftjd|� d|� �dd� d}|tkr�tjd |� d
t| � d �d d� t| }t�|d
t� t
�gdg�\}} |s�|t kr�t �
d|dd� � qt �d|| � qt�|d �}
t|
|� |dk�r|
jd }|�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/ rG rH zConfig specified module z has an unknown frequency z22.1�Z
deprecatedZdeprecated_versionNz Module has been renamed from z to z/. Update any references in /etc/cloud/cloud.cfgz24.1� �handlez\Module `%s` has been removed from cloud-init. It may be removed from `/etc/cloud/cloud.cfg`.r! z,Could not find module named %s (searched %s)r r )r r r r )r5 r) r r
� deprecate�RENAMED_MODULESr � find_moduler rN r �REMOVED_MODULES�LOG�infoZwarning�
import_moduler0 r* rJ r ) r? �raw_mods� mostly_modsZraw_modZraw_namerG r �mod_nameZmod_locsZ looked_locsr/ r r r �_fixup_modules� sd
� � �
��
��zModules._fixup_modules)r] c C sh | j �� }g }g }|D �]F\}}}}z�t�d|||� |�|� d|� �} d| |f }
t| |
| jd�} || j||d�} | �� t|j �}
|
j
}t |�dkr�t j
ddd � | �d
ti� tjd d d
�� }|j| |j | |d�\}}W 5 Q R X |�r| � d|jd�d�| _n
d| | _W 5 Q R X W q tk
�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 r; �parent)r r2 ZcloudrH � z%Config modules with a `log` parameterz23.2rR �logrS �skip)Zlog_mode)rG z ran successfully and took z.3fz secondsz%s previously ranz Running module %s (%s) failed)r: ZcloudifyrY �debugrJ r r> r2 r rT Z
parametersr% r
rU �updater ZTimed�runZdelta�message� Exceptionr
Zlogexc)r? r] ZccZfailuresZ which_ranr/ r rG rH �run_nameZdescZmyrepZ func_argsZfunc_signatureZ func_paramsZtimerZranZ_r�er r r � _run_modules s`
�
��
� ��"zModules._run_modulesc C s& |||d�}|g}| � |�}| �|�S )N)r/ rH rG )r_ rk )r? r^ rH rG Z mod_to_ber\ r] r r r �
run_single7 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, )rQ r_ r: Zdistror r2 r5 r* r8 rJ r rY rZ �joinrk )r? Z section_namer\ r] Z distro_nameZskippedZforcedZ
overriddenZinapplicable_modsZ active_modsr1 r/ r Z_freqZ_argsZworked_distrosr r r � run_sectionC sJ
��zModules.run_section)NN)NN)r r r r r@ �propertyr ZConfigr2 r r rQ r r_ rk rl ro r r r r r9 f s
;C7
r9 ))rC Zlogging�inspectr �typesr �typingr r r r Z cloudinitr r r
r r r
Zcloudinit.distrosr Zcloudinit.helpersr Zcloudinit.reporting.eventsr Zcloudinit.settingsr Zcloudinit.stagesr Z getLoggerr rY r( rX rV r r) r0 rL �boolr8 r9 r r r r �<module> s.
� �