File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/modules/__pycache__/boto_asg.cpython-310.pyc
o
�N�g�� � @ sr d Z ddlZddlZddlZddlZddlZddlZddl Zddl
m m Z ddl
Ze�e�ZdZz>ddlZddlZddlm mZ ddlm mZ ddlZe�d��ej� ddl Z ddl m Z e�d��ej� dZ W n e y{ dZ Y nw d d
� Z!d d � Z"d;d
d�Z#d;dd�Z$ d<dd�Z% d<dd�Z&dd� Z'dd� Z(d=dd�Z)dd � Z*d;d d �Z+d;d d �Z,d;d!d"�Z- d;d#d$�Z. % d>d&d'�Z/d;d(d)�Z0 d;d*d+�Z1d;d,d-�Z2d;d.d/�Z3 0 1 2 d?d3d4�Z4d5d6� Z5 d=d7d8�Z6 d=d9d:�Z7dS )@a�
Connection module for Amazon Autoscale Groups
.. versionadded:: 2014.7.0
:configuration: This module accepts explicit autoscale credentials but can also
utilize IAM roles assigned to the instance through Instance Profiles.
Dynamic credentials are then automatically obtained from AWS API and no
further configuration is necessary. More Information available at:
.. code-block:: text
http://docs.aws.amazon.com/AWSEC2/latest/UserGuide/iam-roles-for-amazon-ec2.html
If IAM roles are not used you need to specify them either in a pillar or
in the minion's config file:
.. code-block:: yaml
asg.keyid: GKTADJGHEIQSXMKKRBJ08H
asg.key: askdjghsdfjkghWupUjasdflkdfklgjsdfjajkghs
A region may also be specified in the configuration:
.. code-block:: yaml
asg.region: us-east-1
If a region is not specified, the default is us-east-1.
It's also possible to specify key, keyid and region via a profile, either
as a passed in dict, or as a string to pull from pillars or minion config:
.. code-block:: yaml
myprofile:
keyid: GKTADJGHEIQSXMKKRBJ08H
key: askdjghsdfjkghWupUjasdflkdfklgjsdfjajkghs
region: us-east-1
:depends: boto
:depends: boto3
� Nz%Y-%m-%dT%H:%M:%SZ�boto)� ClientError�boto3TFc C sF t jj�� } | du r!td tddtd� ttj t dtd d�� | S ) z,
Only load if boto libraries exist.
Tzboto.assign_funcs�asgz
ec2.autoscale)�module�pack�
_get_ec2_connzboto.get_connection_func�ec2)
�salt�utilsZversionsZcheck_boto_reqs� __utils__�__name__�__salt__�setattr�sys�modules)Z
has_boto_reqs� r �I/opt/saltstack/salt/lib/python3.10/site-packages/salt/modules/boto_asg.py� __virtual__Q s �
�r c C s t r
td tddd� d S d S )Nzboto3.assign_funcsZ autoscaling�_get_conn_autoscaling_boto3)Zget_conn_funcname)�HAS_BOTOr r
)�optsr r r �__init__b s
��r c
C s� t ||||d�}d} z|j| gd�}|rW dS d|� �}t�|� W dS tjjyY } z'|rI| jdkrIt�d� t� d � |d
8 }W Y d } ~ q
t�
| � W Y d } ~ dS d } ~ ww ) z�
Check to see if an autoscale group exists.
CLI Example:
.. code-block:: bash
salt myminion boto_asg.exists myasg region=us-east-1
��region�key�keyid�profile� T��namesz-The autoscale group does not exist in region F�
Throttling�.Throttled by AWS API, retrying in 5 seconds...� � N) � _get_conn�get_all_groups�log�debugr � exception�BotoServerError�code�time�sleep�error)
�namer r r r �conn�retriesZ_conn�msg�er r r �existsi s(
��r4 c
C sF t ||||d�}d} z�|j| gd�}|r|d }ni W S t�� }g d�} | D ]V}
|
dkrRg } |jD ]} t�� }
| j|
d< | j|
d < | j|
d
< | �|
� q2| |d< q'|
d krat ||
��
d �||
< q'|
d
krvt ||
�}t dd� |D ��||
< q't ||
�||
< q'|j | d�}g |d< |D ] }|d �t
d|jfd|jfd|jfd|jfd|jfg�� q�|j| d�}i |d< |D ]0}d}|jr�|j�� }t
d|jfd|jfdt|j�fd |j�� fd |fd |jfg�|d |j< q�|W S t j j �y" } z*|�r|j d k�rt �!d � t"�#d!� |d"8 }W Y d}~q
t �$|� i W Y d}~S d}~ww )#z�
Get the configuration for an autoscale group.
CLI Example:
.. code-block:: bash
salt myminion boto_asg.get_config myasg region=us-east-1
r r Tr r )r/ �availability_zones�default_cooldown�desired_capacity�health_check_period�health_check_type�launch_config_name�load_balancers�max_size�min_size�placement_group�vpc_zone_identifier�tags�termination_policies�suspended_processesr@ r �value�propagate_at_launchr? �,rB c s s � | ]}|j V qd S )N)Z process_name)�.0�xr r r � <genexpr>� s � z get_config.<locals>.<genexpr>��as_group�scaling_policiesr/ �adjustment_type�scaling_adjustment�min_adjustment_step�cooldown�scheduled_actionsNr= r< r7 �
start_time�end_time�
recurrencer! r" r# r$ )%r% r& �odict� OrderedDictr@ r rC rD �append�getattr�split�sorted�get_all_policies�dictr/ rL rM rN rO �get_all_scheduled_actionsrR � isoformatr= r< �intr7 rQ rS r r) r* r+ r' r( r, r- r. )r/ r r r r r0 r1 r �retZattrs�attr�_tags�tag�_tagrB �policies�policy�actions�actionrR r3 r r r �
get_config� s�
���
��
��rh c ! C sN t ||||d�}t|t�rtjj�|�}t|t�r tjj�|�}t| t�r,tjj�| �} t| t�r8tjj�| �} g }| r�| D ]F}z|�d�}W n tyW t �
d� Y dS w z|�d�}W n tyo t �
d� Y dS w |�dd�}t j ||| |d�} |�
| � q>t|
t�r�tjj�|
�}
t|t�r�tjj�|�}t|t�r�tjj�|�}d } z9t j| ||||||||| |
|| |
|d �} |�| � t|| |� t|| |� |r�|r�|�| ||� t �d | � W d
S tjj�y& } z2| �r
| jd
k�r
t �d� t�d� | d8 } W Y d} ~ q�t �
| � d| f} t �
| � W Y d} ~ dS d} ~ ww )a
Create an autoscale group.
CLI Example:
.. code-block:: bash
salt myminion boto_asg.create myasg mylc '["us-east-1a", "us-east-1e"]' 1 10 load_balancers='["myelb", "myelb2"]' tags='[{"key": "Name", value="myasg", "propagate_at_launch": True}]'
r r �Tag missing key.FrC �Tag missing value.rD �r rC Z resource_idrD r T)r/ �
launch_configr5 r= r< r7 r; r6 r9 r8 r> r@ r? rA rB zCreated ASG %sr! r" r# r$ NzFailed to create ASG %s)r% �
isinstance�strr
r �json�loads�get�KeyErrorr' r. � autoscale�TagrV �AutoScalingGroupZcreate_auto_scaling_group�_create_scaling_policies�_create_scheduled_actions� put_notification_configuration�infor r) r* r+ r( r, r- )!r/ r: r5 r= r< r7 r; r6 r9 r8 r> r? r@ rA rB rK rP r �notification_arn�notification_typesr r r r0 ra rb rC rD rc r1 �_asgr3 r2 r r r �create� s� "
�
� �
�
�
��r} c & s� t ||||d�}t||||d�}|sdS t|t�r tjj�|�}t|t�r,tjj�|�}t| t�r8tjj�| �} t| t�rDtjj�| �} t|
t�rPtjj�|
�}
t|t�r\tjj�|�}t|t�rhtjj�|�}|jd| gd�gd�� dg �}dd� |D �}g }g � | r�t
d | �} | D ]]}z|� d
�}W n t y� t �
d � d d
|� d�f Y S w z|� d�} W n t y� t �
d� d d
|� d�f Y S w |� dd �} || | | d�} | |vr�|�| � � �| � q�� fdd�|D �} d} z�tj|| ||||||||| |
|| |
d�}!|�r|�r|�|!||� |!�� |�r/t �d|� |�dd� |D �� | �rBt �d| � |�dd� | D �� |!�� |�rN|!�|� t �d | � |j| d �D ] }"|j|"j| d � �qZt || |� |j | d �D ] }#|j |#j| d � �qrt || |� W d S t j!j"�y� }$ z8| �r�|$j#d k�r�t �d!� t$�%d"� | d#8 } W Y d$}$~$q�t �
|$� d%| � �}%t �
|%� d t|$�fW Y d$}$~$S d$}$~$ww )&a
Update an autoscale group.
CLI Example:
.. code-block:: bash
salt myminion boto_asg.update myasg mylc '["us-east-1a", "us-east-1e"]' 1 10 load_balancers='["myelb", "myelb2"]' tags='[{"key": "Name", value="myasg", "propagate_at_launch": True}]'
r )Fzfailed to connect to AWSzauto-scaling-group)�NameZValues)ZFiltersZTagsc S s. g | ]}|d |d |d |� dd�d��qS )ZKeyZValueZ
ResourceIdZPropagateAtLaunchFrk )rq �rF �tr r r �
<listcomp>� s �
��zupdate.<locals>.<listcomp>z
boto3.orderedr ri FzTag z missing keyrC rj z missing valuerD rk c s g | ]}|� vr|�qS r r r �Z desired_tagsr r r� � s r T)�
connectionr/ rl r5 r= r< r7 r; r6 r9 r8 r> r@ r? rA z!Adding/updating tags from ASG: %sc S � g | ]
}t jd i |���qS �r �rs rt r r r r r� � � zDeleting tags from ASG: %sc S r� r� r� r r r r r� � r� zUpdated ASG %srI )Zautoscale_group)T� r! r" r# r$ NzFailed to update ASG )&r% r rm rn r
r ro rp Z
describe_tagsrq r rr r' r. rV rs ru rx �updater( Zcreate_or_update_tags� delete_tagsZresume_processesZsuspend_processesry rZ Z
delete_policyr/ rv r\ Zdelete_scheduled_actionrw r r) r* r+ r, r- )&r/ r: r5 r= r< r7 r; r6 r9 r8 r> r? r@ rA rB rK rP rz r{ r r r r r0 Zconn3Z current_tagsZadd_tagsrb rC rD rc r� r1 r| re Zscheduled_actionr3 r2 r r� r r� b s� #�
���
�
� �
� �
�
��r� c
C sP |r$|D ]!}t jj|d ||d |d |�dd�|d d�}| �|� qdS dS )z*helper function to create scaling policiesr/ rL rM rN NrO )r/ �as_namerL rM rN rO )rs re Z
ScalingPolicyrq Zcreate_scaling_policy)r0 r� rK re r r r rv s
� �rv c C s� |rV|� � D ]Q\}}d|v r t|d t�r tj�|d t�|d< d|v r6t|d t�r6tj�|d t�|d< | j|||�d�|�d�|�d�|�d�|�d�|�d�d� qdS dS ) z5
Helper function to create scheduled actions
rQ rR r7 r= r< rS )r7 r= r< rQ rR rS N)�itemsrm rn �datetime�strptime� DATE_FORMATZ create_scheduled_group_actionrq )r0 r� rP r/ rg r r r rw s, ����rw c
C s� t ||||d�}d} z|�| |� d| � d�}t�|� W dS tjjy^ } z1|rD| jdkrDt�d� t �
d� |d 8 }W Y d
} ~ q
t� | � d | � �}t� |� W Y d
} ~ d S d
} ~ ww )
z�
Delete an autoscale group.
CLI Example:
.. code-block:: bash
salt myminion boto_asg.delete myasg region=us-east-1
r r TzDeleted autoscale group �.r! r" r# r$ Nz!Failed to delete autoscale group F) r% Zdelete_auto_scaling_groupr' ry r r) r* r+ r( r, r- r. )
r/ �forcer r r r r0 r1 r2 r3 r r r �delete. s(
��r� c C s� t | t�r tjj�| �} tjj� � }d| v r.| d �
� D ]\}}tjj � |d�}|�
|� q d| v rJ| d �
� D ]\}}tjj � |d�}|�
|� q8d| v rf| d }tjj � tjjj|dd�d�}|�
|� |�� S )z�
Get a mime multipart encoded string from a cloud-init dict. Currently
supports boothooks, scripts and cloud-config.
CLI Example:
.. code-block:: bash
salt myminion boto.get_cloud_init_mime <cloud init>
Z boothookszcloud-boothook�scriptsz
x-shellscriptz cloud-configF)Zdefault_flow_style)rm rn r
r ro rp �email�mime� multipartZ
MIMEMultipartr� �textZMIMEText�attachZyamlZ safe_dump� as_string)Z
cloud_initZ _cloud_init� script_nameZscript�_scriptZ cloud_configZ
_cloud_configr r r �get_cloud_init_mimeL s&
�
r� c
C s� t ||||d�}d} z|j| gd�}|rW dS d�|�}t�|� W dS tjjyY } z'|rI| jdkrIt�d� t �
d � |d
8 }W Y d } ~ q
t� | � W Y d } ~ dS d } ~ ww ) z�
Check for a launch configuration's existence.
CLI Example:
.. code-block:: bash
salt myminion boto_asg.launch_configuration_exists mylc
r r Tr �4The launch configuration does not exist in region {}Fr! r" r# r$ N� r% � get_all_launch_configurations�formatr' r( r r) r* r+ r, r- r. �
r/ r r r r r0 r1 �lcr2 r3 r r r �launch_configuration_existsl s,
�
��r� c
C s� t | |||d�}d} z|�� W S tjjyF } z(|r5|jdkr5t�d� t� d� |d8 }W Y d}~q
t�
|� g W Y d}~S d}~ww ) z�
Fetch and return all Launch Configuration with details.
CLI Example:
.. code-block:: bash
salt myminion boto_asg.get_all_launch_configurations
r r Tr! r" r# r$ N) r% r� r r) r* r+ r' r( r, r- r. )r r r r r0 r1 r3 r r r r� � s
��r� c C s t | |||�}dd� |D �S )z�
List all Launch Configurations.
CLI Example:
.. code-block:: bash
salt myminion boto_asg.list_launch_configurations
c S � g | ]}|j �qS r �r/ )rF �rr r r r� � � z.list_launch_configurations.<locals>.<listcomp>)r� )r r r r r_ r r r �list_launch_configurations� s
r� c
C s� t ||||d�}d} z|j| gd�}|r|d W S d�|�}t�|� W dS tjjy[ } z'|rK| jdkrKt�d � t �
d
� |d 8 }W Y d} ~ q
t� | � W Y d} ~ dS d} ~ ww ) z�
Dump details of a given launch configuration.
CLI Example:
.. code-block:: bash
salt myminion boto_asg.describe_launch_configuration mylc
r r Tr r r� Nr! r" r# r$ r� r� r r r � describe_launch_configuration� s,
�
��r� �m1.smallc %
C s� t ||||d�}t|t�rtjj�|�}t|
t�r tjj�|
�}
g }|
rPt�� }|
D ]"}|� � D ]\} } t�
� } | � � D ]
\} } t | | | � q<| || < q0q*|g}|rc|sV|rct d |||||||d�}t
jd i d| �d|�d|�d|�d|�d |�d
|�d | �d |�d
| �d| �d|
�d|�d|�d|�d|�d|�d|��}!d}" z|�|!� t�d| � W dS tjjy� }# z1|"r�|#jdkr�t�d� t�d� |"d 8 }"W Y d }#~#q�t�|#� d | � �}$t�|$� W Y d }#~#d S d }#~#ww )!z�
Create a launch configuration.
CLI Example:
.. code-block:: bash
salt myminion boto_asg.create_launch_configuration mylc image_id=ami-0b9c9f62 key_name='mykey' security_groups='["mygroup"]' instance_type='c3.2xlarge'
r z"boto_secgroup.convert_to_group_ids)�vpc_id�vpc_namer r r r r/ �image_id�key_name�security_groups� user_data�
instance_type� kernel_id�
ramdisk_id�block_device_mappings�instance_monitoring�
spot_price�instance_profile_name�
ebs_optimized�associate_public_ip_address� volume_type�delete_on_termination�iops�use_block_device_typesr Tz
Created LC %sr! r" r# r$ NzFailed to create LC Fr )r% rm rn r
r ro rp �blockdevicemappingZBlockDeviceMappingr� ZEBSBlockDeviceTyper r rs ZLaunchConfiguration�create_launch_configurationr' ry r r) r* r+ r( r, r- r. )%r/ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r0 Z_bdmsZ_block_device_mapZblock_device_dictZ block_device�
attributesZ
_block_device� attributerC r� r1 r3 r2 r r r r� � s� #
� �
�������� �
� � �
������
��r� c
C s� t ||||d�}d} z|�| � t�d| � W dS tjjyX } z1|r>|jdkr>t�d� t �
d� |d8 }W Y d }~q
t� |� d
| � �}t� |� W Y d }~d S d }~ww ) z�
Delete a launch configuration.
CLI Example:
.. code-block:: bash
salt myminion boto_asg.delete_launch_configuration mylc
r r Tz
Deleted LC %sr! r" r# r$ NzFailed to delete LC F) r% �delete_launch_configurationr' ry r r) r* r+ r( r, r- r. ) r/ r r r r r0 r1 r3 r2 r r r r� > s&
��r� c
C s� t ||||d�}d}|dkrZ|d8 }z |j| d�}|D ]
} | j|kr(| j W S qt�d| � W dS tjjyU }
z|
j dkrA� t�
d � t � d
� W Y d}
~
nd}
~
ww |dkst�d � dS ) a:
Return the arn for a scaling policy in a specific autoscale group or None
if not found. Mainly used as a helper method for boto_cloudwatch_alarm, for
linking alarms to scaling policies.
CLI Example:
.. code-block:: bash
salt '*' boto_asg.get_scaling_policy_arn mygroup mypolicy
r r r r$ rI zCould not convert: %sNr! z)Throttled by API, will retry in 5 secondsr# z"Maximum number of retries exceeded)
r% rZ r/ Z
policy_arnr' r. r r) r* Z
error_coder( r, r- ) rJ Zscaling_policy_namer r r r r0 r1 rd re r3 r r r �get_scaling_policy_arn[ s,
�
���
r� c
C s� t | |||d�}d} z!d}g }|dur*|j|d�}|dd� |D �7 }|j}|dus|W S tjjyb } z(|rQ| jd krQt�d
� t �
d � |d 8 }W Y d} ~ q
t� | � g W Y d} ~ S d} ~ ww )
a
Return all AutoScale Groups visible in the account
(as a list of boto.ec2.autoscale.group.AutoScalingGroup).
.. versionadded:: 2016.11.0
CLI Example:
.. code-block:: bash
salt-call boto_asg.get_all_groups region=us-east-1 --output yaml
r r Tr� N)�
next_tokenc S s g | ]}|�qS r r �rF �ar r r r� � s z"get_all_groups.<locals>.<listcomp>r! r" r# r$ ) r% r& r� r r) r* r+ r' r( r, r- r. )
r r r r r0 r1 r� �asgsr_ r3 r r r r& ~ s, �
��r& c C s dd� t | |||d�D �S )z�
Return all AutoScale Groups visible in the account
(as a list of names).
.. versionadded:: 2016.11.0
CLI Example:
.. code-block:: bash
salt-call boto_asg.list_groups region=us-east-1
c S r� r r� r� r r r r� � s ��z list_groups.<locals>.<listcomp>r )r& r r r r � list_groups� s �r� � InService�Healthy�private_ip_addressc
sL t ||||d�} t||||d�}
d} z | j| gd�} W n5 tjjyQ }
z'| rA|
jdkrAt�d� t �
d� | d8 } W Y d }
~
qt� |
� W Y d }
~
d
S d }
~
ww t | �dkrft�d | d d
� | D �� d
S | d }g }|j
D ] }|d ur{|j|kr{qo|d ur�|j|kr�qo|�|j� qo|
j|d�}�r��fdd
�|D �S � fdd
�|D �S )z�
return attribute of all instances in the named autoscale group.
CLI Example:
.. code-block:: bash
salt-call boto_asg.get_instances my_autoscale_group_name
r r Tr r! r" r# r$ NFz#name '%s' returns multiple ASGs: %sc S r� r r� )rF r r r r r� � r� z!get_instances.<locals>.<listcomp>r )� instance_idsc s g | ] � � fd d��D ��qS )c s g | ]}t � |��qS r )�_convert_attribute)rF r` ��instancer r r� � s z,get_instances.<locals>.<listcomp>.<listcomp>r )rF )r� r� r r� � s ��c s g | ] }t |� �rt|� ��qS r )rW r� )rF r� )r� r r r� � s ��)r% r r& r r) r* r+ r' r( r, r- r. �len� instances�lifecycle_state�
health_statusrV Z instance_idZget_only_instances)r/ r� r� r� r� r r r r r0 Zec2_connr1 r� r3 r r� �ir� r )r� r� r �
get_instances� sL
�� �
�
�r� c C s8 |dkrt t| |��}dd� |�� D �S t| |��d�S )Nr@ c S s" i | ]
\}}|� d �|� d ��qS )zutf-8)�encode)rF r rC r r r �
<dictcomp>� s �z&_convert_attribute.<locals>.<dictcomp>�ascii)r[ rW r� r� )r� r� r@ r r r r� � s �r� c
C s� t ||||d�}z
|j|| |d�}W n2 tyD } z&td | �}
| j�di ��d�dkr6ddiW Y d } ~ S d
|
iW Y d } ~ S d } ~ ww td d � |d
D ��S )z�
Switch desired instances to StandBy mode
.. versionadded:: 2016.11.0
CLI Example:
.. code-block:: bash
salt-call boto_asg.enter_standby my_autoscale_group_name '["i-xxxxxx"]'
r )� InstanceIds�AutoScalingGroupNameZ ShouldDecrementDesiredCapacity�boto3.get_error�Error�Code�ResourceNotFoundExceptionr4 FNr. c s � � | ] }|d dkV qdS �Z
StatusCodeZFailedNr �rF Zactivityr r r rH $ � �
�z enter_standby.<locals>.<genexpr>�
Activities)r �
enter_standbyr r �responserq �all� r/ r� Z!should_decrement_desired_capacityr r r r r0 r� r3 �errr r r r� s& �
� ���r� c
C s� t ||||d�}z |j|| d�}W n2 tyC } z&td | �}
| j�di ��d�dkr5ddiW Y d } ~ S d
|
iW Y d } ~ S d } ~ ww td d � |d
D ��S )z�
Exit desired instances from StandBy mode
.. versionadded:: 2016.11.0
CLI Example:
.. code-block:: bash
salt-call boto_asg.exit_standby my_autoscale_group_name '["i-xxxxxx"]'
r )r� r� r� r� r� r� r4 FNr. c s r� r� r r� r r r rH J r� z exit_standby.<locals>.<genexpr>r� )r � exit_standbyr r r� rq r� r� r r r r� ) s" �
� ���r� )NNNN)NNNNNNNNNNNNNNNNNN)FNNNN)NNNNNr� NNNFNNFNNTNFNNNN)r� r� r� NNNNN)8�__doc__r� Zemail.mime.multipartr� �loggingr r, Zsalt.utils.compatr
Zsalt.utils.jsonZsalt.utils.odictr rT Zsalt.utils.versions� getLoggerr
r' r� r Zboto.ec2Zboto.ec2.autoscaler rs Zboto.ec2.blockdevicemappingr� Zboto.ec2.instance�setLevel�CRITICALr Zbotocore.exceptionsr r � ImportErrorr r r4 rh r} r� rv rw r� r� r� r� r� r� r� r� r� r& r� r� r� r� r� r r r r �<module> s 0
�
m
�y
� $
!
�&
�
g
�
#
#
�B
�+�