o m[h @sUdZddlZddlZddlmZmZddlmZddlm Z ddl m Z ddl m Z ddlmZd d d ge gd Ze ed <eeZddZdefddZdede dededdf ddZde defddZdS)z5Grub Dpkg: Configure grub debconf installation deviceN)subputil)Cloud)Config) MetaSchema) PER_INSTANCE)ProcessExecutionError cc_grub_dpkgubuntudebian)iddistros frequencyactivate_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_idevssX         r3returnc 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)r*OSErrorr#error)r1rrrrcs rnamecfgcloudargsc Cs|d|di}|si}|dd}t|r"td||dSt|}td|z tjdg|dWdStyQ}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-selections)dataz6Failed to run debconf-set-selections for grub_dpkg: %s) getris_falser#r$get_debconf_configrr&r')r7r8r9r:mycfgr< dconf_selr1rrrhandleos"   rCrAcCstrt|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 ) rrget_cfg_option_strr3r>r booltranslate_boolstrlower)rAr2 idevs_emptyrrrr@s(    r@)__doc__loggingr( cloudinitrrcloudinit.cloudrcloudinit.configrcloudinit.config.schemarcloudinit.settingsrcloudinit.subprr__annotations__ getLogger__name__r#r3rErrGlistrCr@rrrrs&      D