File manager - Edit - /opt/saltstack/salt/lib/python3.10/site-packages/salt/modules/__pycache__/boto_asg.cpython-310.pyc
Back
o ;j~� � @ sl d Z ddlZddlZddlZddlZddlZddlmZ ddl Z ddl Z ddl Z e� e�ZdZz>ddlZddlZddlm mZ ddlm mZ ddlZe� d��ej� ddlZddl m Z e� d��ej� dZ W n e yx d Z Y nw d d � Z d d � Z!d<dd�Z"d<dd�Z# d=dd�Z$ d=dd�Z%dd� Z&dd� Z'd>dd�Z(d d � Z)d<d d �Z*d<d d!�Z+d<d"d#�Z, d<d$d%�Z- & d?d'd(�Z.d<d)d*�Z/ d<d+d,�Z0d<d-d.�Z1d<d/d0�Z2 1 2 3 d@d4d5�Z3d6d7� Z4 d>d8d9�Z5 d>d:d;�Z6dS )Aa� 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 � N)� OrderedDictz%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( ��r5 c C sB t ||||d�}d} z�|j| gd�}|r|d }ni W S t� }g d�} | D ]U} | dkrPg } |jD ]} t� } | j| d< | j| d < | j| d < | �| � q1| |d< q&| d kr_t|| �� d �|| < q&| d krtt|| �}t dd� |D ��|| < q&t|| �|| < q&|j | d�}g |d< |D ] }|d �t d|j fd|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 tj 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 )r0 �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_processesrA r �value�propagate_at_launchr@ �,rC 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_policiesr0 �adjustment_type�scaling_adjustment�min_adjustment_step�cooldown�scheduled_actionsNr> r= r8 � start_time�end_time� recurrencer" r# r$ r% )$r& r' r rA r rD rE �append�getattr�split�sorted�get_all_policies�dictr0 rM rN rO rP �get_all_scheduled_actionsrS � isoformatr> r= �intr8 rR rT r r* r+ r, r( r) r- r. r/ )r0 r r r r r1 r2 r �ret�attrs�attr�_tags�tag�_tagrC �policies�policy�actions�actionrS r4 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.FrD �Tag missing value.rE �r rD Z resource_idrE r T)r0 � launch_configr6 r> r= r8 r< r7 r: r9 r? rA r@ rB rC zCreated ASG %sr" r# r$ r% NzFailed to create ASG %s)r& � isinstance�strr r �json�loads�get�KeyErrorr( r/ � autoscale�TagrU �AutoScalingGroupZcreate_auto_scaling_group�_create_scaling_policies�_create_scheduled_actions� put_notification_configuration�infor r* r+ r, r) r- r. )!r0 r; r6 r> r= r8 r< r7 r: r9 r? r@ rA rB rC rL rQ r �notification_arn�notification_typesr r r r1 ra rb rD rE rc r2 �_asgr4 r3 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 �rG �tr r r � <listcomp>� s � ��zupdate.<locals>.<listcomp>z boto3.orderedr ri FzTag z missing keyrD rj z missing valuerE rk c s g | ]}|� vr|�qS r r r �Z desired_tagsr r r� � s r T)� connectionr0 rl r6 r> r= r8 r< r7 r: r9 r? rA r@ rB 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 %srJ )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/ rU rs ru rx �updater) Zcreate_or_update_tags� delete_tagsZresume_processesZsuspend_processesry rY Z delete_policyr0 rv r[ Zdelete_scheduled_actionrw r r* r+ r, r- r. )&r0 r; r6 r> r= r8 r< r7 r: r9 r? r@ rA rB rC rL rQ rz r{ r r r r r1 Zconn3Z current_tagsZadd_tagsrb rD rE rc r� r2 r| re Zscheduled_actionr4 r3 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 policiesr0 rM rN rO NrP )r0 �as_namerM rN rO rP )rs re Z ScalingPolicyrq Zcreate_scaling_policy)r1 r� rL 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 rR rS r8 r> r= rT )r8 r> r= rR rS rT N)�itemsrm rn �datetime�strptime� DATE_FORMATZ create_scheduled_group_actionrq )r1 r� rQ r0 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/ ) r0 �forcer r r r r1 r2 r3 r4 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/ � r0 r r r r r1 r2 �lcr3 r4 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 r1 r2 r4 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 �r0 )rG �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 r0 �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/ )%r0 r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r1 Z_bdmsZ_block_device_mapZblock_device_dictZ block_device� attributesZ _block_device� attributerD r� r2 r4 r3 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/ ) r0 r r r r r1 r2 r4 r3 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% rJ zCould not convert: %sNr" z)Throttled by API, will retry in 5 secondsr$ z"Maximum number of retries exceeded) r& rY r0 Z policy_arnr( r/ r r* r+ Z error_coder) r- r. ) rK Zscaling_policy_namer r r r r1 r2 rd re r4 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 �rG �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 r1 r2 r� �asgsr^ r4 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� )rG 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)rG r` ��instancer r r� � s z,get_instances.<locals>.<listcomp>.<listcomp>r )rG )r� r� r r� � s ��c s g | ] }t |� �rt|� ��qS r )rV r� )rG r� )r� r r r� � s ��)r& r r' r r* r+ r, r( r) r- r. r/ �len� instances�lifecycle_state� health_statusrU Z instance_idZget_only_instances)r0 r� r� r� r� r r r r r1 Zec2_connr2 r� r4 r r� �ir� r )r� r� r � get_instances� sL �� � � �r� c C s8 |dkrt t| |��}dd� |�� D �S t| |��d�S )NrA c S s" i | ] \}}|� d �|� d ��qS )zutf-8)�encode)rG r rD r r r � <dictcomp>� s �z&_convert_attribute.<locals>.<dictcomp>�ascii)rZ rV r� r� )r� r� rA 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�ResourceNotFoundExceptionr5 FNr/ c s � � | ] }|d dkV qdS �Z StatusCodeZFailedNr �rG Zactivityr r r rI $ � � �z enter_standby.<locals>.<genexpr>� Activities)r � enter_standbyr r �responserq �all� r0 r� Z!should_decrement_desired_capacityr r r r r1 r� r4 �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� r5 FNr/ c s r� r� r r� r r r rI 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)7�__doc__r� Zemail.mime.multipartr� �loggingr r- � collectionsr Zsalt.utils.compatr Zsalt.utils.jsonZsalt.utils.versions� getLoggerr r( r� r Zboto.ec2Zboto.ec2.autoscaler rs Zboto.ec2.blockdevicemappingr� Zboto.ec2.instance�setLevel�CRITICALr Zbotocore.exceptionsr r � ImportErrorr r r5 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 �+�
| ver. 1.4 |
Github
|
.
| PHP 8.2.30 | Generation time: 0 |
proxy
|
phpinfo
|
Settings