U Ad-@sddlZddlmZddlmZmZmZmZddlm Z m Z ddlm Z ddlm Z mZddlmZddlmZdd lmZdd lmZdd lmZe eZd ZGd ddeZddZddZee e!dddZ"GdddZ#dS)N) ModuleType)DictList NamedTupleOptional)configimporter)log) type_utilsutil) ALL_DISTROS) ConfigMerger)ReportEventStack) FREQUENCIES)InitZcc_c@s2eZdZUeed<eed<eed<eeed<dS) ModuleDetailsmodulename frequencyrun_argsN)__name__ __module__ __qualname__r__annotations__strrrr:/usr/lib/python3/dist-packages/cloudinit/config/modules.pyrs rcCsX|dd}|dr.|dt|d}|}|s>dS|tsTdt|f}|S)N-_z.pyrz%s%s)replacelowerendswithlenstrip startswith MOD_PREFIX)rZ canon_namerrrform_module_name&s   r'cCst|drd|jksd|jkr4td|d|d|jdtkrdtd|d|d|jddt|d rtd|d|d dS) NmetardistroszModule '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)modrrrrvalidate_module2s" r.)module_detailscfgreturncCs2t|jjdi}|sdS||s.dSdS)NZactivate_by_schema_keysTF) frozensetrr(get intersectionkeys)r/r0Zactivate_by_schema_keys_keysrrr _is_activeHsr6c@sxeZdZdedddZeejdddZe e ddd Z e e dd d Z e e d d dZdddZddZdS)ModulesN)initcCs2||_||_d|_|dkr(tdddd}||_dS)Nzmodule-reporterz module-descF)r descriptionZreporting_enabled)r8 cfg_files _cached_cfgrreporter)selfr8r:r<rrr__init__TszModules.__init__)r1cCs<|jdkr0t|jj|jj|j|jjd}|j|_t|jS)N)paths datasourceZadditional_fnsZbase_cfg) r;r r8r?r@r:r0copydeepcopy)r=Zmergerrrrr0as z Modules.cfgcCs`g}||jkr|S|j|}|s&|S|D].}|s6q*t|trT|d|iq*t|tri}t|dkr~|d|d<t|dkr|d|d<t|dkr|dd|d<|r||q*t|trDi}d }d |kr|d |d<d }d |kr|d |d<d|kr,|dp&g|d<|rZ|rZ||q*t d |t |fq*|S)aURead 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-rfreqrNargsFrTrz3Failed to read '%s' item in config, unknown type %s) r0r3 isinstancerappendr$listr#dict TypeErrorr obj_name)r=rZ module_listZcfg_modsitemcontentsZvalidrrr _read_modulesosV               zModules._read_modulesc Csg}|D]}|d}|d}|dp*g}t|}|s:q|rX|tkrXtd||d}t|dtt gdg\}} |std|| qt |d } t | ||dkr| j d }| t| |||d q|S) zConvert 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-rErFz6Config specified module %s has an unknown frequency %sNhandlez,Could not find module named %s (searched %s)rr)rrrr)r3r'rLOGZwarningr find_moduler rLr import_moduler.r(rHr) r=raw_mods mostly_modsZraw_modZraw_namerErmod_nameZmod_locsZ looked_locsr-rrr_fixup_modulessP     zModules._fixup_modules)rVc Cs|j}g}g}|D]\}}}}ztd|||||j|t|g} ||d|} d| |f} t| | |jd} | 8|j| |j | |d\} }| rd| | _ n d| | _ W5QRXWqt k r}z"t td|||||fW5d}~XYqXq||fS) Nz(Running module %s (%s) with frequency %szconfig-zrunning %s with frequency %s)rr9parent)rEz%s ran successfullyz%s previously ranzRunning module %s (%s) failed)r8ZcloudifyrRdebugr0rHrr<runrQmessage Exceptionr Zlogexc)r=rVZccZfailuresZ which_ranr-rrErFZ func_argsrun_nameZdescZmyrepZranZ_rerrr _run_modulessD      "zModules._run_modulescCs&|||d}|g}||}||S)N)r-rFrE)rXr`)r=rWrFrEZ mod_to_berUrVrrr run_singles zModules.run_singlecCs||}||}|jjj}g}g}|jdg}g}g} |D]} | \} } } }| dkrZq@| jd}t| |js|| | q@|r|t gkr||kr| |kr| | q@| | | | | | |gq@|rt dd ||rt dd |||r t dd ||| S) zRuns 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, )rOrXr8Zdistrorr0r3r(r6rHr rRinfojoinr`)r=Z section_namerUrVZ distro_nameZskippedZforcedZ overriddenZinapplicable_modsZ active_modsr/r-rZ_freqZ_argsZworked_distrosrrr run_sectionsJ         zModules.run_section)NN)NN)rrrrr>propertyrZConfigr0rrrOrrXr`rarerrrrr7Ss  ;0) r7)$rAtypesrtypingrrrrZ cloudinitrrr Zloggingr r Zcloudinit.distrosr Zcloudinit.helpersr Zcloudinit.reporting.eventsrZcloudinit.settingsrZcloudinit.stagesrZ getLoggerrrRr&rr'r.rJboolr6r7rrrr s"