3
\�me� � @ s� U d Z ddlZddlZddlmZ ddlmZmZ ddlm Z ddl
m Z ddl m
Z
mZ ddlmZ dd lmZ d
Zd d gZd
ddeeeed�gg d�Ze
ee�Z eje�Zdd� Zed�dd�Ze e e e dd�dd�Z e e d�dd �Z dS ) z5Grub Dpkg: Configure grub debconf installation device� N)�dedent)�subp�util)�Cloud)�Config)�
MetaSchema� get_meta_doc)� PER_INSTANCE)�ProcessExecutionErrora� Configure 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.
ZubuntuZdebianZ 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
)�id�name�title� description�distrosZ frequencyZexamplesZactivate_by_schema_keysc C sl d} g }d}d}t � r d}d}y tjdd||gdd �jj� } W np tk
r� } z6t|jt�rhtj d
� nd |j
kr~tj d � n� W Y d
d
}~X n t k
r� t j
td� Y nX | s�tjj| � r�dS y&tjdddd| gdd �jj� j� }W n$ t k
�r t j
td| � Y nX tj ddj|�� dd� |D �}tj ddj|�� |�rXt|�d n| }tj 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/bootZdevicez /boot/efiz
grub-probez-tT)Zcapturez 'grub-probe' not found in $PATHz failed to get canonical pathz)grub-probe 'failed to get canonical path'Nz*grub-probe failed to execute for grub_dpkgZudevadm�infoz--rootz--query=symlinkz3udevadm DEVLINKS symlink query failed for disk='%s'z%considering these device symlinks: %s�,c S s g | ]}d |kr|�qS )z
disk/by-id� )�.0Zdevr r �"/usr/lib/python3.6/cc_grub_dpkg.py�
<listcomp>z s z fetch_idevs.<locals>.<listcomp>z)filtered to these disk/by-id symlinks: %sr z selected %s)�
is_efi_bootedr �stdout�stripr
�
isinstance�reason�FileNotFoundError�LOG�debug�stderr� Exceptionr �logexc�os�path�exists�split�join�sorted)r ZdevicesZ probe_targetZ probe_mount�e�idevsr r r � fetch_idevs? sB
r+ )�returnc C s> y t jjd�S tk
r8 } ztjd| � dS d} ~ X nX dS )z4
Check if the system is booted in EFI mode.
z/sys/firmware/efiz7Failed to determine if system is booted in EFI mode: %sFN)r# r$ r% �OSErrorr �error)r) r r r r � s
r )r �cfg�cloud�argsr, c C s� |j d|j di ��}|s i }|j dd�}tj|�rDtjd| |� d S t|�}tjd|� ytjdg|d� W n0 tk
r� } ztjtd |� W Y d d }~X nX d S )
NZ grub_dpkgz grub-dpkg�enabledTz*%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) �getr Zis_falser r �get_debconf_configr r! r" )r r/ r0 r1 �mycfgr2 Z dconf_selr) r r r �handle� s
r7 )r6 r, c C s� t � r*tj| dd�}|dkr"t� }d| S tj| dd�}|dkrFt� }| jd�}|dkr`| }nt|t�sttj|�}t|�j � }d||f S dS )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_devicesz grub-pc/install_devices_emptyz[grub-pc grub-pc/install_devices string %s
grub-pc grub-pc/install_devices_empty boolean %s
)
r r Zget_cfg_option_strr+ r4 r �boolZtranslate_bool�str�lower)r6 r* Z idevs_emptyr r r r5 � s"
r5 ) �__doc__Zloggingr# �textwrapr Z cloudinitr r Zcloudinit.cloudr Zcloudinit.configr Zcloudinit.config.schemar r Zcloudinit.settingsr Zcloudinit.subpr
ZMODULE_DESCRIPTIONr �metaZ getLogger�__name__r r+ r8 r r9 �listr7 r5 r r r r �<module> s4
D