1#@!#!123s

: / lib / python3.6 / site-packages / cloudinit / sources / helpers / vmware / imc /

Filename : config_custom_script.py

back

# Copyright (C) 2017 Canonical Ltd.
# Copyright (C) 2017-2019 VMware Inc.
#
# Author: Maitreyee Saikia <msaikia@vmware.com>
#
# This file is part of cloud-init. See LICENSE file for license information.

import logging
import os
import stat

from cloudinit import subp, util

LOG = logging.getLogger(__name__)


class CustomScriptNotFound(Exception):
    pass


class CustomScriptConstant:
    CUSTOM_TMP_DIR = "/root/.customization"

    # The user defined custom script
    CUSTOM_SCRIPT_NAME = "customize.sh"
    CUSTOM_SCRIPT = os.path.join(CUSTOM_TMP_DIR, CUSTOM_SCRIPT_NAME)
    POST_CUSTOM_PENDING_MARKER = "/.guest-customization-post-reboot-pending"
    # The cc_scripts_per_instance script to launch custom script
    POST_CUSTOM_SCRIPT_NAME = "post-customize-guest.sh"


class RunCustomScript:
    def __init__(self, scriptname, directory):
        self.scriptname = scriptname
        self.directory = directory
        self.scriptpath = os.path.join(directory, scriptname)

    def prepare_script(self):
        if not os.path.exists(self.scriptpath):
            raise CustomScriptNotFound(
                "Script %s not found!! Cannot execute custom script!"
                % self.scriptpath
            )

        util.ensure_dir(CustomScriptConstant.CUSTOM_TMP_DIR)

        LOG.debug(
            "Copying custom script to %s", CustomScriptConstant.CUSTOM_SCRIPT
        )
        util.copy(self.scriptpath, CustomScriptConstant.CUSTOM_SCRIPT)

        # Strip any CR characters from the decoded script
        content = util.load_file(CustomScriptConstant.CUSTOM_SCRIPT).replace(
            "\r", ""
        )
        util.write_file(
            CustomScriptConstant.CUSTOM_SCRIPT, content, mode=0o544
        )


class PreCustomScript(RunCustomScript):
    def execute(self):
        """Executing custom script with precustomization argument."""
        LOG.debug("Executing pre-customization script")
        self.prepare_script()
        subp.subp([CustomScriptConstant.CUSTOM_SCRIPT, "precustomization"])


class PostCustomScript(RunCustomScript):
    def __init__(self, scriptname, directory, ccScriptsDir):
        super(PostCustomScript, self).__init__(scriptname, directory)
        self.ccScriptsDir = ccScriptsDir
        self.ccScriptPath = os.path.join(
            ccScriptsDir, CustomScriptConstant.POST_CUSTOM_SCRIPT_NAME
        )

    def execute(self):
        """
        This method copy the post customize run script to
        cc_scripts_per_instance directory and let this
        module to run post custom script.
        """
        self.prepare_script()

        LOG.debug("Copying post customize run script to %s", self.ccScriptPath)
        util.copy(
            os.path.join(
                self.directory, CustomScriptConstant.POST_CUSTOM_SCRIPT_NAME
            ),
            self.ccScriptPath,
        )
        st = os.stat(self.ccScriptPath)
        os.chmod(self.ccScriptPath, st.st_mode | stat.S_IEXEC)
        LOG.info("Creating post customization pending marker")
        util.ensure_file(CustomScriptConstant.POST_CUSTOM_PENDING_MARKER)

Name	Size	Last Modified	Owner	Permissions
__pycache__	Dir	April 24 2025 3:40:45	root	0755
__init__.py	0 KB	December 04 2023 11:47:40	root	0644
boot_proto.py	0.301 KB	December 04 2023 11:47:40	root	0644
config.py	4.448 KB	December 04 2023 11:47:40	root	0644
config_custom_script.py	3.028 KB	December 04 2023 11:47:40	root	0644
config_file.py	2.155 KB	December 04 2023 11:47:40	root	0644
config_nic.py	8.273 KB	December 04 2023 11:47:40	root	0644
config_passwd.py	2.365 KB	December 04 2023 11:47:40	root	0644
config_source.py	0.27 KB	December 04 2023 11:47:40	root	0644
guestcust_error.py	0.401 KB	December 04 2023 11:47:40	root	0644
guestcust_event.py	0.45 KB	December 04 2023 11:47:40	root	0644
guestcust_state.py	0.35 KB	December 04 2023 11:47:40	root	0644
guestcust_util.py	19.958 KB	December 04 2023 11:47:40	root	0644
nic.py	3.211 KB	December 04 2023 11:47:40	root	0644
nic_base.py	3.78 KB	December 04 2023 11:47:40	root	0644

2017 © D7net \| D704T team