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_ubuntu_drivers.py
# This file is part of cloud-init. See LICENSE file for license information.
"""Ubuntu Drivers: Interact with third party drivers in Ubuntu."""
import logging
import os
from cloudinit.cloud import Cloud
from cloudinit.distros import Distro
try:
import debconf
HAS_DEBCONF = True
except ImportError:
debconf = None
HAS_DEBCONF = False
from cloudinit import subp, temp_utils, type_utils, util
from cloudinit.config import Config
from cloudinit.config.schema import MetaSchema
from cloudinit.settings import PER_INSTANCE
LOG = logging.getLogger(__name__)
meta: MetaSchema = {
"id": "cc_ubuntu_drivers",
"distros": ["ubuntu"],
"frequency": PER_INSTANCE,
"activate_by_schema_keys": ["drivers"],
}
OLD_UBUNTU_DRIVERS_STDERR_NEEDLE = (
"ubuntu-drivers: error: argument <command>: invalid choice: 'install'"
)
# Use a debconf template to configure a global debconf variable
# (linux/nvidia/latelink) setting this to "true" allows the
# 'linux-restricted-modules' deb to accept the NVIDIA EULA and the package
# will automatically link the drivers to the running kernel.
NVIDIA_DEBCONF_CONTENT = """\
Template: linux/nvidia/latelink
Type: boolean
Default: true
Description: Late-link NVIDIA kernel modules?
Enable this to link the NVIDIA kernel modules in cloud-init and
make them available for use.
"""
X_LOADTEMPLATEFILE = "X_LOADTEMPLATEFILE"
def install_drivers(cfg, pkg_install_func, distro: Distro):
if not isinstance(cfg, dict):
raise TypeError(
"'drivers' config expected dict, found '%s': %s"
% (type_utils.obj_name(cfg), cfg)
)
cfgpath = "nvidia/license-accepted"
# Call translate_bool to ensure that we treat string values like "yes" as
# acceptance and _don't_ treat string values like "nah" as acceptance
# because they're True-ish
nv_acc = util.translate_bool(util.get_cfg_by_path(cfg, cfgpath))
if not nv_acc:
LOG.debug("Not installing NVIDIA drivers. %s=%s", cfgpath, nv_acc)
return
if not subp.which("ubuntu-drivers"):
LOG.debug(
"'ubuntu-drivers' command not available. "
"Installing ubuntu-drivers-common"
)
pkg_install_func(["ubuntu-drivers-common"])
driver_arg = "nvidia"
version_cfg = util.get_cfg_by_path(cfg, "nvidia/version")
if version_cfg:
driver_arg += ":{}".format(version_cfg)
LOG.debug(
"Installing and activating NVIDIA drivers (%s=%s, version=%s)",
cfgpath,
nv_acc,
version_cfg if version_cfg else "latest",
)
# Register and set debconf selection linux/nvidia/latelink = true
tdir = temp_utils.mkdtemp(dir=distro.get_tmp_exec_path(), needs_exe=True)
debconf_file = os.path.join(tdir, "nvidia.template")
try:
util.write_file(debconf_file, NVIDIA_DEBCONF_CONTENT)
with debconf.DebconfCommunicator("cloud-init") as dc:
dc.command(X_LOADTEMPLATEFILE, debconf_file)
except Exception as e:
util.logexc(
LOG, "Failed to register NVIDIA debconf template: %s", str(e)
)
raise
finally:
if os.path.isdir(tdir):
util.del_dir(tdir)
try:
subp.subp(["ubuntu-drivers", "install", "--gpgpu", driver_arg])
except subp.ProcessExecutionError as exc:
if OLD_UBUNTU_DRIVERS_STDERR_NEEDLE in exc.stderr:
LOG.warning(
"the available version of ubuntu-drivers is"
" too old to perform requested driver installation"
)
elif "No drivers found for installation." in exc.stdout:
LOG.warning("ubuntu-drivers found no drivers for installation")
raise
def handle(name: str, cfg: Config, cloud: Cloud, args: list) -> None:
if "drivers" not in cfg:
LOG.debug("Skipping module named %s, no 'drivers' key in config", name)
return
if not HAS_DEBCONF:
LOG.warning(
"Skipping module named %s, 'python3-debconf' is not installed",
name,
)
return
install_drivers(
cfg["drivers"], cloud.distro.install_packages, cloud.distro
)