Current Dir: /snap/core22/2955/usr/lib/python3/dist-packages/cloudinit/config/

[DIR]  schemas [ delete | rename ]
[DIR]  __pycache__ [ delete | rename ]
[FILE] cc_ansible.py [ edit | delete | rename ]
[FILE] cc_apk_configure.py [ edit | delete | rename ]
[FILE] cc_apt_configure.py [ edit | delete | rename ]
[FILE] cc_apt_pipelining.py [ edit | delete | rename ]
[FILE] cc_bootcmd.py [ edit | delete | rename ]
[FILE] cc_byobu.py [ edit | delete | rename ]
[FILE] cc_ca_certs.py [ edit | delete | rename ]
[FILE] cc_chef.py [ edit | delete | rename ]
[FILE] cc_disable_ec2_metadata.py [ edit | delete | rename ]
[FILE] cc_disk_setup.py [ edit | delete | rename ]
[FILE] cc_fan.py [ edit | delete | rename ]
[FILE] cc_final_message.py [ edit | delete | rename ]
[FILE] cc_growpart.py [ edit | delete | rename ]
[FILE] cc_grub_dpkg.py [ edit | delete | rename ]
[FILE] cc_install_hotplug.py [ edit | delete | rename ]
[FILE] cc_keyboard.py [ edit | delete | rename ]
[FILE] cc_keys_to_console.py [ edit | delete | rename ]
[FILE] cc_landscape.py [ edit | delete | rename ]
[FILE] cc_locale.py [ edit | delete | rename ]
[FILE] cc_lxd.py [ edit | delete | rename ]
[FILE] cc_mcollective.py [ edit | delete | rename ]
[FILE] cc_mounts.py [ edit | delete | rename ]
[FILE] cc_ntp.py [ edit | delete | rename ]
[FILE] cc_package_update_upgrade_install.py [ edit | delete | rename ]
[FILE] cc_phone_home.py [ edit | delete | rename ]
[FILE] cc_power_state_change.py [ edit | delete | rename ]
[FILE] cc_puppet.py [ edit | delete | rename ]
[FILE] cc_raspberry_pi.py [ edit | delete | rename ]
[FILE] cc_reset_rmc.py [ edit | delete | rename ]
[FILE] cc_resizefs.py [ edit | delete | rename ]
[FILE] cc_resolv_conf.py [ edit | delete | rename ]
[FILE] cc_rh_subscription.py [ edit | delete | rename ]
[FILE] cc_rsyslog.py [ edit | delete | rename ]
[FILE] cc_runcmd.py [ edit | delete | rename ]
[FILE] cc_salt_minion.py [ edit | delete | rename ]
[FILE] cc_scripts_per_boot.py [ edit | delete | rename ]
[FILE] cc_scripts_per_instance.py [ edit | delete | rename ]
[FILE] cc_scripts_per_once.py [ edit | delete | rename ]
[FILE] cc_scripts_user.py [ edit | delete | rename ]
[FILE] cc_scripts_vendor.py [ edit | delete | rename ]
[FILE] cc_seed_random.py [ edit | delete | rename ]
[FILE] cc_set_hostname.py [ edit | delete | rename ]
[FILE] cc_set_passwords.py [ edit | delete | rename ]
[FILE] cc_snap.py [ edit | delete | rename ]
[FILE] cc_spacewalk.py [ edit | delete | rename ]
[FILE] cc_ssh.py [ edit | delete | rename ]
[FILE] cc_ssh_authkey_fingerprints.py [ edit | delete | rename ]
[FILE] cc_ssh_import_id.py [ edit | delete | rename ]
[FILE] cc_timezone.py [ edit | delete | rename ]
[FILE] cc_ubuntu_autoinstall.py [ edit | delete | rename ]
[FILE] cc_ubuntu_drivers.py [ edit | delete | rename ]
[FILE] cc_ubuntu_pro.py [ edit | delete | rename ]
[FILE] cc_update_etc_hosts.py [ edit | delete | rename ]
[FILE] cc_update_hostname.py [ edit | delete | rename ]
[FILE] cc_users_groups.py [ edit | delete | rename ]
[FILE] cc_wireguard.py [ edit | delete | rename ]
[FILE] cc_write_files.py [ edit | delete | rename ]
[FILE] cc_write_files_deferred.py [ edit | delete | rename ]
[FILE] cc_yum_add_repo.py [ edit | delete | rename ]
[FILE] cc_zypper_add_repo.py [ edit | delete | rename ]
[FILE] modules.py [ edit | delete | rename ]
[FILE] schema.py [ edit | delete | rename ]
[FILE] __init__.py [ edit | delete | rename ]

Viewing: /snap/core22/2955/usr/lib/python3/dist-packages/cloudinit/config/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.info(
                "***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]

Upload File: