o

x[hN�
@s�UdZd
dl
Z
d
dlZd
dlmZ
d
dlmZ
zd
dlZdZWn
e	y,


dZdZYn
wd
dl
mZmZm
Z
mZ
d
dlmZ
d
d	lmZ
d
d
lmZ
e
�e�
Zddg
ed
g
d�Zeed<dZdZdZdefdd�Zdededededdf
dd�Z dS)z<Ubuntu Drivers: Interact with third party drivers in Ubuntu.�N)
�Cloud)
�DistroTF)�subp�
temp_utils�
type_utils�util)
�Config)
�
MetaSchema)
�PER_INSTANCE�cc_ubuntu_drivers�ubuntu�drivers)�id�distros�	frequency�activate_by_schema_keys�metazDubuntu-drivers: error: argument <command>: invalid choice: 'install'z�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�distroc
Cs�
t|t
�std
t�|�
|f�
�
d}t�t�||��
}|s&t�	d||�
dSt
�d�
s5t�	d�

|
dg
�

d}t�|d�}|rF|d	�|�
7}t�	d
|||rO|n
d�
t
j|��dd
�}tj�|d�}zGz#t�|t�
t�d�
�}	|	�t|�
Wd�
n1s�w



Y
Wnty�
}

z
t�tdt|
�
�
�d}
~
w
wWtj�|�
r�t�|�

n
tj�|�
r�t�|�

wwzt
�
ddd|g�

WdSt
jy�
}
zt|j vr�t�!d�

�d|j"vr�t�!d�

�d}~w
w)Nz.'drivers' config expected dict, found '%s': %sznvidia/license-acceptedz$Not installing NVIDIA drivers. %s=%szubuntu-driverszI'ubuntu-drivers' command not available.  Installing ubuntu-drivers-commonzubuntu-drivers-common�nvidiaznvidia/versionz:{}z<Installing and activating NVIDIA drivers (%s=%s, version=%s)�latestT)�dir�	needs_exeznvidia.templatez
cloud-initz.Failed to register NVIDIA debconf template: %s�installz--gpgpuz[the available version of ubuntu-drivers is too old to perform requested driver installationz"No drivers found for installation.z0ubuntu-drivers found no drivers for installation)#�
isinstance�dict�	TypeErrorr�obj_namer�translate_bool�get_cfg_by_path�LOG�debugr�which�formatr�mkdtemp�get_tmp_exec_path�os�path�join�
write_file�NVIDIA_DEBCONF_CONTENT�debconf�DebconfCommunicator�commandr�	Exception�logexc�str�isdir�del_dir�ProcessExecutionError� OLD_UBUNTU_DRIVERS_STDERR_NEEDLE�stderr�warning�stdout)�cfg�pkg_install_funcr�cfgpath�nv_acc�
driver_arg�version_cfg�tdir�debconf_file�dc�
e�exc�rC�D/usr/lib/python3/dist-packages/cloudinit/config/cc_ubuntu_drivers.py�install_drivers9sz




��





�








�




��


����

��
�





�
�


��rE�namer8�cloud�args�returncCsFd
|
v
rt�
d|�
dStst�d|�
dSt|
d
|jj|j�
dS)Nr
z4Skipping module named %s, no 'drivers' key in configz<Skipping module named %s, 'python3-debconf' is not installed)r r!�HAS_DEBCONFr6rEr�install_packages)rFr8rGrHrCrCrD�handleys






�
�rL)!�__doc__�loggingr&�cloudinit.cloudr�cloudinit.distrosrr+rJ�ImportError�	cloudinitrrrr�cloudinit.configr�cloudinit.config.schemar	�cloudinit.settingsr
�	getLogger�__name__r r�__annotations__r4r*rrEr0�listrLrCrCrCrD�<module>
s8





�






��

"@