3
\me @ s d dl Z d dlZd dlZd dlmZmZ e jeZG dd de Z
G dd dZ G dd dZ G d d
d
e Z
G d d d e ZdS )
N)subputilc @ s e Zd ZdS )CustomScriptNotFoundN)__name__
__module__ __qualname__ r r */usr/lib/python3.6/config_custom_script.pyr s r c @ s* e Zd ZdZdZejjeeZdZ dZ
dS )CustomScriptConstantz/root/.customizationz customize.shz)/.guest-customization-post-reboot-pendingzpost-customize-guest.shN) r r r CUSTOM_TMP_DIRZCUSTOM_SCRIPT_NAMEospathjoin
CUSTOM_SCRIPTPOST_CUSTOM_PENDING_MARKERPOST_CUSTOM_SCRIPT_NAMEr r r r r
s
r
c @ s e Zd Zdd Zdd ZdS )RunCustomScriptc C s || _ || _tjj||| _d S )N)
scriptname directoryr r
r
scriptpath)selfr r r r r __init__! s zRunCustomScript.__init__c C sp t jj| js td| j tjtj t j
dtj tj | jtj tj
tj jdd}tjtj |dd d S )Nz3Script %s not found!! Cannot execute custom script!zCopying custom script to %s
id )mode)r r
existsr r r Z
ensure_dirr
r LOGdebugr copyZ load_filereplaceZ
write_file)r Zcontentr r r prepare_script& s
z RunCustomScript.prepare_scriptN)r r r r r r r r r r s r c @ s e Zd Zdd ZdS )PreCustomScriptc C s& t jd | j tjtjdg dS )z7Executing custom script with precustomization argument.z"Executing pre-customization scriptZprecustomizationN)r r r r r
r )r r r r execute> s
zPreCustomScript.executeN)r r r r" r r r r r! = s r! c s$ e Zd Z fddZdd Z ZS )PostCustomScriptc s. t t| j|| || _tjj|tj| _ d S )N)
superr# r ccScriptsDirr r
r r
r ccScriptPath)r r r r% ) __class__r r r F s zPostCustomScript.__init__c C sn | j tjd| j tjtjj| j t
j | j tj | j}tj
| j|jt jB tjd tjt
j dS )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 r r r& r r r r
r r r
r statchmodst_modeS_IEXECinfoZ ensure_filer )r str r r r" M s
zPostCustomScript.execute)r r r r r"
__classcell__r r )r' r r# E s r# )Zloggingr r( Z cloudinitr r Z getLoggerr r Exceptionr r
r r! r# r r r r s