File: //proc/1233/root/usr/lib/python3/dist-packages/docker/context/__pycache__/context.cpython-310.pyc
o
�2.a� � @ sz d dl Z d dlZd dlmZmZ d dlmZ d dlmZ d dl m
Z
d dl m Z d dl m Z d dl m
Z
G d d
� d
�ZdS ) � N)�copyfile�rmtree)� TLSConfig)�ContextException)� get_meta_dir)�
get_meta_file)� get_tls_dir)�get_context_hostc @ s� e Zd ZdZ d+dd�Z d,dd�Zd d
� Zed d � �Zed
d� �Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zed d � �Zed d � �Zed!d"� �Zed#d$� �Zed%d&� �Zed'd(� �Zed)d*� �ZdS )-�Contextz
A context.NFc C s� |st d��|| _d | _|| _i | _i | _d| _d| _|s5|r#|dkr%dn|}|t||�| d�i| _d S |� � D ]7\}}t
|t �sJt d�
||���|| j|< |dkrTq9|�dt||��| j| d< t|�d| ��| j| d< q9d S ) NzName not providedz IN MEMORY�swarm�docker��Host�
SkipTLSVerifyz>Unknown endpoint format for
context {}: {}r r )� Exception�name� context_type� orchestrator� endpoints�tls_cfg� meta_path�tls_pathr �items�
isinstance�dictr �format�get�bool) �selfr r �hostr �tls�default_endpoint�k�v� r$ �8/usr/lib/python3/dist-packages/docker/context/context.py�__init__ sJ �����
�
���zContext.__init__r c C s@ t || �|d�| j|< |r|| j| d< |r || j|< d S d S )Nr
�DefaultNamespace)r r r )r r r r �skip_tls_verify�
def_namespacer$ r$ r% � set_endpoint9 s
��zContext.set_endpointc C s | � � S �N)�__call__�r r$ r$ r% �inspectF s zContext.inspectc C s` t �|�}|r.| |d |d �dd �|�dd �d�}|d �dd �|_|�� t|�|_|S d S )N�Name�Metadata�StackOrchestrator� Endpoints)r r �Type)r
�
_load_metar r � _load_certsr r )�clsr �meta�instancer$ r$ r% � load_contextI s
�
zContext.load_contextc
C s� t |�}tj�|�s d S i }zt|��
}t�|�}W d � n1 s#w Y W n ttt fy@ } zt
d� ||���d }~ww |d � � D ]&\}}|dkrPqG|�
dtd d��|d | d< t|�
dd��|d | d< qG|S )Nz@Detected corrupted meta file for
context {} : {}r2 r r Fr T)r �os�path�isfile�open�json�load�OSError�KeyError�
ValueErrorr r r r r r )r6 r � meta_file�metadata�f�er" r# r$ r$ r% r4 W s2
�����
�
�zContext._load_metac C s i }t | j�}| j�� D ]m}tj�tj�||��sq d }d }d }t�tj�||��D ]+}|� d�r:tj�|||�}q*|� d�rHtj�|||�}q*|� d�rUtj�|||�}q*t
|||g�ryd }|dkrn| jd � dd�snd}t ||f||d�||< q || _
|| _d S ) N�ca�cert�keyr r FT)� client_cert�ca_cert�verify)r r r �keysr: r; �isdir�join�listdir�
startswith�allr r r r ) r �certs�tls_dir�endpointrK rH rI �filenamerL r$ r$ r% r5 r s8
��
��
zContext._load_certsc
C s: t | j�}tj�|�st�|� tt| j�d��}|�t �
| j �� W d � n1 s,w Y t | j�}| j
�� D ]S\}}tj�tj�||��sTt�tj�||�� |j}|rht|tj�||tj�|��� |jr�|j\}}t|tj�||tj�|��� t|tj�||tj�|��� q;t | j�| _t | j�| _d S )N�w)r r r: r; rN �makedirsr= r �writer> �dumpsr0 r r r rO rK r �basenamerH r r ) r �meta_dirrE rT rU r �ca_file� cert_file�key_filer$ r$ r% �save� s4
�
�
�
�� z Context.savec C s8 t j�| j�r t| j� t j�| j�rt| j� d S d S r+ )r: r; rN r r r r- r$ r$ r% �remove� s
�zContext.removec C s d| j j� d| j� d�S )N�<z: 'z'>)� __class__�__name__r r- r$ r$ r% �__repr__� s zContext.__repr__c C s t j| �� dd�S )N� )�indent)r> rZ r, r- r$ r$ r% �__str__� s zContext.__str__c C s" | j }|�| j� |�| j� |S r+ )r0 �update� TLSMaterial�Storage)r �resultr$ r$ r% r, � s zContext.__call__c C s
| j d u S r+ )r r- r$ r$ r% �is_docker_host� s
zContext.is_docker_hostc C � | j S r+ )r r- r$ r$ r% r/ � � z Context.Namec C sF | j r| j dkr| j�dd �}|r|�dd �S d S | j| j �dd �S )Nr r r )r r r )r rU r$ r$ r% r � s z Context.Hostc C rn r+ )r r- r$ r$ r% � Orchestrator� ro zContext.Orchestratorc C s$ i }| j r
d| j i}| j|| jd�S )Nr1 )r/ r0 r2 )r r r )r r7 r$ r$ r% r0 � s
�zContext.Metadatac C s2 | j }|r |dkr d}|| j�� v r| j| S d S )Nr r )r r rM )r rI r$ r$ r% r � s
zContext.TLSConfigc C sH i }| j �� D ]\}}|j\}}tttjj|j||g��||< qd|iS )Nrj ) r r rH �list�mapr: r; r[ rK )r rS rU r rH rI r$ r$ r% rj � s
��zContext.TLSMaterialc C s d| j | jd�iS )Nrk )� MetadataPath�TLSPath)r r r- r$ r$ r% rk � s
��zContext.Storage)NNNF)r NNFN)rd �
__module__� __qualname__�__doc__r& r* r. � classmethodr9 r4 r5 r` ra re rh r, rm �propertyr/ r rp r0 r rj rk r$ r$ r$ r% r
sD
�+
�
r
)r: r> �shutilr r �
docker.tlsr �
docker.errorsr �docker.context.configr r r r r
r$ r$ r$ r% �<module> s