U
    ]=                     @   s8   d dl mZ d dlmZ ddlmZ G dd deZdS )    )	APIClient)APIError   )Modelc                       s   e Zd ZdZdZ fddZedd Zdd Ze	jje_dddZ
dd Ze	jje_dd Ze	jje_dd Zdd Ze	jje_dddZ  ZS )Swarmzu
    The server's Swarm state. This a singleton that must be reloaded to get
    the current state of the Swarm.
    ZIDc              
      sZ   t t| j|| | jrVz|   W n0 tk
rT } z|jjdkrD W 5 d }~X Y nX d S )N)i  i  )superr   __init__clientreloadr   ZresponseZstatus_code)selfargskwargse	__class__ 5/usr/lib/python3/dist-packages/docker/models/swarm.pyr      s    zSwarm.__init__c                 C   s   | j ddS )z
        The version number of the swarm. If this is not the same as the
        server, the :py:meth:`update` function will not work and you will
        need to call :py:meth:`reload` before calling it again.
        ZVersionZIndex)attrsgetr   r   r   r   version   s    zSwarm.versionc                 C   s   | j j S N)r	   apiget_unlock_keyr   r   r   r   r       s    zSwarm.get_unlock_keyN0.0.0.0:2377Fc           
      K   sB   ||||||d}| j jjf ||d< | j jjf |}	|   |	S )a:  
        Initialize a new swarm on this Engine.

        Args:
            advertise_addr (str): Externally reachable address advertised to
                other nodes. 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``. If the port number is omitted,
                the port number from the listen address is used.

                If not specified, it will be automatically detected when
                possible.
            listen_addr (str): 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``.
                If the port number is omitted, the default swarm listening port
                is used. Default: ``0.0.0.0:2377``
            force_new_cluster (bool): Force creating a new Swarm, even if
                already part of one. Default: False
            default_addr_pool (list of str): Default Address Pool specifies
                default subnet pools for global scope networks. Each pool
                should be specified as a CIDR block, like '10.0.0.0/8'.
                Default: None
            subnet_size (int): SubnetSize specifies the subnet size of the
                networks created from the default subnet pool. Default: None
            data_path_addr (string): Address or interface to use for data path
                traffic. For example, 192.168.1.1, or an interface, like eth0.
            task_history_retention_limit (int): Maximum number of tasks
                history stored.
            snapshot_interval (int): Number of logs entries between snapshot.
            keep_old_snapshots (int): Number of snapshots to keep beyond the
                current snapshot.
            log_entries_for_slow_followers (int): Number of log entries to
                keep around to sync up slow followers after a snapshot is
                created.
            heartbeat_tick (int): Amount of ticks (in seconds) between each
                heartbeat.
            election_tick (int): Amount of ticks (in seconds) needed without a
                leader to trigger a new election.
            dispatcher_heartbeat_period (int):  The delay for an agent to send
                a heartbeat to the dispatcher.
            node_cert_expiry (int): Automatic expiry for nodes certificates.
            external_ca (dict): Configuration for forwarding signing requests
                to an external certificate authority. Use
                ``docker.types.SwarmExternalCA``.
            name (string): Swarm's name
            labels (dict): User-defined key/value metadata.
            signing_ca_cert (str): The desired signing CA certificate for all
                swarm node TLS leaf certificates, in PEM format.
            signing_ca_key (str): The desired signing CA key for all swarm
                node TLS leaf certificates, in PEM format.
            ca_force_rotate (int): An integer whose purpose is to force swarm
                to generate a new signing CA certificate and key, if none have
                been specified.
            autolock_managers (boolean): If set, generate a key and use it to
                lock data stored on the managers.
            log_driver (DriverConfig): The default log driver to use for tasks
                created in the orchestrator.

        Returns:
            (str): The ID of the created node.

        Raises:
            :py:class:`docker.errors.APIError`
                If the server returns an error.

        Example:

            >>> client.swarm.init(
                advertise_addr='eth0', listen_addr='0.0.0.0:5000',
                force_new_cluster=False, default_addr_pool=['10.20.0.0/16],
                subnet_size=24, snapshot_interval=5000,
                log_entries_for_slow_followers=1200
            )

        )advertise_addrlisten_addrforce_new_clusterdefault_addr_poolsubnet_sizedata_path_addr
swarm_spec)r	   r   create_swarm_specZ
init_swarmr
   )
r   r   r   r   r   r   r    r   Zinit_kwargsZnode_idr   r   r   init$   s    Rz
Swarm.initc                 O   s   | j jj||S r   )r	   r   
join_swarmr   r   r   r   r   r   join   s    z
Swarm.joinc                 O   s   | j jj||S r   )r	   r   leave_swarmr%   r   r   r   leave   s    zSwarm.leavec                 C   s   | j j | _dS )z
        Inspect the swarm on the server and store the response in
        :py:attr:`attrs`.

        Raises:
            :py:class:`docker.errors.APIError`
                If the server returns an error.
        N)r	   r   Zinspect_swarmr   r   r   r   r   r
      s    	zSwarm.reloadc                 C   s   | j j|S r   )r	   r   unlock_swarm)r   keyr   r   r   unlock   s    zSwarm.unlockc                 K   s<   | ddkrd|d< | jjj| j| jjjf ||||dS )a  
        Update the swarm's configuration.

        It takes the same arguments as :py:meth:`init`, except
        ``advertise_addr``, ``listen_addr``, and ``force_new_cluster``. In
        addition, it takes these arguments:

        Args:
            rotate_worker_token (bool): Rotate the worker join token. Default:
                ``False``.
            rotate_manager_token (bool): Rotate the manager join token.
                Default: ``False``.
            rotate_manager_unlock_key (bool): Rotate the manager unlock key.
                Default: ``False``.
        Raises:
            :py:class:`docker.errors.APIError`
                If the server returns an error.

        Znode_cert_expiryNl     +  )r   r!   rotate_worker_tokenrotate_manager_tokenrotate_manager_unlock_key)r   r	   r   Zupdate_swarmr   r"   )r   r,   r-   r.   r   r   r   r   update   s    zSwarm.update)Nr   FNNN)FFF)__name__
__module____qualname____doc__Zid_attributer   propertyr   r   r   r#   r&   r$   r(   r'   r
   r+   r)   r/   __classcell__r   r   r   r   r      s.   


       
^


  r   N)Z
docker.apir   Zdocker.errorsr   resourcer   r   r   r   r   r   <module>   s   