o w7e @sUdZddlZddlZddlmZddlmZmZddlm Z ddl m Z ddl m Z mZddlmZdd lmZd Zd d gZd ddeeeedggdZe ed<eeZeeZddZdefddZdede de deddf ddZde defdd Z dS)!z5Grub Dpkg: Configure grub debconf installation deviceN)dedent)subputil)Cloud)Config) MetaSchema get_meta_doc) PER_INSTANCE)ProcessExecutionErroraConfigure which device is used as the target for grub installation. This module can be enabled/disabled using the ``enabled`` config key in the ``grub_dpkg`` config dict. This module automatically selects a disk using ``grub-probe`` if no installation device is specified. The value which is placed into the debconf database is in the format which the grub postinstall script expects. Normally, this is a /dev/disk/by-id/ value, but we do fallback to the plain disk name if a by-id name is not present. If this module is executed inside a container, then the debconf database is seeded with empty values, and install_devices_empty is set to true. ubuntudebian cc_grub_dpkgz Grub Dpkgz*Configure grub debconf installation deviceaE grub_dpkg: enabled: true # BIOS mode (install_devices needs disk) grub-pc/install_devices: /dev/sda grub-pc/install_devices_empty: false # EFI mode (install_devices needs partition) grub-efi/install_devices: /dev/sda )idnametitle descriptiondistros frequencyexamplesactivate_by_schema_keysmetac Csbd}g}d}d}trd}d}ztjdd||gdd j}Wn8tyI}zt|jtr3t d n d |j vr>t d nWYd }~nd }~wt yWt tdYnw|r`tj|sbdSztjdddd|gdd j}Wnt yt td|Ynwt dd|dd|D}t dd||rt|dn|}t d||S)z Fetches the /dev/disk/by-id device grub is installed to. Falls back to plain disk name if no by-id entry is present. diskz/bootdevicez /boot/efiz grub-probez-tT)capturez'grub-probe' not found in $PATHzfailed to get canonical pathz)grub-probe 'failed to get canonical path'Nz*grub-probe failed to execute for grub_dpkgudevadminfoz--rootz--query=symlinkz3udevadm DEVLINKS symlink query failed for disk='%s'z%considering these device symlinks: %s,cSsg|]}d|vr|qS)z disk/by-id).0devrr?/usr/lib/python3/dist-packages/cloudinit/config/cc_grub_dpkg.py zszfetch_idevs..z)filtered to these disk/by-id symlinks: %srz selected %s) is_efi_bootedrstdoutstripr isinstancereasonFileNotFoundErrorLOGdebugstderr Exceptionrlogexcospathexistssplitjoinsorted)rdevices probe_target probe_mounteidevsrrr! fetch_idevs?sX         r9returnc CsBztjdWSty }z td|WYd}~dSd}~ww)z4 Check if the system is booted in EFI mode. z/sys/firmware/efiz7Failed to determine if system is booted in EFI mode: %sNF)r.r/r0OSErrorr)error)r7rrr!r#s r#rcfgcloudargsc Cs|d|di}|si}|dd}t|r"td||dSt|}td|z tdg|WdStyP}zttd|WYd}~dSd}~ww) N grub_dpkgz grub-dpkgenabledTz*%s disabled by config grub_dpkg/enabled=%sz-Setting grub debconf-set-selections with '%s'zdebconf-set-selectionsz6Failed to run debconf-set-selections for grub_dpkg: %s) getris_falser)r*get_debconf_configrr,r-)rr=r>r?mycfgrA dconf_selr7rrr!handles"   rGrEcCstrt|dd}|durt}d|St|dd}|dur#t}|d}|dur0| }n t|ts:t|}t| }d||fS)zd Returns the debconf config for grub-pc or grub-efi depending on the systems boot mode. zgrub-efi/install_devicesNz+grub-pc grub-efi/install_devices string %s zgrub-pc/install_deviceszgrub-pc/install_devices_emptyz[grub-pc grub-pc/install_devices string %s grub-pc grub-pc/install_devices_empty boolean %s ) r#rget_cfg_option_strr9rBr&booltranslate_boolstrlower)rEr8 idevs_emptyrrr!rDs(    rD)!__doc__loggingr.textwrapr cloudinitrrcloudinit.cloudrcloudinit.configrcloudinit.config.schemarrcloudinit.settingsr cloudinit.subpr MODULE_DESCRIPTIONrr__annotations__ getLogger__name__r)r9rIr#rKlistrGrDrrrr!s<        D