Google
Edit File: modules.cpython-39.pyc
a '�Dg�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.9/site-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| jvs d| jvr4td| � d|� d���| jd tvrdtd| � 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 u r(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 u r0t| 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 vr|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 |v r�|d �� |d< d }d |v �r|d �� |d< d|v �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� sT �� ��zModules._read_modulesc C s. g }|D �] }|d }|� d�}|� d�p,g }t|�}|s<q|rf|tvrftjd|� d|� �dd� d}|tv r�tjd |� d t| � d �d d� t| }t�|d t� t �gdg�\}} |s�|t v r�t � d|dd� � qt �d|| � qt�|d �} t| |� |du �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� sh �� ��� �� ��zModules._fixup_modules)r] c C s� | j �� }g }g }|D �]t\}}}}�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 d � n1 s�0 Y |�r| � d|jd�d�| _n d| | _W d � n1 �s:0 Y W q t�y� } z*t�td||� |�||f� W Y d }~qd }~0 0 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 � 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 sT � �� � �(��.&zModules._run_modulesc C s& |||d�}|g}| � |�}| �|�S )N)r/ rH rG )r_ rl )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 ]�} | \} } } }| du rZq@| jd }t| | j�s||� | � q@|r�|t gkr�||vr�| |vr�|� | � 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 �joinrl )r? Z section_namer\ r] Z distro_nameZskippedZforcedZ overriddenZinapplicable_modsZ active_modsr1 r/ r Z_freq�_argsZworked_distrosr r r � run_sectionC sJ ��zModules.run_section)NN)NN)r r r r r@ �propertyr ZConfigr2 r r rQ r r_ rl rm rq 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, ��