File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/modules/__pycache__/swarm.cpython-310.pyc
o
�N�g�5 � @ s� d Z ddlZdd� Zz
ddlZee�ZW n ey dZY nw dZdd� Zd d
� Z d d � Z
e e e
fd
d�Ze e e fdd�Ze
fdd�Ze e e e e e e fdd�Ze fdd�Ze fdd�Ze fdd�Ze e
fdd �Ze e e e e fd d �ZdS ) a�
Docker Swarm Module using Docker's Python SDK
=============================================
:codeauthor: Tyler Jones <jonestyler806@gmail.com>
.. versionadded:: 2018.3.0
The Docker Swarm Module is used to manage and create Docker Swarms.
Dependencies
------------
- Docker installed on the host
- Docker python sdk >= 2.5.1
Docker Python SDK
-----------------
.. code-block:: bash
pip install -U docker
More information: https://docker-py.readthedocs.io/en/stable/
� Nc s ddg}t � fdd�|D ��S )N� APIClient�from_envc 3 s � | ]}t � |�V qd S )N)�hasattr)�.0�attr��mod� �F/opt/saltstack/salt/lib/python3.10/site-packages/salt/modules/swarm.py� <genexpr> s � z$_is_docker_module.<locals>.<genexpr>)�all)r Zrequired_attrsr r r
�_is_docker_module s r
F�swarmc C s t rtS dS )zC
Load this module if the docker python module is installed
)FzGThe swarm module failed to load: Docker python module is not available.)�
HAS_DOCKER�__virtualname__r r r r
� __virtual__- s r c C s t rt�� td< td td< d S )N�client�id� server_name)r �dockerr � __context__Z
__grains__)�selfr r r
�__init__9 s r c C s t jdd�} | �� }|d S )z�
Get the Docker Swarm Manager or Worker join tokens
CLI Example:
.. code-block:: bash
salt '*' swarm.swarm_tokens
�unix://var/run/docker.sock�Zbase_urlZ
JoinTokens)r r Z
inspect_swarm)r �servicer r r
� swarm_tokens? s
r c
C s� z i }t d j�| ||� d�t d �}|�|t� d�� W |S tjjyW } z*i }d|j v r:|�|j dd�� n|�t
|j �dd�� W Y d }~|S W Y d }~|S d }~w t yi i }|�d
d i� Y |S w ) a�
Initialize Docker on Minion as a Swarm Manager
advertise_addr
The ip of the manager
listen_addr
Listen address used for inter-manager communication,
as well as determining the networking interface used
for the VXLAN Tunnel Endpoint (VTEP).
This can either be an address/port combination in
the form 192.168.1.1:4567,
or an interface followed by a port number,
like eth0:4567
force_new_cluster
Force a new cluster if True is passed
CLI Example:
.. code-block:: bash
salt '*' swarm.swarm_init advertise_addr='192.168.50.10' listen_addr='0.0.0.0' force_new_cluster=False
r zRDocker swarm has been initialized on {} and the worker/manager Join token is belowr )�CommentZTokensz%This node is already part of a swarm.F)r �result)�Errorr Nr z]Please make sure you are passing advertise_addr, listen_addr and force_new_cluster correctly.) r r �init�format�updater r �errorsZAPIErrorZ explanation�str� TypeError)Zadvertise_addr� listen_addrZforce_new_cluster� salt_return�output�errr r r
�
swarm_initN s8 ���
�
� ����r* c C sf z i }t d jj| g||d� t d d }|�|| d�� W |S ty2 i }|�ddi� Y |S w )a�
Join a Swarm Worker to the cluster
remote_addr
The manager node you want to connect to for the swarm
listen_addr
Listen address used for inter-manager communication if the node gets promoted to manager,
as well as determining the networking interface used for the VXLAN Tunnel Endpoint (VTEP)
token
Either the manager join token or the worker join token.
You can get the worker or manager token via ``salt '*' swarm.swarm_tokens``
CLI Example:
.. code-block:: bash
salt '*' swarm.joinswarm remote_addr=192.168.50.10 listen_addr='0.0.0.0' token='SWMTKN-1-64tux2g0701r84ofq93zppcih0pe081akq45owe9ts61f30x4t-06trjugdu7x2z47j938s54il'
r )Z remote_addrsr& Z
join_tokenr z has joined the Swarm)r Z Manager_Addrr zuPlease make sure this minion is not part of a swarm and you are passing remote_addr, listen_addr and token correctly.)r r �joinr" r% )Z remote_addrr& �tokenr' r( r r r
� joinswarm� s"
� ����r- c C s4 i }t d jj| d� t d d }|�d|i� |S )z�
Force the minion to leave the swarm
force
Will force the minion/worker/manager to leave the swarm
CLI Example:
.. code-block:: bash
salt '*' swarm.leave_swarm force=False
r ��forcer z has left the swarmr )r r Zleaver" )r/ r' r( r r r
� leave_swarm� s
r0 c C s� z<i }t jjd|d�}t jj||id�} td jj|| ||| d� td d | }
|�|
td || |||||d� � W |S tyO i }|�d d
i� Y |S w ) a
Create Docker Swarm Service Create
image
The docker image
name
Is the service name
command
The docker command to run in the container at launch
hostname
The hostname of the containers
replicas
How many replicas you want running in the swarm
target_port
The target port on the container
published_port
port that's published on the host/os
CLI Example:
.. code-block:: bash
salt '*' swarm.service_create image=httpd name=Test_Service command=None hostname=salthttpd replicas=6 target_port=80 published_port=80
Z
replicated)�replicas)�portsr )�name�image�command�modeZ
endpoint_specr z* has a Docker Swarm Service running named ) ZInfoZMinion�Name�ImageZCommand�Hostname�ReplicasZ Target_PortZPublished_Portr zPlease make sure you are passing arguments correctly [image, name, command, hostname, replicas, target_port and published_port]) r �typesZ ServiceMode� EndpointSpecr Zservices�creater" r% ) r4 r3 r5 �hostnamer1 � target_port�published_portr' Z replica_moder2 Zechobackr r r
�service_create� sN (
������ ����rA c C sL z�i }t jdd�}|j| d�}tjj�|�}tjj�|�}|d d }|d d }|d d d }|d d d
} |d }
|d } |d
} |d d }
|d d d d }|d d }|d d d d }| D ](}|d }|d }|d }|d }|�|||
||| | |||||t d |d�� qgW |S t
y� i }|�d d i� Y |S w ) z�
Swarm Service Information
service_name
The name of the service that you want information on about the service
CLI Example:
.. code-block:: bash
salt '*' swarm.swarm_service_info service_name=Test_Service
r r )r �Version�Index�Specr7 r< ZModeZPorts�IDZ CreatedAtZ UpdatedAtZLabelsZ
Replicatedr: ZEndpointZ
VirtualIPsZ TaskTemplateZ
ContainerSpecr8 Z
PublishedPortZ
TargetPortZ PublishMode�Protocolr )z Service Namer: z
Service IDZNetworkz Network Modez
Creation Datez Update DatezPublished Portz Target PortzPublished ModerF z Docker Imagez Minion IdrB r z service_name arg is missing?) r r Zinspect_service�salt�utils�json�dumps�loadsr" r r% )Z service_namer' r r �getdata�dump�versionr3 Z network_moder2 Zswarm_idZ create_dateZ update_date�labelsr1 �networkr4 �itemsr@ r? Zpublished_mode�protocolr r r
�swarm_service_info s\
��� ��rS c C s\ zi }t jdd�}|�| �} |�| td d�� W |S ty- i }|�ddi� Y |S w )z�
Remove Swarm Service
service
The name of the service
CLI Example:
.. code-block:: bash
salt '*' swarm.remove_service service=Test_Service
r r r )zService Deletedz Minion IDr zservice arg is missing?)r r �remove_servicer" r r% )r r' r r r r
rT O s
� ��rT c C s� zbi }t jdd�}|jd| id�}tjj�|�}tjj�|�}|D ]>}|d d d }|d d }|d d } |d
}
|d d } |d d
} |d }
|d d }|�||| |
| | |
|d�� q!W |S t yu i }|�ddi� Y |S w )z�
Displays Information about Swarm Nodes with passing in the server
server
The minion/server name
CLI Example:
.. code-block:: bash
salt '*' swarm.node_ls server=minion1
r r r3 )�filtersZ DescriptionZEngineZ
EngineVersion�Platformr9 rE rD �Role� Availability�StatusrB rC )zDocker VersionrV r9 rE ZRolesrX rY rB r z>The server arg is missing or you not targeting a Manager node?)
r r ZnodesrG rH rI rJ rK r" r% )Zserverr' r r rL rM rQ Zdocker_version�platformZ hostnamesZids�role� availability�statusrN r r r
�node_lsi sF
��� ���r^ c C sf t jdd�}z|dkr|j| dd�}|W S |j| dd�}|W S ty2 i }|�ddi� | Y S w ) aU
Remove a node from a swarm and the target needs to be a swarm manager
node_id
The node id from the return of swarm.node_ls
force
Forcefully remove the node/minion from the service
CLI Example:
.. code-block:: bash
salt '*' swarm.remove_node node_id=z4gjbe9rwmqahc2a91snvolm5 force=false
r r �TrueTr. Fr z/Is the node_id and/or force=True/False missing?)r r � remove_noder% r" )�node_idr/ r r r' r r r
r` � s �r` c C sh t jdd�}zi }| ||d�}|j|||d� |�d|i� W |S ty3 i }|�ddi� Y |S w )a�
Updates docker swarm nodes/needs to target a manager node/minion
availability
Drain or Active
node_name
minion/node
role
role of manager or worker
node_id
The Id and that can be obtained via swarm.node_ls
version
Is obtained by swarm.node_ls
CLI Example:
.. code-block:: bash
salt '*' swarm.update_node availability=drain node_name=minion2 role=worker node_id=3k9x7t8m4pel9c0nqr3iajnzp version=19
r r )rX r7 rW )ra rN � node_speczNode Informationr zOMake sure all args are passed [availability, node_name, role, node_id, version])r r � update_noder" r% )r\ Z node_namer[ ra rN r r' rb r r r
rc � s ����rc )�__doc__Zsalt.utils.jsonrG r
r r � ImportErrorr r r r r$ �int�boolr* r- r0 rA rS rT r^ r` rc r r r r
�<module> s: � 6 *
� R ; 0