File: //lib/python3.6/site-packages/awscli/customizations/eks/__pycache__/kubeconfig.cpython-36.pyc
3
�T_�$ � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlm Z m
Z
G dd� de�Z G dd� de �Z G d d
� d
e �Z
d d � ZG d
d� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZdS )� N)� OrderedDict)�EKSError)�ordered_yaml_load�ordered_yaml_dumpc @ s e Zd ZdZdS )�KubeconfigErrorz& Base class for all kubeconfig errors.N)�__name__�
__module__� __qualname__�__doc__� r r � /usr/lib/python3.6/kubeconfig.pyr s r c @ s e Zd ZdZdS )�KubeconfigCorruptedErrorz+ Raised when a kubeconfig cannot be parsed.N)r r r r
r r r r r
s r
c @ s e Zd ZdZdS )�KubeconfigInaccessableErrorz< Raised when a kubeconfig cannot be opened for read/writing.N)r r r r
r r r r r ! s r c C s( t d dg fdg fd d
d t � fd
g fg�S )N�
apiVersion�v1�clusters�contexts�current-context� �kind�ConfigZ preferences�users)r r )r r )r r )r r r r r �_get_new_kubeconfig_content% s r c @ s& e Zd Zddd�Zdd� Zdd� ZdS ) �
KubeconfigNc C s || _ |d krt� }|| _d S )N)�pathr �content)�selfr r r r r �__init__2 s zKubeconfig.__init__c C s
t | j�S )z+ Return the stored content in yaml format. )r r )r r r r � dump_content8 s zKubeconfig.dump_contentc C s& d| j krdS |dd� | j d D �kS )zg
Return true if this kubeconfig contains an entry
For the passed cluster name.
r Fc S s g | ] }|d �qS )�namer )�.0�clusterr r r �
<listcomp>C s z*Kubeconfig.has_cluster.<locals>.<listcomp>)r )r r r r r � has_cluster< s
zKubeconfig.has_cluster)N)r r r r r r# r r r r r 1 s
r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�KubeconfigValidatorc C s t d d �j| _d S )N)r r �_validation_content)r r r r r H s z KubeconfigValidator.__init__c C s* t |t�std��| j|� | j|� dS )z�
Raises KubeconfigCorruptedError if the passed content is invalid
:param config: The config to validate
:type config: Kubeconfig
z(Internal error: Not a Kubeconfig object.N)�
isinstancer r
�_validate_config_types�_validate_list_entry_types)r �configr r r �validate_configM s
z#KubeconfigValidator.validate_configc C s~ t |jt�std��xd| jj� D ]V\}}||jkr |j| dk r t |j| t|�� r tdj|t|j| �t|����q W dS )z�
Raises KubeconfigCorruptedError if any of the entries in config
are the wrong type
:param config: The config to validate
:type config: Kubeconfig
zContent not a dictionary.Nz%{0} is wrong type:{1} (Should be {2}))r& r �dictr
r% �items�type�format)r r) �key�valuer r r r' Z s
z*KubeconfigValidator._validate_config_typesc C sd x^| j j� D ]P\}}||jkr t|j| �tkr x*|j| D ] }t|t�s<tdj|���q<W q W dS )z�
Raises KubeconfigCorruptedError if any lists in config contain objects
which are not dictionaries
:param config: The config to validate
:type config: Kubeconfig
z Entry in {0} not a dictionary.N) r% r, r r- �listr& r r
r. )r r) r/ r0 �elementr r r r( q s
z.KubeconfigValidator._validate_list_entry_typesN)r r r r r* r' r( r r r r r$ G s
r$ c @ s e Zd Zddd�Zdd� ZdS )�KubeconfigLoaderNc C s |d krt � }|| _d S )N)r$ �
_validator)r Z validatorr r r r � s zKubeconfigLoader.__init__c C s� y"t |d��}t|�}W dQ R X W nr tk
rd } z&|jtjkrFd}ntdj|���W Y dd}~X n2 tjk
r� } zt dj|���W Y dd}~X nX t
||�}| j j |� |S )aP
Loads the kubeconfig found at the given path.
If no file is found at the given path,
Generate a new kubeconfig to write back.
If the kubeconfig is valid, loads the content from it.
If the kubeconfig is invalid, throw the relevant exception.
:param path: The path to load a kubeconfig from
:type path: string
:raises KubeconfigInaccessableError: if the kubeconfig can't be opened
:raises KubeconfigCorruptedError: if the kubeconfig is invalid
:return: The loaded kubeconfig
:rtype: Kubeconfig
�rNz&Can't open kubeconfig for reading: {0}z'YamlError while loading kubeconfig: {0})
�openr �IOError�errno�ENOENTr r. �yamlZ YAMLErrorr
r r4 r* )r r �streamZloaded_content�eZ
loaded_configr r r �load_kubeconfig� s
z KubeconfigLoader.load_kubeconfig)N)r r r r r= r r r r r3 � s
r3 c @ s e Zd Zdd� ZdS )�KubeconfigWriterc C s� t jj|j�}yt j|� W n< tk
rX } z |jtjkrHtdj|���W Y dd}~X nX y(t |jd��}t
|j |� W dQ R X W n0 t k
r� } ztdj|���W Y dd}~X nX dS )a
Write config to disk.
OK if the file doesn't exist.
:param config: The kubeconfig to write
:type config: Kubeconfig
:raises KubeconfigInaccessableError: if the kubeconfig
can't be opened for writing
z'Can't create directory for writing: {0}Nzw+z&Can't open kubeconfig for writing: {0})
�osr �dirname�makedirs�OSErrorr8 ZEEXISTr r. r6 r r r7 )r r) Z directoryr< r; r r r �write_kubeconfig� s z!KubeconfigWriter.write_kubeconfigN)r r r rC r r r r r> � s r> c @ s( e Zd Zdd� Zddd�Zd dd�ZdS )
�KubeconfigAppenderc C s� ||j krg |j |< |j | }t|t�s>tdj|t|�t���d}x@t|�D ]4\}}d|krLd|krL|d |d krL|||< d}qLW |s�|j|� ||j |< |S )z�
Insert entry into the array at content[key]
Overwrite an existing entry if they share the same name
:param config: The kubeconfig to insert an entry into
:type config: Kubeconfig
z1Tried to insert into {0},which is a {1} not a {2}Fr T)r r& r1 r r. r- � enumerate�append)r r) r/ �entryZarray�foundZcounterZexisting_entryr r r � insert_entry� s&
z KubeconfigAppender.insert_entryNc C s4 t dt d|d fd|d fg�fd|p,|d fg�S )zE Generate a context to associate cluster and user with a given alias.�contextr! r �user)r )r r! rK �aliasr r r �
_make_context� s
z KubeconfigAppender._make_contextc C sL | j |||d�}| j|d|� | j|d|� | j|d|� |d |jd< |S )ag
Insert the passed cluster entry and user entry,
then make a context to associate them
and set current-context to be the new context.
Returns the new context
:param config: the Kubeconfig to insert the pair into
:type config: Kubeconfig
:param cluster: the cluster entry
:type cluster: OrderedDict
:param user: the user entry
:type user: OrderedDict
:param alias: the alias for the context; defaults top user entry name
:type context: str
:return: The generated context
:rtype: OrderedDict
)rL r r r r zcurrent-context)rM rI r )r r) r! rK rL rJ r r r �insert_cluster_user_pair� s z+KubeconfigAppender.insert_cluster_user_pair)N)N)r r r rI rM rN r r r r rD � s
rD )r? r: Zloggingr8 Zbotocore.compatr Z$awscli.customizations.eks.exceptionsr Z&awscli.customizations.eks.ordered_yamlr r r r
r r �objectr r$ r3 r> rD r r r r �<module> s ;*