File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/proxy/__pycache__/junos.cpython-310.pyc
o
�N�gz � @ s d Z ddlZz4dZddlZddlZddlZddlZddlm Z m
Z
m Z m Z m
Z
mZmZmZ ddlmZ ddlmZ W n eyE dZY nw dgZi Ze�e�ZG d d
� d
�ZdZ d d � Z d
d� Z dd� Z dd� Z dd� Z!dd� Z"dd� Z#dd� Z$dd � Z%d d � Z&d d � Z'd!d"� Z(d#d$� Z)dS )%a�
Interface with a Junos device via proxy-minion. To connect to a junos device via junos proxy, specify the host information in the pillar in '/srv/pillar/details.sls'
.. code-block:: yaml
proxy:
proxytype: junos
host: <ip or dns name of host>
username: <username>
port: 830
password: <secret>
In '/srv/pillar/top.sls' map the device details with the proxy name.
.. code-block:: yaml
base:
'vmx':
- details
After storing the device information in the pillar, configure the proxy in '/etc/salt/proxy'
.. code-block:: yaml
master: <ip or hostname of salt-master>
Run the salt proxy via the following command:
.. code-block:: bash
salt-proxy --proxyid=vmx
� NT)�ConnectAuthError�ConnectClosedError� ConnectError�ConnectRefusedError�ConnectTimeoutError�
ProbeError�RpcError�RpcTimeoutError)�TimeoutExpiredError)�IOProcF�junosc @ s e Zd ZdZdZdd� ZdS )� RebootActivez�
Class to get static variable, to indicate when a reboot/shutdown
is being processed and the keep_alive should not probe the
connection since it interferes with the shutdown process.
Fc K s d S �N� )�self�kwargsr r �D/opt/saltstack/salt/lib/python3.10/site-packages/salt/proxy/junos.py�__init__N s zRebootActive.__init__N)�__name__�
__module__� __qualname__�__doc__�reboot_shutdownr r r r r r
E s r
c C s t sdS tS )z6
Only return if all the modules are available
)FzMMissing dependency: The junos proxy minion requires the 'jnpr' Python module.)� HAS_JUNOS�__virtualname__r r r r � __virtual__V s r c C s d| d< t �d� d| d d i}g d�}d| d �� v r(| d �d�| d d< | d �� }|D ]}||v r>| d | ||< q0t �d |� tjjdi |��td
< ztd
�� W n" t t
t t t
fyz } zt �d |� dtd < W Y d
}~d
S d
}~ww d|v r�t| d d �}z|td
_W n ty� } zt �dt|�� W Y d
}~n d
}~ww t �d|� ztd
jtjjjjd� W n ty� } z t �d|� W Y d
}~nd
}~ww ztd
jtjjjjd� W n ty� } z t �d|� W Y d
}~nd
}~ww dtd < d
S )z\
Open the connection to the Junos device, login, and bind to the
Resource class
F�multiprocessingzOpening connection to junos�host�proxy)�user�username�password�passwd�portZ gather_facts�modeZbaudZattemptsZ
auto_probeZssh_private_key_fileZ
ssh_config� normalizeZ huge_treer r zArgs: %s�connz2%s : not able to initiate connection to the device� initializedN�timeoutz"Not able to set timeout due to: %sz RPC timeout set to %d seconds)Zcuz(Bind failed with Config class due to: %s)�swz$Bind failed with SW class due to: %sTr )�log�debug�keys�pop�jnprr ZDevice� thisproxy�openr r r r r �error�intr( � Exception�str�bind�utilsZconfigZConfigr) ZSW)�opts�argsZ
optional_argsZ
proxy_keys�arg�exr( r r r �initd sb
� � �� �� �� �� r; c C s t �dd�S )Nr' F)r/ �getr r r r r' � s r' c C s t d S )Nr& )r/ r r r r r& � s r& c C �
dt _d S )NT�r
r r r r r �
reboot_active� �
r? c C r= )NFr> r r r r � reboot_clear� r@ rA c C s t jS r r> r r r r �get_reboot_active� s rB c C s< t � }t� td _|j}|s td i d�| d d �� |S )zk
Validate and return the connection status with the remote device.
.. versionadded:: 2018.3.0
r& zevent.fire_masterzjunos/proxy/{}/stopr r )r& �pingr/ � connectedZ__salt__�format)r7 �devZlocal_connectedr r r �alive� s �rG c C s� t � } | jdu r
dS t| jd�rI| jjjdur | jjj�� s,| jjjdu rGt| jjt�rG| jjj� � dkr@| jjj
� � r@t | �S t
�d� dS dS t | �S )z
Ping? Pong!
NF�_session� z1skipped ping() call as proxy already getting dataT)r& Z_conn�hasattrrH Z
_transportZ is_active�
isinstancer �_buffer�tellZ_q�empty�_rpc_file_listr* r+ )rF r r r rC � s
� �
rC c
C s� z | j jddd� W dS ttfy+ z| �� W Y dS tttfy* Y Y dS w tyJ } zdt |�v r?W Y d }~dS W Y d }~d S d }~ww )Nz /dev/null� )�pathZ dev_timeoutTFz,'NoneType' object has no attribute 'timeout')
ZrpcZ file_listr r �closer r r
�AttributeErrorr4 )rF r: r r r rO � s � ���rO c C s dS )z(
Returns the name of this proxy
r r r r r r � proxytype s rT c C s` t td j�} d| v rt | d �| d< d| v r.| d D ]}t | d | d �| d | d< q| S )Nr& � version_infoZ
junos_info�object)�dictr/ �facts)rX �rer r r �get_serialized_facts s "rZ c C s8 t �d| d � z td �� W dS ty Y dS w )zz
This is called when the proxy-minion is exiting to make sure the
connection to the device is closed cleanly.
z Proxy module %s shutting down!!�idr& N)r* r+ r/ rR r3 )r7 r r r �shutdown s �r\ )*r �loggingr Z
jnpr.junosr. Zjnpr.junos.utilsZjnpr.junos.utils.configZjnpr.junos.utils.swZjnpr.junos.exceptionr r r r r r r r Zncclient.operations.errorsr
Z+ncclient.transport.third_party.junos.ioprocr � ImportErrorZ__proxyenabled__r/ � getLoggerr r* r
r r r; r' r& r? rA rB rG rC rO rT rZ r\ r r r r �<module> s@ %(
�
D"