File: //lib/python3/dist-packages/awscli/customizations/eks/__pycache__/update_kubeconfig.cpython-310.pyc
o
.&�aS/ � @ s� d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlm Z d dl
m Z m Z m
Z
mZmZmZ d dlmZ e�e�Ze j�d�Zd ZG d
d � d e�ZG d d
� d
e�Z G dd� de�Z dS )� N)� OrderedDict)� BasicCommand)� uni_print)�EKSClusterError)�
Kubeconfig�KubeconfigError�KubeconfigLoader�KubeconfigWriter�KubeconfigValidator�KubeconfigAppender)�ordered_yaml_dumpz~/.kube/configz%client.authentication.k8s.io/v1alpha1c @ st e Zd ZdZe�ddd�Zdddd�dd d
d�d d d
d�d
dd
dd
d�ddd
dd�ddd
d�gZdd� Zdd� Z dS )�UpdateKubeconfigCommandzupdate-kubeconfig�eksz_description.rst�namez�The name of the cluster for which to create a kubeconfig entry. This cluster must exist in your account and in the specified or configured default Region for your AWS CLI installation.T)r � help_text�required�
kubeconfiga Optionally specify a kubeconfig file to append with your configuration. By default, the configuration is written to the first file path in the KUBECONFIG environment variable (if it is set) or the default kubeconfig path (.kube/config) in your home directory.Fzrole-arnz�To assume a role for cluster authentication, specify an IAM role ARN with this option. For example, if you created a cluster while assuming an IAM role, then you must also assume that role to connect to the cluster the first time.zdry-run�
store_truezRPrint the merged kubeconfig to stdout instead of writing it to the specified file.)r �action�defaultr r �verbosez_Print more detailed output when writing to the kubeconfig file, including the appended entries.)r r r r �aliaszBAlias for the cluster context name. Defaults to match cluster ARN.c C s* t d� |D ] }t t|�� t d� qdS )z�
Display entries in yaml format
:param entries: a list of OrderedDicts to be printed
:type entries: list
z
Entries:
�
N)r r )�self�entries�entry� r �M/usr/lib/python3/dist-packages/awscli/customizations/eks/update_kubeconfig.py�_display_entriesg s
�z(UpdateKubeconfigCommand._display_entriesc C s� t | j|j|j|�}|�� }|�� }ttj� dd�|j
�}|� |d �}|� |d �}t
� } | �||||j�}
|jrBt|�� � d S t� } | �|� |rXtd�|
d |j�� n td�|
d |j�� |jrp| �|
||g� d S d S )N�
KUBECONFIG� r zUpdated context {0} in {1}
z Added new context {0} to {1}
)� EKSClient�_sessionr �role_arn�get_cluster_entry�get_user_entry�KubeconfigSelector�os�environ�getr �choose_kubeconfig� has_clusterr �insert_cluster_user_pairr �dry_runr � dump_contentr �write_kubeconfig�format�pathr r ) r � parsed_args�parsed_globals�client�new_cluster_dict�
new_user_dict�config_selector�config�updating_existing�appender�new_context_dict�writerr r r � _run_mains sN � ���
�
�
��z!UpdateKubeconfigCommand._run_mainN)
�__name__�
__module__� __qualname__�NAMEr � FROM_FILE� DESCRIPTION� ARG_TABLEr r= r r r r r
' sJ ��
� � � � ��7 r
c @ s* e Zd Z ddd�Zdd� Zdd� ZdS ) r& Nc s� |du rt � }|� _|du rt|�}|� _|dur"� �|�g� _dS |dkr(t}� fdd�|�tj �D �� _t
� j�dkrCtg� _dS dS )a6
Parse KUBECONFIG into a list of absolute paths.
Also replace the empty list with DEFAULT_PATH
:param env_variable: KUBECONFIG as a long string
:type env_variable: string
:param path_in: The path passed in through the CLI
:type path_in: string or None
Nr c s& g | ]}t |�� �d kr� �|��qS )r )�len�strip� _expand_path)�.0�element�r r r �
<listcomp>� s
�z/KubeconfigSelector.__init__.<locals>.<listcomp>r ) r
�
_validatorr �_loaderrG �_paths� DEFAULT_PATH�splitr' �pathseprE )r � env_variable�path_in� validator�loaderr rJ r �__init__� s
� �zKubeconfigSelector.__init__c C s� | j D ]7}z| j�|�}|�|�r t�d�|�� |W S W q ty: } zt�d�||�� W Y d}~qd}~ww | j�| j d �S )a�
Choose which kubeconfig file to read from.
If name is already an entry in one of the $KUBECONFIG files,
choose that one.
Otherwise choose the first file.
:param cluster_name: The name of the cluster which is going to be added
:type cluster_name: String
:return: a chosen Kubeconfig based on above rules
:rtype: Kubeconfig
z Found entry to update at {0}zPassing {0}:{1}Nr ) rN rM �load_kubeconfigr+ �LOG�debugr0 r �warning)r � cluster_name�candidate_path�
loaded_config�er r r r* � s
�
� ��z$KubeconfigSelector.choose_kubeconfigc C s t j�t j�|��S )z4 A helper to expand a path to a full absolute path. )r' r1 �abspath�
expanduser)r r1 r r r rG � s z KubeconfigSelector._expand_path)NN)r> r? r@ rV r* rG r r r r r& � s
�! r& c @ s. e Zd Zd
dd�Zdd� Zdd� Zdd � ZdS ) r! Nc C s" || _ || _|| _d | _|| _d S �N)r" �
_cluster_name� _role_arn�_cluster_description�_globals)r �sessionr[ r# r3 r r r rV � s
zEKSClient.__init__c C s� | j du rG| jdu r| j�d�}n| jjd| jj| jj| jjd�}|j| jd�}|d | _ d| j vr6t d��| j d dvrGt d �
| j d ���| j S )
z�
Use an eks describe-cluster call to get the cluster description
Cache the response in self._cluster_description.
describe-cluster will only be called once.
Nr )� region_name� endpoint_url�verify)r �cluster�statuszCluster not found)�ACTIVE�UPDATINGzCluster status is {0}) rd re r" �
create_client�regionrh �
verify_ssl�describe_clusterrb r r0 )r r4 �full_descriptionr r r �_get_cluster_description� s$
�
�z"EKSClient._get_cluster_descriptionc C sX | � � �dddi�d }| � � �d�}| � � �d�}tdtd|fd|fg�fd |fg�S )
ze
Return a cluster entry generated using
the previously obtained description.
�certificateAuthority�datar �endpoint�arnrj zcertificate-authority-data�serverr )rs r) r )r � cert_datarv rw r r r r$ s
����zEKSClient.get_cluster_entryc C s� | � � �d��d�d }td| � � �dd�fdtdtdtfd d
|d d d
| jgfdg�fg�fg�}| jdurD|d d d �d| jg� | jj rYtdd| jj fg�g|d d d< |S )zb
Return a user entry generated using
the previously obtained description.
rw �:� r r �user�exec�
apiVersion�argsz--regionr z get-tokenz--cluster-name)�command�awsNz--role)r � AWS_PROFILE�value�env)
rs r) rP r � API_VERSIONrb rc �extendr" �profile)r ro �generated_userr r r r% s< �� ���
�
�zEKSClient.get_user_entryra )r> r? r@ rV rs r$ r% r r r r r! � s
r! ) r' �logging�botocore.compatr � awscli.customizations.commandsr �awscli.customizations.utilsr �$awscli.customizations.eks.exceptionsr �$awscli.customizations.eks.kubeconfigr r r r r
r �&awscli.customizations.eks.ordered_yamlr � getLoggerr> rX r1 r` rO r� r
�objectr& r! r r r r �<module> s
zG