�
܍~c� � �z � U d Z ddlZddlmZ ddlmZ ddlmZ ddlZdZ n# e
$ r dZdZ Y nw xY wddl m Z dd l
mZ dd
l
mZmZmZmZ dd lmZ dd lmZmZ dd
lmZ e j e� � Z dgZ ddd ed� � e ed� � gedgd�Z ee d< ee � � Z dZ dZ!dZ"defd�Z#d e$d ed ed e d e%d!df d"�Z&dS )#z<Ubuntu Drivers: Interact with third party drivers in Ubuntu.� N)�dedent)�Cloud)�DistroTF)�Logger)�log)�subp�
temp_utils�
type_utils�util)�Config)�
MetaSchema� get_meta_doc)� PER_INSTANCE�ubuntu�cc_ubuntu_driverszUbuntu Driversz,Interact with third party drivers in Ubuntu.zo This module interacts with the 'ubuntu-drivers' command to install
third party driver packages.zN drivers:
nvidia:
license-accepted: true
�drivers)�id�name�title� description�distros�examples� 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 �� � t | t � � s't dt j | � � �d| ��� � �d}t j t j | |� � � � }|s t � d||� � d S t j d� � s&t � d� � |dg� � d}t j | d � � }|r|d
� |� � z
}t � d |||r|nd � � t j |� � � d
�� � }t j � |d� � } t j |t( � � t+ j d� � 5 } | � t0 |� � d d d � � n # 1 swxY w Y n;# t2 $ r.}
t j t dt7 |
� � � � � d }
~
ww xY wt j � |� � rt j |� � n9# t j � |� � rt j |� � w w xY w t j
ddd|g� � d S # t j $ rR} t> | j v rt �! d� � n#d| j" v rt �! d� � � d } ~ ww xY w)Nz''drivers' config expected dict, found 'z': znvidia/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�excs �D/usr/lib/python3/dist-packages/cloudinit/config/cc_ubuntu_drivers.py�install_driversrN O s< � � �c�4� � �
��i��"�3�'�'�'�'���
.�
�
�
�
(�G� �
��!5�c�7�!C�!C�
D�
D�F� � � � � �8�'�6�J�J�J��� �:�&� '� '� 4� � � �
/�
�
�
� ��1�2�3�3�3��J��&�s�,<�=�=�K�� 0��e�l�l�;�/�/�/�
��I�I�F���"�0� � �� � � � � �&�":�":�"<�"<�� M� M� M�D��7�<�<��&7�8�8�L� � �� �&<�=�=�=�
�
(� �
6�
6� 9�"� �J�J�)�<� 8� 8� 8� 9� 9� 9� 9� 9� 9� 9� 9� 9� 9� 9���� 9� 9� 9� 9��� � � � � � � �A�3�q�6�6�
�
�
� ����� ���� �7�=�=�� � � � �L�� � � ��� �7�=�=�� � � � �L�� � � � � ����
� � �#�Y� �:�F�G�G�G�G�G�� � %� � � � +�s�z� 9� 9� �K�K�D�
�
�
�
� 2�S�Z�
?�
?� �K�K�J� K� K� K�
��������sg �.F= � F1�% F= �1F5�5F= �8F5�9F= �<H, �=
G5�)G0�0G5�5H, �,6I"�&J � K!�A
K � K!r rB �cloudr �args�returnc �� � d|vr|� d| � � d S t s|� d| � � d S t |d |j j |j � � d S )Nr z4Skipping module named %s, no 'drivers' key in configz<Skipping module named %s, 'python3-debconf' is not installed)r+ � HAS_DEBCONFr@ rN r �install_packages)r rB rO r rP s rM �handlerU � s� � � ���� � � �H�$�O�O�O��� � � � � � J� �
�
�
� ��� �I��� �5�u�|�� � � � � )'�__doc__r0 �textwrapr �cloudinit.cloudr �cloudinit.distrosr r5 rS � ImportError�loggingr � cloudinitr r r r
r �cloudinit.configr �cloudinit.config.schemar
r �cloudinit.settingsr � getLogger�__name__r* r r �__annotations__r> r4 r rN r: �listrU � rV rM �<module>rf s' �� C� B� B� � � � � � � � � � � !� !� !� !� !� !� $� $� $� $� $� $���N�N�N��K�K��� � � ��G��K�K�K����� � � � � � � $� $� $� $� $� $� 8� 8� 8� 8� 8� 8� 8� 8� 8� 8� 8� 8� #� #� #� #� #� #� <� <� <� <� <� <� <� <� +� +� +� +� +� +��g���!�!�� �*�� � �
;��6� (�� �
���
�
�
�� � )�{�)� ��j� � � �. �,�t�
�
�� K� !�� � *� �=�6� =� =� =� =�@�
����#(��/5��=A�� �� � � � � s �" � .�. |