golden hour

/usr/lib/python3.9/site-packages/cloudinit/config

File/Folder	Size	Actions
__init__.py	40 B
__pycache__	-
cc_ansible.py	7.68 KB
cc_apk_configure.py	4.23 KB
cc_apt_configure.py	38.08 KB
cc_apt_pipelining.py	1.85 KB
cc_bootcmd.py	1.65 KB
cc_byobu.py	2.73 KB
cc_ca_certs.py	8.67 KB
cc_chef.py	12.19 KB
cc_disable_ec2_metadata.py	1.6 KB
cc_disk_setup.py	29.61 KB
cc_fan.py	1.95 KB
cc_final_message.py	2.51 KB
cc_growpart.py	18.37 KB
cc_grub_dpkg.py	5.42 KB
cc_install_hotplug.py	3.19 KB
cc_keyboard.py	1.49 KB
cc_keys_to_console.py	2.09 KB
cc_landscape.py	3.06 KB
cc_locale.py	1.19 KB
cc_lxd.py	13.47 KB
cc_mcollective.py	4.11 KB
cc_mounts.py	18.16 KB
cc_ntp.py	18.97 KB
cc_package_update_upgrade_install.py	3.79 KB
cc_phone_home.py	3.75 KB
cc_power_state_change.py	5.92 KB
cc_puppet.py	10.72 KB
cc_reset_rmc.py	4.33 KB
cc_resizefs.py	10.63 KB
cc_resolv_conf.py	3.17 KB
cc_rh_subscription.py	15.21 KB
cc_rsyslog.py	11.55 KB
cc_runcmd.py	1.6 KB
cc_salt_minion.py	3.96 KB
cc_scripts_per_boot.py	1.27 KB
cc_scripts_per_instance.py	1.26 KB
cc_scripts_per_once.py	1.24 KB
cc_scripts_user.py	1.25 KB
cc_scripts_vendor.py	1.26 KB
cc_seed_random.py	3 KB
cc_set_hostname.py	3.4 KB
cc_set_passwords.py	10.32 KB
cc_snap.py	3.36 KB
cc_spacewalk.py	2.64 KB
cc_ssh.py	11 KB
cc_ssh_authkey_fingerprints.py	3.76 KB
cc_ssh_import_id.py	5.43 KB
cc_timezone.py	1.14 KB
cc_ubuntu_autoinstall.py	2.99 KB
cc_ubuntu_drivers.py	4.08 KB
cc_ubuntu_pro.py	13.65 KB
cc_update_etc_hosts.py	2.46 KB
cc_update_hostname.py	2.18 KB
cc_users_groups.py	2.86 KB
cc_wireguard.py	6.67 KB
cc_write_files.py	6 KB
cc_write_files_deferred.py	1.28 KB
cc_yum_add_repo.py	4.53 KB
cc_zypper_add_repo.py	4.93 KB
modules.py	13.23 KB
schema.py	50.26 KB
schemas	-

Edit: cc_package_update_upgrade_install.py

# Copyright (C) 2012 Yahoo! Inc.
#
# Author: Joshua Harlow <harlowja@yahoo-inc.com>
#
# This file is part of cloud-init. See LICENSE file for license information.

"""Package Update Upgrade Install: update, upgrade, and install packages"""

import logging
import os
import time

from cloudinit import subp, util
from cloudinit.cloud import Cloud
from cloudinit.config import Config
from cloudinit.config.schema import MetaSchema
from cloudinit.distros import ALL_DISTROS
from cloudinit.log.loggers import flush_loggers
from cloudinit.settings import PER_INSTANCE

REBOOT_FILES = ("/var/run/reboot-required", "/run/reboot-needed")
REBOOT_CMD = ["/sbin/reboot"]

meta: MetaSchema = {
    "id": "cc_package_update_upgrade_install",
    "distros": [ALL_DISTROS],
    "frequency": PER_INSTANCE,
    "activate_by_schema_keys": [
        "apt_update",
        "package_update",
        "apt_upgrade",
        "package_upgrade",
        "packages",
    ],
}

LOG = logging.getLogger(__name__)

def _multi_cfg_bool_get(cfg, *keys):
    for k in keys:
        if util.get_cfg_option_bool(cfg, k, False):
            return True
    return False

def _fire_reboot(
    wait_attempts: int = 6, initial_sleep: int = 1, backoff: int = 2
):
    """Run a reboot command and panic if it doesn't happen fast enough."""
    subp.subp(REBOOT_CMD)
    start = time.monotonic()
    wait_time = initial_sleep
    for _i in range(wait_attempts):
        time.sleep(wait_time)
        wait_time *= backoff
        elapsed = time.monotonic() - start
        LOG.debug("Rebooted, but still running after %s seconds", int(elapsed))
    # If we got here, not good
    elapsed = time.monotonic() - start
    raise RuntimeError(
        "Reboot did not happen after %s seconds!" % (int(elapsed))
    )

def handle(name: str, cfg: Config, cloud: Cloud, args: list) -> None:
    # Handle the old style + new config names
    update = _multi_cfg_bool_get(cfg, "apt_update", "package_update")
    upgrade = _multi_cfg_bool_get(cfg, "package_upgrade", "apt_upgrade")
    reboot_if_required = _multi_cfg_bool_get(
        cfg, "apt_reboot_if_required", "package_reboot_if_required"
    )
    pkglist = util.get_cfg_option_list(cfg, "packages", [])

errors = []
    if update or upgrade:
        try:
            cloud.distro.update_package_sources()
        except Exception as e:
            util.logexc(LOG, "Package update failed")
            errors.append(e)

if upgrade:
        try:
            cloud.distro.package_command("upgrade")
        except Exception as e:
            util.logexc(LOG, "Package upgrade failed")
            errors.append(e)

if len(pkglist):
        try:
            cloud.distro.install_packages(pkglist)
        except Exception as e:
            util.logexc(
                LOG, "Failure when attempting to install packages: %s", pkglist
            )
            errors.append(e)

# TODO(smoser): handle this less violently
    # kernel and openssl (possibly some other packages)
    # write a file /var/run/reboot-required after upgrading.
    # if that file exists and configured, then just stop right now and reboot
    for reboot_marker in REBOOT_FILES:
        reboot_fn_exists = os.path.isfile(reboot_marker)
        if reboot_fn_exists:
            break
    if (upgrade or pkglist) and reboot_if_required and reboot_fn_exists:
        try:
            LOG.warning(
                "Rebooting after upgrade or install per %s", reboot_marker
            )
            # Flush the above warning + anything else out...
            flush_loggers(LOG)
            _fire_reboot()
        except Exception as e:
            util.logexc(LOG, "Requested reboot did not happen!")
            errors.append(e)

if len(errors):
        LOG.warning(
            "%s failed with exceptions, re-raising the last one", len(errors)
        )
        raise errors[-1]