o w7e/@sddlZddlmZddlmZddlmZmZmZm Z ddl m Z m Z ddl m Zddl mZmZddlmZdd lmZdd lmZdd lmZdd lmZeeZd ZGdddeZddZ ddZ!dede"de#fddZ$GdddZ%dS)N) signature) ModuleType)DictList NamedTupleOptional)configimporter)log) type_utilsutil) ALL_DISTROS) ConfigMerger)ReportEventStack) FREQUENCIES)Initcc_c@s2eZdZUeed<eed<eed<eeed<dS) ModuleDetailsmodulename frequencyrun_argsN)__name__ __module__ __qualname__r__annotations__strrrr:/usr/lib/python3/dist-packages/cloudinit/config/modules.pyr s rcCsX|dd}|dr|dt|d}|}|sdS|ts*dt|f}|S)N-_z.pyrz%s%s)replacelowerendswithlenstrip startswith MOD_PREFIX)r canon_namerrrform_module_name's   r*cCst|drd|jvsd|jvrtd|d|d|jdtvr2td|d|d|jddt|d rBtd|d|d dS) NmetardistroszModule '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)modrrrrvalidate_module3s$   r2module_detailscfgreturncCs2t|jjdi}|sdS||sdSdS)Nactivate_by_schema_keysTF) frozensetrr+get intersectionkeys)r3r4activate_by_schema_keys_keysrrr _is_activeIsr<c@sxeZdZddefddZedejfddZde e fdd Z de e fd d Z d e e fd dZdddZddZdS)ModulesNinitcCs2||_||_d|_|durtdddd}||_dS)Nzmodule-reporterz module-descF)r descriptionreporting_enabled)r> cfg_files _cached_cfgrreporter)selfr>rArCrrr__init__Us zModules.__init__r5cCs<|jdurt|jj|jj|j|jjd}|j|_t|jS)N)paths datasourceadditional_fnsbase_cfg) rBrr>rFrGrAr4copydeepcopy)rDmergerrrrr4bs  z Modules.cfgcCsPg}||jvr |S|j|}|s|S|D]}|sqt|tr)|d|iqt|trbi}t|dkr>|d|d<t|dkrL|d|d<t|dkrZ|dd|d<|ra||qt|tri}d }d |vry|d |d<d }d |vr|d |d<d|vr|dpg|d<|r|r||qt d |t |f|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. r1rfreqr!NargsFrTrz3Failed to read '%s' item in config, unknown type %s) r4r8 isinstancerappendr&listr%dict TypeErrorr obj_name)rDr module_listcfg_modsitemcontentsvalidrrr _read_modulespsX           zModules._read_modulesc Csg}|D]a}|d}|d}|dpg}t|}|sq|r,|tvr,td||d}t|dtt gdg\}} |sEtd|| qt |d } t | ||durZ| 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. r1rOrPz6Config specified module %s has an unknown frequency %sNhandlez,Could not find module named %s (searched %s)rr)rrrr)r8r*rLOGwarningr find_moduler rVr import_moduler2r+rRr) rDraw_mods mostly_modsraw_modraw_namerOrmod_namemod_locs looked_locsr1rrr_fixup_modulessL    zModules._fixup_modulesrdc CsP|j}g}g}|D]\}}}}zqtd|||||d|} d| |f} t| | |jd} ||j||d} | =t|j } | j }t |dkrYt j ddd | d ti|j| |j | |d \}}|rmd | | _nd | | _Wdn1s|wYWq ty}zt td|||||fWYd}~q d}~ww||fS)Nz(Running module %s (%s) with frequency %szconfig-zrunning %s with frequency %s)rr?parent)rr4cloudrPz%Config modules with a `log` parameterz23.2) deprecateddeprecated_versionr )rOz%s ran successfullyz%s previously ranzRunning module %s (%s) failed)r>cloudifyr_debugrRrrCr4rr^ parametersr%r deprecateupdaterunmessage Exceptionlogexc)rDrdccfailures which_ranr1rrOrPrun_namedescmyrep func_argsfunc_signature func_paramsran_rerrr _run_modulessV          zModules._run_modulescCs&|||d}|g}||}||S)N)r1rPrO)rjr)rDrgrPrO mod_to_bercrdrrr run_single s  zModules.run_singlecCs||}||}|jjj}g}g}|jdg}g}g} |D]A} | \} } } }| dur-q | jd}t| |js>| | q |rX|t gkrX||vrX| |vrS| | q | | | | | | |gq |rmt dd ||ryt dd |||rt 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 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\rjr>distrorr4r8r+r<rRr r_infojoinr)rD section_namercrd distro_nameskippedforced overriddeninapplicable_mods active_modsr3r1r_freq_argsworked_distrosrrr run_sectionsJ         zModules.run_section)NN)rrrrrEpropertyrConfigr4rrr\rrjrrrrrrrr=Ts  ;0 2 r=)&rJinspectrtypesrtypingrrrr cloudinitrr r loggingr r cloudinit.distrosr cloudinit.helpersrcloudinit.reporting.eventsrcloudinit.settingsrcloudinit.stagesr getLoggerrr_r(rr*r2rTboolr<r=rrrrs&