File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/__pycache__/azurearm.cpython-310.pyc
o
�N�g- � @ s� d Z ddlZddlZddlZddlmZ ddlZddlZddl Zddl
Zddl Zddl Zddl
mZmZ zddlmZmZ ddlmZmZ dZW n eyS dZY nw e�e�Zd d
� Zd d � Z d
d� Z dd� Z dd� Z dd� Z ddd�Z!dS )a�
Azure (ARM) Utilities
.. versionadded:: 2019.2.0
:maintainer: <devops@eitr.tech>
:maturity: new
:depends:
* `azure <https://pypi.python.org/pypi/azure>`_ >= 2.0.0rc6
* `azure-common <https://pypi.python.org/pypi/azure-common>`_ >= 1.1.4
* `azure-mgmt <https://pypi.python.org/pypi/azure-mgmt>`_ >= 0.30.0rc6
* `azure-mgmt-compute <https://pypi.python.org/pypi/azure-mgmt-compute>`_ >= 0.33.0
* `azure-mgmt-network <https://pypi.python.org/pypi/azure-mgmt-network>`_ >= 0.30.0rc6
* `azure-mgmt-resource <https://pypi.python.org/pypi/azure-mgmt-resource>`_ >= 0.30.0
* `azure-mgmt-storage <https://pypi.python.org/pypi/azure-mgmt-storage>`_ >= 0.30.0rc6
* `azure-mgmt-web <https://pypi.python.org/pypi/azure-mgmt-web>`_ >= 0.30.0rc6
* `azure-storage <https://pypi.python.org/pypi/azure-storage>`_ >= 0.32.0
* `msrestazure <https://pypi.python.org/pypi/msrestazure>`_ >= 0.4.21
:platform: linux
� N)�
itemgetter)�SaltInvocationError�SaltSystemExit)�ServicePrincipalCredentials�UserPassCredentials)�MetadataEndpointError� get_cloud_from_metadata_endpointTFc C s t sdS dS )NFT)� HAS_AZURE� r
r
�G/opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/azurearm.py� __virtual__6 s r c
K s� d| v rt d | d �}| �|� g d�}ddg}z$| �d�r.| �d��d�r.t| d �}nt�d�}t|| �dd ��}W n tt t
fyQ t � d
�
| d ���w t|��| �rx| d re| d re| d
sitd��t| d | d | d
|d�}nAt|��| �r�| d r�| d s�td��t| d | d |d�}n"d| v r�z
ddlm} ||d�}W n t y� tdd��w td��d| vr�td��tjj�| d �}|||fS )z'
Acquire Azure ARM Credentials
Zprofilez
config.option)� client_id�secret�tenant�username�password�cloud_environment�httpzmsrestazure.azure_cloudZAZURE_PUBLIC_CLOUDz0The Azure cloud environment {} is not available.r
r r z_The client_id, secret, and tenant parameters must all be populated if using service principals.)r r zfThe username and password parameters must both be populated if using username/password authentication.)r �subscription_idr )�MSIAuthenticationzHMSI authentication support not availabe (requires msrestazure >= 0.4.14)��msgz�Unable to determine credentials. A subscription_id with username and password, or client_id, secret, and tenant or a profile with the required parameters populatedz#A subscription_id must be specified)Z__salt__�update�get�
startswithr � importlib�
import_module�getattr�AttributeError� ImportErrorr �sys�exit�format�set�issubsetr r r Z"msrestazure.azure_active_directoryr r �salt�utilsZ stringutilsZto_str) �kwargsZazure_credentialsZ service_principal_creds_kwargsZuser_pass_creds_kwargs� cloud_envZcloud_env_module� credentialsr r r
r
r �_determine_auth= sv
��
��������� �� �
r* c K s� ddddddddd d
d d � }| |vrt d
�| �d��|| }| dv r%d}n | dv r,d}n| }zt�d| �}t||� d��}W n tyO t�d| � d���w td i |��\}}} | dkrg||| j j
d�}
n |||| j j
d�}
|
j � dt
jj� �� |
S ) zT
Dynamically load the selected client and return a management client object
ZComputeManagementZAuthorizationManagementZ
DnsManagementZStorageManagementZManagementLockZMonitorManagementZNetworkManagement�PolicyZResourceManagementZ SubscriptionZWebSiteManagement) ZcomputeZ
authorizationZdnsZstorage�managementlockZmonitor�network�policy�resource� subscriptionZwebz8The Azure ARM client_type {} specified can not be found.r )r. r0 r/ )r, zresource.locks� azure.mgmt.�Clientz
The azure z client is not available.r0 )r) �base_url)r) r r3 zSalt/Nr
)r r" r r r r r r! r* Z endpointsZresource_manager�configZadd_user_agentr% �version� __version__) Z client_typer' Z
client_mapZ map_value� module_nameZ
client_moduler2 r) r r( �clientr
r
r �
get_client� sV ��� ���r9 c K sJ z
t t|�d��}W n ttfy t td�}Y nw |d| �� |� dS )z/
Log an azurearm cloud error exception
Zazurearm_log_level�errorz*An AzureARM %s CloudError has occurred: %sN)r �logr r � TypeError�
capitalize)r8 �messager' Z cloud_loggerr
r
r �log_cloud_error� s �
�r? c C s8 g } z
t | �}|�|�� � W n
ty Y |S w q)zK
Extract all pages within a paged object as a list of dictionaries
)�next�appendZas_dict�
StopIteration)Z paged_objectZ paged_returnZpager
r
r �paged_object_to_list� s ��rC c K s� i }zt �d| � d��}t||�}W n ty" t�d�|| ���w dt|�v r�|j� � D ]�\}}|�
|�}|dur�|d d � � rVt |t
�rVt| |d fi |��||< q.|d d dkrht |t
�rh|||< q.|d d d kr�t |t�r�g } |D ]V}
|d d
� � r�t |
t
�r�| �t| |d |d �d �d
|d �d �� fi |
��� qy|d d
dkr�t |
t
�r�| �|
� qy|d d
� � s�|d d
dkr�| �|
� qy| ||< q.|||< q.|d i |��S )
z6
Assemble an object from incoming parameters.
r1 z.modelsz5The {} model in the {} Azure module is not available.�_attribute_mapN�typer �{�[� �]r
)r r r r r r! r" �dirrD �itemsr �isupper�
isinstance�dict�create_object_model�listrA �index�rindex) r7 Z object_namer' Z
object_kwargsZ model_moduleZModel�attrrK �paramZobj_listZ list_itemr
r
r rO � sd ���
� �
� �����
�
�rO c C sB i }|sg }t |t�sd|d< |S t|�t| �kr"| |d�|d< |S z
dd� || fD �\}}W n ty> d|d< | Y S tyL d|d< | Y S w t|�D ]M\}}|D ]F}|| } ||v rs|| �|i ��d d
��d �d }
n|| �|�}
t | t�r�| � � } t |
t�r�|
� � }
| |
kr�||d�|d< | S qWqQ|S )
a
Compare lists of dictionaries representing Azure objects. Only keys found in the "new" dictionaries are compared to
the "old" dictionaries, since getting Azure objects from the API returns some read-only data which should not be
used in the comparison. A list of parameter names can be passed in order to compare a bare object name to a full
Azure ID path for brevity. If string types are found in values, comparison is case insensitive. Return comment
should be used to trigger exit from the calling function.
z+must be provided as a list of dictionaries!Zcomment)�old�newZchangesc s s � | ] }t |td �d�V qdS )�name)�keyN)�sortedr )�.0r4 r
r
r � <genexpr>3 s �
�z(compare_list_of_dicts.<locals>.<genexpr>z:configurations must be provided as a list of dictionaries!z7configuration dictionaries must contain the "name" key!�id� �/���)
rM rP �lenr< �KeyError� enumerater �split�str�lower) rU rV Zconvert_id_to_name�retZ
local_configsZremote_configs�idx�valrX Z local_valZ
remote_valr
r
r �compare_list_of_dicts sJ
� � �
��ri )N)"�__doc__r �loggingr �operatorr Z salt.configr% Z salt.loaderZsalt.utils.argsZsalt.utils.stringutilsZsalt.utils.versionsZ salt.versionZsalt.exceptionsr r Zazure.common.credentialsr r Zmsrestazure.azure_cloudr r r r � getLogger�__name__r; r r* r9 r? rC rO ri r
r
r
r �<module> s6 �
P<5