Google
Edit File: DataSourceWSL.cpython-39.pyc
a '�Dg|9 � @ s� d Z ddlZddlZddlZddlmZ ddlmZmZmZ ddl Z ddl m Z m Z m Z ddlmZ ddlmZ ddlmZ e�e�Zd Zd Zd Zd Zed �dd�Z ee d �dd�Z ed �dd�Z ed �dd�Z eee d�dd�Z ee d �dd�Z!G dd � d �Z"ee ee#d �d d �Z$eeee" ee" f d�d d!�Z%ee" ee" ee d"�d#d$�Z&G d%d&� d&e j'�Z(e(e j)ffgZ*d'd(� Z+dS ))z?Datasource to support the Windows Subsystem for Linux platform.� N)�PurePath)�List�Optional�Tuple)�sources�subp�util)�Distro)�type_from_starts_with)�Pathsz/usr/bin/wslpathziid-datasource-wsl� %s.user-dataz agent.yaml��returnc C s" t � tddg�\} }t| �� �jS )zL Returns the name of the current WSL instance as seen from outside. z-am�/)r � WSLPATH_CMDr �rstrip�name)Z root_net_path�_� r �C/usr/lib/python3.9/site-packages/cloudinit/sources/DataSourceWSL.py� instance_name s r c C sH d} d}g }t �� �� D ]*}|d | kr||d v r|�|d � q|S )z� Return a list of mount points of the Windows drives inside the current WSL instance, if drives are mounted, or an empty list otherwise Z9pz aname=drvfsZfstypeZoptsZ mountpoint)r �mounts�values�append)ZFS_TYPEZOPTIONS_CONTAINZmounted�mntr r r �mounted_win_drives+ s r c C sd t � } | std��d}| D ]2}|| }t�|tj�s6qt�d|� t|� S tdd�| � ��dS )z? Returns the Linux path to the Windows host's cmd.exe. z Windows drives are not mounted.z%s/Windows/System32/cmd.exezFound cmd.exe at <%s>z,Couldn't find cmd.exe in any mount point: %sz, N) r �IOError�os�access�X_OK�LOG�debugr �join)r � candidater �cmdr r r �cmd_executable; s �r% c C sV t � } t�d| �� ddg�\}}|�� }|s6t�d��t�td|g�\}}t|�� �S )z� Finds the user's home directory path as a WSL path. raises: IOError when no mountpoint with cmd.exe is found ProcessExecutionError when either cmd.exe is unable to retrieve the user's home directory z/initz/Czecho %USERPROFILE%z4No output from cmd.exe to show the user profile dir.z-au)r% r �as_posixr ZProcessExecutionErrorr r )r$ �homer �outr r r � find_homeS s � r) )� user_homer c C s2 t j�| d�}t j�|�s*t�d|� dS t|�S )z� Returns the Windows user profile .cloud-init directory translated as a Linux path accessible inside the current WSL instance, or None if not found. z .cloud-initz*cloud-init user data dir %s doesn't exist.N)r �pathr" �isdirr r! r )r* �seed_dirr r r �cloud_init_data_dirq s r. c C s6 t �� \}}}|r|n|}d| d||f d| dgS )z Return a list of candidate file names that may contain user-data in some supported format, ordered by precedence. r z%s-%s.user-dataz%s-all.user-datazdefault.user-data)r Zget_linux_distro)r Zdistribution_idZ version_idZversion_codename�versionr r r � candidate_user_data_file_names s �r0 c @ s, e Zd ZdZed�dd�Zed�dd�ZdS ) � ConfigDataz~Models a piece of configuration data as a dict if possible, while retaining its raw representation alongside its file path)r+ c C s8 t �|�| _|| _d | _dt| j�kr4t �| j�| _d S )Nztext/cloud-config)r �load_text_file�rawr+ � config_dictr � load_yaml)�selfr+ r r r �__init__� s zConfigData.__init__r c C s | j d uS �N)r4 )r6 r r r �is_cloud_config� s zConfigData.is_cloud_configN)�__name__� __module__� __qualname__�__doc__r r7 �boolr9 r r r r r1 � s r1 )� cloudinitdirr r c C s� dt i}| du r|S tj�| �� d| �}zt�t�|��}W n ty\ t � d|� Y n0 |rjd|vr�d|� d|� �}t � |� t |��|S )z` Returns the relevant metadata loaded from cloudinit dir based on the instance name � instance-idNz %s.meta-dataz<No instance metadata found at %s. Using default instance-id.z Metadata at z5 does not contain instance-id key. Instead received: ) �DEFAULT_INSTANCE_IDr r+ r" r& r r5 r2 �FileNotFoundErrorr r! �error� ValueError)r? r �metadataZ metadata_path�msgr r r �load_instance_metadata� s* � � �� rG c C s� t j�| d�}t j�|�s dS tt j�|tt� ��}d}t j�|�r`t� d|t | �� t |�}tt j�|t ��}d}t j�|�r�t |�}||fS )zg Read .ubuntupro user-data if present and return a tuple of agent and landscape user-data. z.ubuntupro/.cloud-init)NNNzYLandscape configuration found: %s. Organization policy ignores any local user-data in %s.) r r+ r" r, r �LANDSCAPE_DATA_FILEr �isfiler r! r. r1 �AGENT_DATA_FILE)r* Zpro_dirZlandscape_pathZlandscape_dataZ agent_path� agent_datar r r �load_ubuntu_pro_data� s& � � rL )rK � user_datar c C s� | du st | j�dkr6|du s,t |j�dkr0dS |jS |du sLt |j�dkrl| du sbt | j�dkrfdS | jS t| �� |�� g�s�t�d� d| j�� |j�� f S i }d}g }t|j t �r�|j }|� di �� di �� dd�}t| j t ��rv|r�t�d � | j }|D ]&}||v �r|� |� || ||< �q|�r@t�d � |�� |�rv|� di �� d��rvt�d |� ||d d d< d t�|��� S ) a Merge agent.yaml data provided by Ubuntu Pro for WSL and user data provided either by Landscape or the local user, according to the UP4W specific rules. When merging is not possible, provide #include directive to allow cloud-init to merge separate parts. Nr zaUnable to merge {agent_data.path} and {user_data.path}. Providing as separate user-data #include.z#include %s %s � Z landscapeZclient�tagsz.Merging both user_data and agent.yaml configs.z% agent.yaml overrides config keys: , zFLandscape client conf updated with user-data landscape.client.tags: %sz@#cloud-config # WSL datasouce Merged agent.yaml and user_data %s)�lenr3 �allr9 r r! r+ r& � isinstancer4 �dict�getr r" �yaml�dump�strip)rK rM ZmergedZ user_tagsZoverridden_keysZagent�keyr r r �merge_agent_landscape_data� s^ �� � ��� ��rY c sV e Zd ZdZdeed�� fdd� Zeed�dd�Ze d �d d �Z e d �d d �Z � Z S )� DataSourceWSLZWSLN)�distro�pathsc s t � �||||� d| _d S )NrN )�superr7 r )r6 �sys_cfgr[ r\ Zud_proc�� __class__r r r7 = s zDataSourceWSL.__init__)r- r c C sn dd� t �|�D �}|s$td| ��dd� t| j�D �}|D ] }||�� v r<t|| � S q<td| ��dS )z� Finds the most precendent of the candidate files that may contain user-data, if any, or None otherwise. c S s i | ]}|j �� |j�qS r )r �casefoldr+ )�.0Zefr r r � <dictcomp>M s z5DataSourceWSL.find_user_data_file.<locals>.<dictcomp>z%s directory is emptyc S s g | ] }|� � �qS r )ra )rb �fr r r � <listcomp>S s �z5DataSourceWSL.find_user_data_file.<locals>.<listcomp>z6%s doesn't contain any of the expected user-data filesN)r �scandirr r0 r �keysr )r6 r- Zexisting_filesZ folded_names�filenamer r r �find_user_data_fileA s � � �z!DataSourceWSL.find_user_data_filer c C sv | � � }|sdS z&tt� �}t|t� �}||�d�kW S ttfyp } z t� dt |�� W Y d }~dS d }~0 0 d S )NFr@ z2Unable to check_instance_id from metadata file: %s) Zget_instance_idr. r) rG r rT r rD r Zwarning�str)r6 r^ �currentZdata_dirrE �errr r r �check_instance_id_ s �z DataSourceWSL.check_instance_idc C sl t �t�st�dt� dS t� | _z t� }W n2 ty^ } zt�d|� W Y d }~dS d }~0 0 t|�}d }d }zt || j�| _ W n: t tfy� } z t� dt |�� W Y d }~dS d }~0 0 | jjdkr�t|�\}}z&|d u � r�|d u� r�t| �|��}W nL t tf�yH } z.|�r tjntj }|d|t |�� W Y d }~n d }~0 0 t||g��s\dS t||�| _dS )Nz5No WSL command %s found. Cannot detect WSL datasourceFz#Unable to detect WSL datasource: %szUnable to load metadata: %sZubuntuz+Unable to load any user-data file in %s: %sT)r �whichr r r! r r) r r. rG rE rD rC rj r[ r rL r1 ri �info�anyrY Z userdata_raw)r6 r* �er- rK rM rl �logr r r � _get_datar sL � � � zDataSourceWSL._get_data)N) r: r; r< Zdsnamer r r7 r ri r> rm rs � __classcell__r r r_ r rZ : s rZ c C s t �| t�S r8 )r Zlist_from_depends� datasources)Zdependsr r r �get_datasource_list� s rv ),r= Zloggingr �typing�pathlibr r r r rU Z cloudinitr r r Zcloudinit.distrosr Zcloudinit.handlersr Zcloudinit.helpersr Z getLoggerr: r r rA rH rJ rj r r r% r) r. r0 r1 rS rG rL rY Z DataSourcerZ ZDEP_FILESYSTEMru rv r r r r �<module> sF � "� ! � Uo �