File: //usr/lib/python3/dist-packages/cloudinit/config/__pycache__/modules.cpython-310.pyc
o
�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 de de de!fdd�Z"G dd� d�Z#dS )� N)�
ModuleType)�Dict�List�
NamedTuple�Optional)�config�importer)�log)�
type_utils�util)� ALL_DISTROS)� ConfigMerger)�ReportEventStack)� FREQUENCIES)�Init�cc_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�s*dt|f }|S )N�-�_z.pyr � z%s%s)�replace�lower�endswith�len�strip�
startswith�
MOD_PREFIX)r �
canon_namer r r �form_module_name& s
r) c C s� t | d�rd| jvsd| jvrtd| � d|� d���| jd tvr2td| � d|� d| jd � d���t | d �rBtd| � d|� d
���d S ) N�metar �distroszModule 'z
' with name 'z4' MUST have a 'meta' attribute of type 'MetaSchema'.z' has an invalid frequency �.�schemazX' has a JSON 'schema' attribute defined. Please define schema in cloud-init-schema,json.)�hasattrr* �
ValueErrorr )�modr r r r �validate_module2 s( ������
��r1 �module_details�cfg�returnc C s2 t | jj�di ��}|sdS |�|�� �sdS dS )N�activate_by_schema_keysTF)� frozensetr r* �get� intersection�keys)r2 r3 � activate_by_schema_keys_keysr r r �
_is_activeH s �r; c @ sx e Zd Zddefdd�Zedejfdd��Zde e
fdd �Z de e fd
d �Z
d e e fd
d�Zddd�Zdd� ZdS )�ModulesN�initc C s2 || _ || _d | _|d u rtdddd�}|| _d S )Nzmodule-reporterz module-descF)r � description�reporting_enabled)r= � cfg_files� _cached_cfgr �reporter)�selfr= r@ rB r r r �__init__T s �
zModules.__init__r4 c C s< | j d u rt| jj| jj| j| jjd�}|j| _ t�| j �S )N)�paths�
datasource�additional_fns�base_cfg) rA r
r= rE rF r@ r3 �copy�deepcopy)rC �mergerr r r r3 a s
� z Modules.cfgc C sP g }|| j vr |S | j �|�}|s|S |D ]�}|sqt|t�r)|�d|�� i� qt|t�rbi }t|�dkr>|d �� |d< t|�dkrL|d �� |d< t|�dkrZ|dd� |d< |ra|�|� qt|t�r�i }d }d
|v ry|d
�� |d< d }d |v r�|d �� |d< d|v r�|d p�g |d< |r�|r�|�|� qt d
|t
� |�f ��|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.
r0 � r � �freqr N�argsFr Tr z3Failed to read '%s' item in config, unknown type %s) r3 r7 �
isinstancer �appendr% �listr$ �dict� TypeErrorr
�obj_name)rC r � module_list�cfg_mods�item�contents�validr r r �
_read_moduleso sX
��
�
� ��zModules._read_modulesc C s� g }|D ]a}|d }|� d�}|� d�pg }t|�}|s q|r,|tvr,t�d||� d}t�|dt�t �gdg�\}} |sEt�d|| � qt�
|d �}
t |
|� |du rZ|
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.
r0 rN rO z6Config specified module %s has an unknown frequency %sN� �handlez,Could not find module named %s (searched %s)r r )r r r r )r7 r) r �LOG�warningr � find_moduler
rU r �
import_moduler1 r* rQ r ) rC �raw_mods� mostly_mods�raw_mod�raw_namerN r �mod_name�mod_locs� looked_locsr0 r r r �_fixup_modules� sL
���
��zModules._fixup_modulesrc c C s | j �� }g }g }|D ]|\}}}}zUt�d|||� || j|t|g} |�|� d|� �}
d|
|f } t|
| | jd�} | �! |j|
|j | |d�\}
}|
rQd|
| _
nd|
| _
W d � n1 s`w Y W q t y� } zt �
td||� |�||f� W Y d }~q d }~ww ||fS ) Nz(Running module %s (%s) with frequency %szconfig-z running %s with frequency %s)r r> �parent)rN z%s ran successfullyz%s previously ranz Running module %s (%s) failed)r= �cloudifyr^ �debugr3 rQ r rB �runr] �message� Exceptionr �logexc)rC rc �cc�failures� which_ranr0 r rN rO � func_args�run_name�desc�myrep�ran�_r�er r r � _run_modules� s>
�
�
�
� �� ��zModules._run_modulesc C s& |||d�}|g}| � |�}| �|�S )N)r0 rO rN )ri r{ )rC rf rO rN � mod_to_berb rc r r r �
run_single s �
zModules.run_singlec C s | � |�}| �|�}| jjj}g }g }| j�dg �}g }g } |D ]A}
|
\} } }
}| du r-q | jd }t|
| j�s>|� | � q |rX|t
gkrX||vrX| |vrS|� | � q |� | � | � | | |
|g� q |rmt � dd�
|�� |ryt � 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
�unverified_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, )r[ ri r= �distror r3 r7 r* r; rQ r r^ �info�joinr{ )rC � section_namerb rc � distro_name�skipped�forced�
overridden�inapplicable_mods� active_modsr2 r0 r �_freq�_args�worked_distrosr r r � run_section sJ
��
zModules.run_section)NN)r r r r rD �propertyr �Configr3 r r r[ r ri r{ r} r� r r r r r< S s
;0
) r< )$rI �typesr �typingr r r r � cloudinitr r r �loggingr
r �cloudinit.distrosr �cloudinit.helpersr
�cloudinit.reporting.eventsr �cloudinit.settingsr �cloudinit.stagesr � getLoggerr r^ r' r r) r1 rS �boolr; r< r r r r �<module> s$