File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/runners/__pycache__/f5.cpython-310.pyc
o
�N�g�% � @ s� d Z ddlmZ z
ddlmZ dZW n ey dZY nw dd� ZG dd � d �Z d
d � Z
dd
d�Z dd� Z dd� Z
dd� Zdd� ZdS )a�
Runner to provide F5 Load Balancer functionality
:depends: - pycontrol Python module
:configuration: In order to connect to a F5 Load Balancer, you must specify
in the Salt master configuration the currently available load balancers
.. code-block:: yaml
load_balancers:
bigip1.example.com:
username: admin
password: secret
bigip2.example.com:
username: admin
password: secret
� )�CommandExecutionErrorNTFc C s t sdS dS )NFT)�
HAS_PYCONTROL� r r �C/opt/saltstack/salt/lib/python3.10/site-packages/salt/runners/f5.py� __virtual__ s r c @ sV e Zd Zdd� Zdd� Zdd� Zddd �Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dS )�F5Mgmtc C s || _ || _|| _| �� d S )N)�lb�username�password�_connect)�selfr r r
r r r �__init__% s zF5Mgmt.__init__c C sF zt j| j| j| jdddgd�| _W dS ty" td| j� ���w )z
Connect to F5
TzLocalLB.VirtualServerz LocalLB.Pool)�hostnamer r
ZfromurlZwsdlszUnable to connect to )�f5ZBIGIPr r r
�bigIP� Exception)r r r r r + s
�
��zF5Mgmt._connectc
sT | j jj}|j�d�}||_||_||_|j�d�} � fdd�| D �}
|
r)|
|_nt d��|j�d�} |g| _
|j�d�} |j�d�}
| j |
_ ||
_
|j�d �}|
g|_
|j�d
�}|j�d �}|j|_� |_|j�d �}||_|j�d �}d |_|j�d
�}|||g|_
z|j| dg||gd� W dS ty� } z
td|� d|� ���d}~ww )z)
Create a virtual server
z Common.VirtualServerDefinitionzCommon.ProtocolTypec s. g | ]}|d � d�d � �� kr|d �qS )r �_� ��split�upper��.0�i��protocolr r �
<listcomp>I s . z$F5Mgmt.create_vs.<locals>.<listcomp>zUnknown protocolz Common.VirtualServerSequencez'LocalLB.VirtualServer.VirtualServerTypez+LocalLB.VirtualServer.VirtualServerResourcez3LocalLB.VirtualServer.VirtualServerResourceSequencezLocalLB.ProfileContextTypez*LocalLB.VirtualServer.VirtualServerProfileZ
oneconnectz2LocalLB.VirtualServer.VirtualServerProfileSequencez255.255.255.255)Z definitionsZ wildmasksZ resourcesZprofiles�Unable to create `z` virtual server
NT)r �LocalLB�
VirtualServer� typefactory�create�name�address�portr r �itemZRESOURCE_TYPE_POOL�typeZdefault_pool_nameZPROFILE_CONTEXT_TYPE_ALLZprofile_contextZ profile_namer )r r" �ipr$ r �profile� pool_name�vsZvs_defZcommon_protocols�pZ
vs_def_seqZvs_typeZ vs_resourceZ resource_seqZ
vs_contextZ
vs_profileZvs_profile_httpZvs_profile_connZvs_profile_seq�er r r � create_vs< sd
�� ��� ����zF5Mgmt.create_vs� ROUND_ROBINc
s� | j jjj�d�}� fdd�|D �}|rA| �|�sAz| j jjj|g|gg gd� W d S ty@ } z
td|� d|� ���d}~ww td��)
z7
Create a pool on the F5 load balancer
�LocalLB.LBMethodc s0 g | ]}|d � dd�d � �� kr|d �qS �r r � ���r r ��methodr r r � s *z&F5Mgmt.create_pool.<locals>.<listcomp>)�
pool_namesZ
lb_methods�membersr z` pool
NzUnsupported methodT)r r �Poolr r! �
check_poolr )r r" r4 � lbmethodsZsupported_methodr, r r3 r � create_pool� s
�
����zF5Mgmt.create_poolc
C s� | � |�s t|� d���| jjjj�d�}g |_| jjjj�d�}||_||_ |j�
|� z| jjjj |g|gd� W d S t yU } z
t d|� d|� d|� ���d}~ww )
z&
Add a node to a pool
z pool does not existsz Common.IPPortDefinitionSequencezCommon.IPPortDefinition)r5 r6 zUnable to add `z` to `z`
NT)
r8 r r r r7 r r! �itemsr# r$ �appendZ
add_memberr )r r" r$ r) Z members_seq�memberr, r r r �add_pool_member� s&
�
����zF5Mgmt.add_pool_memberc C �4 | j jj}|�� D ]}|�d�d |kr dS q dS )z/
Check to see if a pool exists
�/r2 TF)r r r7 �get_listr )r r" Zpools�poolr r r r8 � �
�zF5Mgmt.check_poolc C r? )z9
Check to see if a virtual server exists
r@ r2 TF)r r r rA r )r r" r* �vr r r �check_virtualserver� rC zF5Mgmt.check_virtualserverc C s6 | j jjj|gd�d }|D ]
}||jkr dS qdS )z?
Check a pool member exists in a specific pool
)r5 r TF)r r r7 Z
get_memberr# )r r= r) r6 Zmemr r r �check_member_pool� s
�zF5Mgmt.check_member_poolc C s | j jjj�d�}dd� |D �S )z4
List all the load balancer methods
r/ c S s g | ] }|d � dd�d �qS r0 )r )r r4 r r r r � s z$F5Mgmt.lbmethods.<locals>.<listcomp>)r r r7 r r! )r �methodsr r r r9 � s zF5Mgmt.lbmethodsN�r. ) �__name__�
__module__� __qualname__r
r r- r: r> r8 rE rF r9 r r r r r $ s
E
r c
C s^ t d �| d�rtt d | �� �\}}ntd| � d���t| ||�} | �||||||� dS )z�
Create a virtual server
CLI Examples:
.. code-block:: bash
salt-run f5.create_vs lbalancer vs_name 10.0.0.1 80 tcp http poolname
�load_balancersN�Unable to find `�` load balancerT)�__opts__�get�list�valuesr r r- )
r r" r' r$ r r( r) r r
�F5r r r r- � s
r- r. c C sV t d �| d�rtt d | �� �\}}ntd| � d���t| ||�}|�||� dS )z�
Create a pool on the F5 load balancer
CLI Examples:
.. code-block:: bash
salt-run f5.create_pool load_balancer pool_name loadbalance_method
salt-run f5.create_pool load_balancer my_pool ROUND_ROBIN
rL NrM rN T)rO rP rQ rR r r r: )r r" r4 r r
rS r r r r: � s r: c C sX t d �| d�rtt d | �� �\}}ntd| � d���t| ||�}|�|||� dS )z�
Add a node to a pool
CLI Examples:
.. code-block:: bash
salt-run f5.add_pool_member load_balancer 10.0.0.1 80 my_pool
rL NrM rN T)rO rP rQ rR r r r> )r r" r$ r) r r
rS r r r r> � s
r> c C �P t d �| d�rtt d | �� �\}}ntd| � d���t| ||�}|�|�S )z�
Check to see if a pool exists
CLI Examples:
.. code-block:: bash
salt-run f5.check_pool load_balancer pool_name
rL NrM rN )rO rP rQ rR r r r8 �r r" r r
rS r r r r8 �
r8 c C rT )z�
Check to see if a virtual server exists
CLI Examples:
.. code-block:: bash
salt-run f5.check_virtualserver load_balancer virtual_server
rL NrM rN )rO rP rQ rR r r rE rU r r r rE $ rV rE c C sR t d �| d�rtt d | �� �\}}ntd| � d���t| ||�}|�||�S )z�
Check a pool member exists in a specific pool
CLI Examples:
.. code-block:: bash
salt-run f5.check_member_pool load_balancer 10.0.0.1 my_pool
rL NrM rN )rO rP rQ rR r r rF )r r= r) r r
rS r r r rF 6 s
rF rH )�__doc__Zsalt.exceptionsr Zpycontrol.pycontrolZ pycontrolr r � ImportErrorr r r- r: r> r8 rE rF r r r r �<module> s" � 4