o x[h" @sddlZddlZddlZddlmZmZeeZGddde Z GdddZ GdddZ Gd d d e Z Gd d d e ZdS) N)subputilc@s eZdZdS)CustomScriptNotFoundN)__name__ __module__ __qualname__rr[/usr/lib/python3/dist-packages/cloudinit/sources/helpers/vmware/imc/config_custom_script.pyrsrc@s*eZdZdZdZejeeZdZ dZ dS)CustomScriptConstantz/root/.customizationz customize.shz)/.guest-customization-post-reboot-pendingzpost-customize-guest.shN) rrrCUSTOM_TMP_DIRCUSTOM_SCRIPT_NAMEospathjoin CUSTOM_SCRIPTPOST_CUSTOM_PENDING_MARKERPOST_CUSTOM_SCRIPT_NAMErrrr r s r c@seZdZddZddZdS)RunCustomScriptcCs ||_||_tj|||_dSN) scriptname directoryr rr scriptpath)selfrrrrr __init__!szRunCustomScript.__init__cCsptj|jstd|jttjt dtj t |jtj t tj dd}tjtj |dddS)Nz3Script %s not found!! Cannot execute custom script!zCopying custom script to %s id)mode)r rexistsrrr ensure_dirr r LOGdebugrcopyload_text_filereplace write_file)rcontentrrr prepare_script&s&  zRunCustomScript.prepare_scriptN)rrrrr&rrrr r s rc@seZdZddZdS)PreCustomScriptcCs&td|ttjdgdS)z7Executing custom script with precustomization argument.z"Executing pre-customization scriptprecustomizationN)rr r&rr r)rrrr execute>s zPreCustomScript.executeN)rrrr)rrrr r'=s r'cs$eZdZfddZddZZS)PostCustomScriptcs.tt|||||_tj|tj|_ dSr) superr*r ccScriptsDirr rrr r ccScriptPath)rrrr, __class__rr rFs  zPostCustomScript.__init__cCsn|td|jttj|j t j |jt |j}t |j|jt jBtdtt jdS)z This method copy the post customize run script to cc_scripts_per_instance directory and let this module to run post custom script. z'Copying post customize run script to %sz*Creating post customization pending markerN)r&rr r-rr!r rrrr rstatchmodst_modeS_IEXECinfo ensure_filer)rstrrr r)Ms  zPostCustomScript.execute)rrrrr) __classcell__rrr.r r*Es r*)loggingr r0 cloudinitrr getLoggerrr Exceptionrr rr'r*rrrr s