golden hour
/opt/saltstack/salt/lib/python3.10/site-packages/salt/cloud/clouds/__pycache__
⬆️ Go Up
Upload
File/Folder
Size
Actions
__init__.cpython-310.pyc
171 B
Del
OK
aliyun.cpython-310.pyc
19.81 KB
Del
OK
azurearm.cpython-310.pyc
39.67 KB
Del
OK
clc.cpython-310.pyc
8.91 KB
Del
OK
cloudstack.cpython-310.pyc
10.87 KB
Del
OK
digitalocean.cpython-310.pyc
27.39 KB
Del
OK
dimensiondata.cpython-310.pyc
12.62 KB
Del
OK
ec2.cpython-310.pyc
92.28 KB
Del
OK
gce.cpython-310.pyc
48.78 KB
Del
OK
gogrid.cpython-310.pyc
11.07 KB
Del
OK
hetzner.cpython-310.pyc
11.82 KB
Del
OK
joyent.cpython-310.pyc
22.78 KB
Del
OK
libvirt.cpython-310.pyc
16.69 KB
Del
OK
linode.cpython-310.pyc
59.17 KB
Del
OK
lxc.cpython-310.pyc
10.9 KB
Del
OK
msazure.cpython-310.pyc
71.44 KB
Del
OK
oneandone.cpython-310.pyc
18.39 KB
Del
OK
opennebula.cpython-310.pyc
90.2 KB
Del
OK
openstack.cpython-310.pyc
23.53 KB
Del
OK
packet.cpython-310.pyc
12.06 KB
Del
OK
parallels.cpython-310.pyc
11.89 KB
Del
OK
profitbricks.cpython-310.pyc
25.34 KB
Del
OK
proxmox.cpython-310.pyc
27.25 KB
Del
OK
pyrax.cpython-310.pyc
2.59 KB
Del
OK
qingcloud.cpython-310.pyc
16.64 KB
Del
OK
saltify.cpython-310.pyc
11.91 KB
Del
OK
scaleway.cpython-310.pyc
9.36 KB
Del
OK
softlayer.cpython-310.pyc
13.26 KB
Del
OK
softlayer_hw.cpython-310.pyc
13.5 KB
Del
OK
tencentcloud.cpython-310.pyc
20.55 KB
Del
OK
vagrant.cpython-310.pyc
8.32 KB
Del
OK
virtualbox.cpython-310.pyc
9.57 KB
Del
OK
vmware.cpython-310.pyc
94.14 KB
Del
OK
vultrpy.cpython-310.pyc
12.8 KB
Del
OK
xen.cpython-310.pyc
25.66 KB
Del
OK
Edit: azurearm.cpython-310.pyc
o �xeZ � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z ddlZddlmZ ddlZddlZddlZddlZddlZddlZddlZddlmZmZm Z m Z dZ z(ddl m m! m"Z# ddl$m m% m"Z& ddl'm(Z( dd l)m*Z* d Z W n e+y� Y nw d Z,e�-e.�Z/d d � Z0dd� Z1dd� Z2e2dPdd��Z3e2dQdd��Z4dd� Z5e2dd� �Z6e2dd� �Z7e2dPd d ��Z8e2dQd d ��Z9e2dQd d!��Z:e2dQd"d#��Z;e2dQd$d%��Z<e2dQd&d'��Z=e2dQd(d)��Z>e2dQd*d+��Z?e2dPd,d-��Z@d.d/� ZAd0d1� ZBe2dPd2d3��ZCdQd4d5�ZDe2d6d7� �ZEe2dPd8d9��ZFe2dQd:d;��ZGd<d=� ZHdQd>d?�ZIe2dPd@dA��ZJe2dPdBdC��ZKe2dPdDdE��ZLe2dPdFdG��ZMe2dPdHdI��ZNe2 dPdJdK��ZOe2dQdLdM��ZPe2dQdNdO��ZQdS )RaG Azure ARM Cloud Module ====================== .. versionadded:: 2016.11.0 .. versionchanged:: 2019.2.0 The Azure ARM cloud module is used to control access to Microsoft Azure Resource Manager .. warning:: This cloud provider will be removed from Salt in version 3007 in favor of the `saltext.azurerm Salt Extension <https://github.com/salt-extensions/saltext-azurerm>`_ :maintainer: <devops@eitr.tech> :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 :configuration: Required provider parameters: if using username and password: * ``subscription_id`` * ``username`` * ``password`` if using a service principal: * ``subscription_id`` * ``tenant`` * ``client_id`` * ``secret`` if using Managed Service Identity authentication: * ``subscription_id`` Optional provider parameters: **cloud_environment**: Used to point the cloud driver to different API endpoints, such as Azure GovCloud. Possible values: * ``AZURE_PUBLIC_CLOUD`` (default) * ``AZURE_CHINA_CLOUD`` * ``AZURE_US_GOV_CLOUD`` * ``AZURE_GERMAN_CLOUD`` * HTTP base URL for a custom endpoint, such as Azure Stack. The ``/metadata/endpoints`` path will be added to the URL. **userdata** and **userdata_file**: Azure Resource Manager uses a separate VirtualMachineExtension object to pass userdata scripts to the virtual machine. Arbitrary shell commands can be passed via the ``userdata`` parameter, or via a file local to the Salt Cloud system using the ``userdata_file`` parameter. Note that the local file is not treated as a script by the extension, so "one-liners" probably work best. If greater functionality is desired, a web-hosted script file can be specified via ``userdata_file: https://raw.githubusercontent.com/account/repo/master/azure-script.py``, which will be executed on the system after VM creation. For Windows systems, script files ending in ``.ps1`` will be executed with ``powershell.exe``. The ``userdata`` parameter takes precedence over the ``userdata_file`` parameter when creating the custom script extension. **win_installer**: This parameter, which holds the local path to the Salt Minion installer package, is used to determine if the virtual machine type will be "Windows". Only set this parameter on profiles which install Windows operating systems. Example ``/etc/salt/cloud.providers`` or ``/etc/salt/cloud.providers.d/azure.conf`` configuration: .. code-block:: yaml my-azure-config with username and password: driver: azurearm subscription_id: 3287abc8-f98a-c678-3bde-326766fd3617 username: larry password: 123pass Or my-azure-config with service principal: driver: azurearm subscription_id: 3287abc8-f98a-c678-3bde-326766fd3617 tenant: ABCDEFAB-1234-ABCD-1234-ABCDEFABCDEF client_id: ABCDEFAB-1234-ABCD-1234-ABCDEFABCDEF secret: XXXXXXXXXXXXXXXXXXXXXXXX cloud_environment: AZURE_US_GOV_CLOUD The Service Principal can be created with the new Azure CLI (https://github.com/Azure/azure-cli) with: az ad sp create-for-rbac -n "http://<yourappname>" --role <role> --scopes <scope> For example, this creates a service principal with 'owner' role for the whole subscription: az ad sp create-for-rbac -n "http://mysaltapp" --role owner --scopes /subscriptions/3287abc8-f98a-c678-3bde-326766fd3617 *Note: review the details of Service Principals. Owner role is more than you normally need, and you can restrict scope to a resource group or individual resources. � N��wraps)� cpu_count)� ThreadPool)�SaltCloudConfigError�SaltCloudExecutionFailure�SaltCloudExecutionTimeout�SaltCloudSystemExitF)�BlockBlobService)� CloudErrorT�azurearmc C s t � du rdS t� du rdS tS )z) Check for Azure configurations. F)Fz�The following dependencies are required to use the AzureARM driver: Microsoft Azure SDK for Python >= 2.0rc6, Microsoft Azure Storage SDK for Python >= 0.32, MS REST Azure (msrestazure) >= 0.4)�get_configured_provider�get_dependencies�__virtualname__� r r �N/opt/saltstack/salt/lib/python3.10/site-packages/salt/cloud/clouds/azurearm.py� __virtual__� s r c C s" zt �� W S ty t Y S w �N)Z__active_provider_name__�value�AttributeErrorr r r r �_get_active_provider_name� s �r c s t � �� fdd��}|S )z= Decorator wrapper to warn about msazure deprecation c s6 t jjjddtd� � | i t jjjdi |����}|S )NZChlorinez�This cloud provider will be removed from Salt in version 3007 due to the deprecation of the 'Classic' API for Azure. Please migrate to Azure Resource Manager by March 1, 2023 (https://docs.microsoft.com/en-us/azure/virtual-machines/classic-vm-deprecation))�categoryr )�salt�utilsZversionsZ warn_until� FutureWarning�argsZ clean_kwargs)r �kwargs�ret��functionr r �wrapped� s � z%_deprecation_message.<locals>.wrappedr )r r r r r �_deprecation_message� s r! c C s� |du ri }d|vrt d��d|vrt d��g }z(tdd�}|jj|d d�}|jD ]}t|j�|d kr=|�� }|d }q*W |S ty\ } zt j j � d|j � W Y d}~|S d}~ww ) z* Get a resource type api versions N�resource_providerz%A resource_provider must be specified� resource_typez!A resource_type must be specified�resource�� client_type�Zresource_provider_namespace� api_versions)r �get_conn� providers�get�resource_types�strr# �as_dictr r r r �log_cloud_error�message)�callr r( �resconn�provider_queryr$ � resource_dict�excr r r �get_api_versions� s0 � ��� ��r6 c C s� i }z!t dd�}|jj| |d�}|�� }|dur || }W |S |}W |S tyD } ztjj�d|j � d|j i}W Y d}~|S d}~ww )z( Get an AzureARM resource by id r$ r% )� resource_id� api_versionN�Error) r) Z resourcesZ get_by_idr. r r r r r/ r0 )r7 r8 Z extract_valuer r2 Zresource_queryr4 r5 r r r �get_resource_by_id� s$ � ����r: c C s6 g d�} | D ]}t �tt� pt|�}|r| S q|S )z8 Return the first configured provider instance. ))�subscription_id�tenant� client_id�secret)r; �username�password)r; )�configZis_provider_configured�__opts__r r )Z key_combosZcombo�providerr r r r � s ��r c C s t �tdti�S )z* Warn if dependencies aren't met. r )rA Zcheck_driver_dependenciesr �HAS_LIBSr r r r r s r c C s� i }t jj�tjdt� tdd��|d< tjdt� tdd�}|dur%||d< tjdt� tdd�}|durPtjdt� tdd�}tjdt� tdd�}|�|||d �� tjd t� tdd�}|rntjd t� tdd�}|�||d �� t jj j dd | i|��}|S )z7 Return a connection object for a client type. r; F�� search_global�cloud_environmentNr< r= r> )r= r> r< r? r@ )r? r@ r& r ) r r � stringutils�to_strrA �get_cloud_config_valuer rB �updater Z get_client) r& Z conn_kwargs� cloud_envr< r= r> r? r@ �clientr r r r) s@ �� � � � � � �r) c C s* |si }t � }|�|� tjd|tdd�S )zB Return the location that is configured for this provider �locationFrE )r rK rA rJ rB )r1 r Zvm_dictr r r � get_locationC s �rO c C s� | dkrt d��i }g |d< z:tdd�}|jjdd�}g }|jD ]}t|j�dkr1|�� }|d }q |D ]}|�� � d d �}|d � |� q4W |S t yi } zt j j�d| j� d | ji}W Y d } ~ |S d } ~ ww ) z1 Return a dict of all available regions. �actionzfThe avail_locations function must be called with -f or --function, or with the --list-locations option� locationsr$ r% �Microsoft.Computer' ZvirtualMachines� � r9 N)r r) r* r+ r, r- r# r. �lower�replace�appendr r r r r/ r0 ) r1 r r2 r3 rQ r$ r4 rN Z lowercaser5 r r r �avail_locationsQ s8 � � �����rX c s� | dkrt d��tdd�� t� �g }i }� �fdd�}z� jj�d�}|D ] }|�� }|�|d � q%W n tyO } ztj j � d|j � W Y d }~nd }~ww t t� d �}|�||�} | �� d d � | �� D �}|S ) z? Return a dict of all available images on the provider rP z`The avail_images function must be called with -f or --function, or with the --list-images option�computer% c s i }z]� j j�| d�}|D ]O}|�� }� j j�| |d d�}|D ];}|�� }� j j�| |d |d d�}|D ]$} | �� } d�| |d |d | d f�} | |d |d | d d�|| < q6q q W |S ty� } ztjj � d| j � | | j i}W Y d} ~ |S d} ~ ww ) z: Get all images from a specific publisher )rN �publisher_name�name)rN rZ �offer)rN rZ r\ �skus�|�� publisherr\ �sku�versionrY N) �virtual_machine_imagesZ list_offersr. Z list_skus�list�joinr r r r r/ r0 ) r` �dataZoffersZ offer_objr\ r] Zsku_objra �resultsZ version_objrb r[ r5 ��compconnZregionr r �_get_publisher_images� sZ ����� ����#���z+avail_images.<locals>._get_publisher_images�rN r[ N� c S �$ i | ]}|� � D ]\}}||�qqS r ��items��.0�result�k�vr r r � <dictcomp>� � $ z avail_images.<locals>.<dictcomp>)r r) rO rc Zlist_publishersr. rW r r r r r/ r0 r r � map_async�waitr+ ) r1 Z publishersr rj Zpublishers_queryZ publisher_objr` r5 �poolrg r rh r � avail_imagess s4 � /�� �� rz c C s� | dkrt d��tdd�}i }t� }z|jj|d�}|D ] }|�� }|||d < q W |S tyL } ztjj � d|j � d|j i}W Y d}~|S d}~ww ) z< Return a list of sizes available from the provider rP z^The avail_sizes function must be called with -f or --function, or with the --list-sizes optionrY r% rk r[ r9 N) r r) rO Zvirtual_machine_sizesrd r. r r r r r/ r0 )r1 ri r rN ZsizesZsize_obj�sizer5 r r r � avail_sizes� s( � ����r| c C sT | dkrt d��i }t� }|D ]}d|i||< dD ] }|| �|�|| |< qq|S )z( List VMs on this Azure account rP z=The list_nodes function must be called with -f or --function.r[ )�id�imager{ �state� private_ips� public_ips)r �list_nodes_fullr+ )r1 r �nodes�nodeZpropr r r � list_nodes� s � �r� c s� | dkrt d��tddd�d�}|d � tdd �}i }� fd d �}t� D ]<}g }|jj|d �}|D ]}|�� } || d < |�| � q1tt � d �} | � ||�} | � � dd� | � � D �} |� | � q$|S )z@ List all VMs on the subscription with full information rP zBThe list_nodes_full function must be called with -f or --function.�Microsoft.NetworkZnetworkInterfaces�r" r# �r r rY r% c s� i }| d | d< | d d | d< | d | d< g | d< g | d < | || d < z| d d }d � |d |d |d |d g�| d< W n4 ttfys z| d d d d | d< W n ttfyp | �d i ��d i ��d�| d< Y nw Y nw z<| d d }t|�D ]/\}}t|d � d �}t|| d �\}}}| d d | �|� | d �|� | d �|� qW n t y� Y nw | || d < |S )z Get node info. Zvm_idr} �hardware_profile�vm_sizer{ Zprovisioning_stater r� r� r[ �storage_profile�image_referencer^ r` r\ ra rb r~ �os_disk�uri�network_profile�network_interfaces�resource_group) re � TypeError�KeyErrorr+ � enumerater: �_get_network_interfacerK �extend� Exception) r� Znode_retZ image_refZ netifaces�index�netifaceZ netiface_nameZpubipsZprivips��netapi_versionr r �_get_node_info sV � � ���� � �� � z'list_nodes_full.<locals>._get_node_info��resource_group_namer� rl c S rm r rn rp r r r ru D rv z#list_nodes_full.<locals>.<dictcomp>)r r6 r) �list_resource_groups�virtual_machinesrd r. rW r r rw rx r+ rK ) r1 �netapi_versionsri r r� �groupr� Z nodes_queryZnode_objr� ry rg Z group_retr r� r r� � s4 ��� / r� c C s� | dkrt d��tdd�}i }z|j�� }|D ] }|�� }|||d < qW |S tyG } ztjj� d|j � d|j i}W Y d}~|S d}~ww )z? List resource groups associated with the subscription rP zFThe list_hosted_services function must be called with -f or --functionr$ r% r[ r9 N) r r) �resource_groupsrd r. r r r r r/ r0 )r1 r2 r �groupsZ group_objr� r5 r r r r� J s& � ����r� c C sZ |dkrt d��ztd�| }W n ty! t�d| � i }Y nw td |t� t� |S )z? Show the details from AzureARM concerning an instance rP z<The show_instance action must be called with -a or --action.r z Failed to get data for node '%s'zcloud.cache_node)r r� r� �log�debug� __utils__r rB )r[ r1 r� r r r � show_instancec s � �r� c C s� |du ri }t dd�}|�d�du r tjdi tdd�|d< g }|j�|d |d �}|j}|jD ]}|�|j� q0|j� |d |d �}|� � |D ]}|j � |d |�}|� � qJ||iS )z% Delete a network interface. N�networkr% r� TrE � iface_name) r) r+ rA rJ rB r� r[ �ip_configurationsrW �deleterx �public_ip_addresses)r1 r �netconnZips�ifacer� �ip_�pollerr r r �delete_interfacew s0 �� � r� c C sl t dd�}z|jj|| d�}|�� }W |S ty5 } ztjj�d|j � d|j i}W Y d}~|S d}~ww )z4 Get the public ip address details by name. r� r% )r� �public_ip_address_name�errorN) r) r� r+ r. r r r r r/ r0 )r[ r� r� Z pubip_queryZpubipr5 r r r �_get_public_ip� s � ���r� c C s� g }g }t ddd�d�}|d }tdd�}|jj|| d�}|�� }t|d �D ];\} } | �d �d ur9|�| d � d | �d i �v rbt| d d |d�} t| |�} |�| d � |d | d � | � q'|||fS )z" Get a network interface. r� ZpublicIPAddressesr� r� r r� r% )r� Znetwork_interface_namer� �private_ip_addressNr} �public_ip_addressr[ � ip_address) r6 r) r� r+ r. r� rW r: r� rK ) r[ r� r� r� r� r� r� Znetiface_queryr� r� Z ip_configZpublic_ip_nameZ public_ipr r r r� � s2 �� �� � r� c C s� | dkrt d��ttd�}ttd�}ttd�}ttd�}t|t�s#i }|}tdd�}|�d �d u r6t� |d < |�d�d u rHtj d|t d d �|d< |�d �d u rZtj d |t d d �|d < |�d�d u rltj d|t d d �|d< |�d�d u r|d� |d �|d< z|j j|d |d |d d�}W n t y� } zt d� | j|d |d |d ���d } ~ ww i } d } d|v �r |d } t| t��r g } | �� D ]T\}}|D ]M}z|j�|d ||�}| �d|�� d i� W q� t y� } zt�dt| �� W Y d } ~ q�d } ~ w t�y } zt�dt| �� W Y d } ~ q�d } ~ ww q�| | d< d|�� v �r0|d | d< |j| d< n|j| d< |�d�d u �r�d � |d �}|jj|d |||d |jd �d �}d }|� � z+|j�|d |�}|j �r�|t|j �d!�| d"< |d-d � |d �|d#�| ��g} W n/W n t �y� } z t�d| � W Y d } ~ nd } ~ ww |d$7 }|d%k�r�t d&��t �!d'� �q]nd � |d �}|d-||d#�| ��g} d }|�d(�d u�r�|j"j|d |d( d)�}||d || d*�}|j#�|d |d |�}z|� � W n t$�y } z t�%d+| � W Y d } ~ nd } ~ ww d } z t&|d |d �W S t �y= |d$7 }|d%k�r6t d,��t �!d'� Y nw �q).z% Create a network interface. rP zGThe create_network_interface action must be called with -a or --action.�IPAllocationMethod�NetworkInterface� NetworkInterfaceIPConfiguration�PublicIPAddressr� r% rN NFrE �subnetZnetwork_resource_groupr� r� z {}-iface0r[ )r� Zvirtual_network_nameZ subnet_namez3{} (Resource Group: "{}", VNET: "{}", Subnet: "{}")�#load_balancer_backend_address_poolsr} zThere was a cloud error: %sz2There was an error getting the Backend Pool ID: %sr� Z private_ip_allocation_methodZallocate_public_ipTz{}-ip)rN Zpublic_ip_allocation_method)r� r� � parametersr �r} r� )r[ r� � �x z(Timed out waiting for public IP Address.� Zsecurity_group)r� Znetwork_security_group_name)rN �network_security_groupr� zlNetwork interface creation could not be polled. It is likely that we are reusing an existing interface. (%s)z,Timed out waiting for operation to complete.r )'r �getattr�network_models� isinstance�dictr) r+ rO rA rJ rB �format�subnetsr r0 ro r� rW r. r� r� r- r� �keysZstaticZdynamicr� �create_or_updaterx r� r} � ValueError�time�sleepZnetwork_security_groupsr� r� �warningr� )r1 r r� r� r� r� �vm_r� Z subnet_objr5 Z ip_kwargsr� Z pool_dictsZpool_idsZload_balZbe_poolsry Z lbbep_dataZ pub_ip_namer� �countZ pub_ip_dataZ priv_ip_namer� Z iface_paramsr r r �create_network_interface� sD � � � � � ����� � ����� ��� � ���� �� ������� ��� � ��r� c D C sZ t dd�}ttd�}ttd�}ttd�}ttd�}ttd�}ttd�}ttd �} ttd �} ttd �} ttd �} ttd �} ttd�}ttd�}ttd�}ttd�}ttd�}tjdt� tdd�}| �d�du rjd| d< | �d�du rvt� | d< | �d�du r�tjd| tdd�| d< | �d �du r�tjd | tdd�| d < t d | d �\}}}|d | d < d!� | d �}tjd"| tdtjd#| tdd�d$�}d}tjd%| tddd$�}|du�r z t j j �|d&�� }|�� }W d � n1 s�w Y W n t�y } z td'� || jd( ���d} ~ ww tjd)| tddd$�} i } tjd*| tdd�} | �sB|du�rB| |d+� |�d,�} | | gd-�}!|| |!d.�}"|"| d/< d}#nt j j�tjd0| tdtjd1| tdd�d$��}#| �sc|#du�r�| �s�t|#t��smtd2��t|#�d3k �s{t|#�d4k�rtd5��d6}$td7d8� |#D ���r�|$d97 }$td:d8� |#D ���r�|$d97 }$td;d8� |#D ���r�|$d97 }$td<d8� |#D ���r�|$d97 }$|$d=k �r�td>��|#| d?< tjd@| tddd$�}%|%du�r�t|%t��r�d dA� || d |%�i}%nd}%t� }&|&jj }'t| �dB�t��rt j j � | dB �}(n| �dB�}(d})t|(t ��rg })ng }(d6}*g }+|(D ]�},t|,t��r'd |,i},|,� d |,�d |,�d dC� | d t|*����� |,� dD|,�dE|,�dFdG��� |,� dH|,�dIdJ�� |*|+v �rn|*d97 }*|*dKk�rit!�"dL� n|*|+v �sZ|,� dM|*� |*d97 }*dN|,v �r�||,dN dO�|,dP< |,dN= n*|,�dP�dQk�r�|dR� | dS |'| d |,dM �dO�|,dP< ndP|,v �r�||,dP dO�|,dP< dT|,v �r�dU|,dV< ndW|,v �r�dW|,dV< ndX|,dV< |)�#|d�i |,��� �qd}-| dT �$dY��s�| �dP�dQk�rI| dT �$dY��r�|| dT dO�}.n,d}.dZ| dT v �r| dT �%dZ�\}/}0}1}2||/|0|1|2d[�}-n| dT �$d\��r&|| dT d]�}-| �r,d^}3nd_}3| |j&|j'||d`� | dS |'|�dO�|3|.| �da�db�}4n8d}.d}3| |j'| �da�dc�}4dZ| dT v �rr| dT �%dZ�\}/}0}1}2||/|0|1|2d[�}-n| dT �$d\��r�|| dT d]�}-tjdd| tddd$�}5tjde| tddd$�}6tjdf| tddd$�}7|5�r�t(j)�*|5��r�t j j �|5d&�� }8|8�� }6W d � n 1 �s�w Y |6�r|7�rtjdg| tddd$�}9|9�r t j j+�,t�dh| t��\| di< | dj< | �d �}:dk| v �r�dl�-|:| dk g�}:t j j+�.tdm | dj |:� t j j+�/t| |6�}6d};|6du�s |5du�r�z_| �r9dn}<do}=dp}>|5�r6|5�0dq��r6dr}?n ds}?ndt}<du}=dv}>ds}?i }@|6�rK|6|@dw< n |5�$dY��rg|5g|@dx< |?dy |5|5�1dz�d9 d� |@dw< | d | d | d d{ | d |<|=|>d|@dd|� };W n t�y� } z t!�2d}| � W Y d} ~ nd} ~ ww || d d|t|| dF �3� |�d~�||4|)|-d�|d�|| d d��| ��| | | d d]�gd��|%d��}At4d� d�d�d�� | d �t4d� d�| g d���td� td� d�� z'|j5j6| d | d |Ad��}B|B�7� |B�8� }C|C�9� }C|;�r t:|;d�� W |CS W |CS t;�y, } zt j j<�=d| j>� i }CW Y d} ~ |CS d} ~ ww )�z" Request a VM from Azure. rY r% � CachingTypes�DataDisk�DiskCreateOptionTypes�HardwareProfile�ImageReference�LinuxConfiguration�SshConfiguration� SshPublicKey�NetworkInterfaceReference�NetworkProfile�OSDisk� OSProfile�StorageProfile�VirtualHardDisk�VirtualMachine�VirtualMachineSizeTypesr; FrE �driverNr rN r� Tr[ rP )r1 r r} Ziface_idz{}-vol0� ssh_usernameZ win_username�rF �default�ssh_publickeyfile�rz*Failed to read ssh publickey file '{}': {}���� disable_password_authentication� win_installerz /home/{}/.ssh/authorized_keys)Zkey_data�path)Z public_keys)r� ZsshZlinux_configuration� ssh_passwordZ win_passwordz$The admin password must be a string.� �{ z9The admin password must be between 8-123 characters long.r c s � � | ]}|� � V qd S r )�isdigit�rq �charr r r � <genexpr> � � z#request_instance.<locals>.<genexpr>r� c s r� r )�isupperr� r r r r� r� c s r� r )�islowerr� r r r r� r� c s s � | ]}|t jv V qd S r )�string� punctuationr� r r r r� s � � zkThe admin password must contain at least 3 of the following types: upper, lower, digits, special charactersZadmin_password�availability_setzS/subscriptions/{}/resourceGroups/{}/providers/Microsoft.Compute/availabilitySets/{}�volumesz {}-datadisk{}� disk_size_gbZlogical_disk_size_in_gbr{ �d �cachingZ host_cachingZReadOnly� z"Maximum lun count has been reached�lunZ media_link)r� �vhdZ unmanagedz)https://{}.blob.{}/vhds/{}-datadisk{}.vhd�storage_accountr~ � from_image� create_option�attach�empty�httpr^ r_ z/subscriptionsr� �Windows�Linuxz https://{}.blob.{}/vhds/{}.vhdZos_disk_size_gb)r r r[ r �os_typer~ r� )r r� � userdata_file�userdata�userdata_template�userdata_sendkeysZkeysizeZpriv_keyZpub_keyZ append_domain�.Zpki_dirrR ZCustomScriptExtensionz1.8z.ps1z/powershell -ExecutionPolicy Unrestricted -File rT �Microsoft.Azure.Extensions� CustomScript�2.0�commandToExecuteZfileUrisz./�/Z_custom_userdata_script) r� �virtual_machine_name�extension_namerN r` � virtual_machine_extension_type�type_handler_version�auto_upgrade_minor_version�settings�protected_settingsz Failed to encode userdata: %s)r� )r� � data_disksr� )Zadmin_usernameZ computer_name)r� )rN Zplanr� r� Z os_profiler� r� �cloud.fire_event�eventzrequesting instancezsalt/cloud/{}/requesting�cloud.filter_eventZ requesting�r[ �profilerC r� �sock_dir� transport�r r$ r% )r� �vm_namer� r� r )?r) r� �compute_modelsrA rJ r rB r+ rO r� r� r r �filesZfopen�readr� r r rH rI r� r- r �len�any�_get_cloud_environment�suffixes�storage_endpointZyamlZ safe_loadrd � setdefaultr� r� rW � startswith�splitZnoner �osr� �exists�cloudZgen_keysre Z accept_keyr �endswith�rfind� exceptionrU r� r� r� rx rr r. � create_or_update_vmextensionr r r/ r0 )Dr� r ri r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r; Z iface_datar� r� Z disk_nameZ vm_usernameZssh_publickeyfile_contentsr� Zspkc_r5 r� Z os_kwargsr� Z sshpublickeyZsshconfigurationZlinuxconfigurationZ vm_passwordZ complexityr� rL Zstorage_endpoint_suffixr� r r ZlunsZvolumeZimg_refZ source_imageZimg_pubZimg_offZimg_skuZimg_verr r� r r r Zfh_r Zkey_idZcustom_extensionr` r r Zcommand_prefixr �paramsZ vm_create� vm_resultr r r �request_instance s� � � � � �� � �� ������������ � � � �� ��� � � � ������ � � � � � � ��� � �������� �� � ����r< c C s� z| d rt jtt� p d| d | d�du rW dS W n ty" Y nw | �d�du r.d| d< td d d d �| d �td d| g d��td td d� td | d | d d| d � | �d�sft| d�| d< t � d| d | d � t | d�}|r|d|v r�d�| d t |��}t � |� t|��dd� }z)tjjj|| d | d ft jd | td d �t jd | td d �t jd!| td"d �d#�}W n" tttfy� } zz t �|� W tt |���tt |���d}~ww || d$< | �d%�s�t �d%| t�| d%< t �d&| t�| d'< td( | t�}t| d d)d*�}t � d+| d � t �d,| d t�|�� |�|� td d d-d.�| d �td d/| g d��td td d� |S )0z. Create a single VM from a data dict. r# r )r� F�bootstrap_interfaceN�publicr r zstarting createzsalt/cloud/{}/creatingr[ r! Zcreatingr" r$ r% r& zcloud.cachedir_index_addr� rN r� zCreating Cloud VM %s in %sr� zError creating VM {}! ({})c S sP t | dd�}|s dS d}|dkr|d d }|dkr |d d }|du r&dS |S ) z" Query node data. rP �r1 FNr> r� r Zprivater� )r� )r[ r= rf r� r r r �_query_node_dataJ s z create.<locals>._query_node_dataZwait_for_ip_timeoutiX )r� Zwait_for_ip_interval� Z wait_for_ip_interval_multiplierr� )Z update_args�timeout�intervalZinterval_multiplierZssh_hostr� r� r@ zcloud.bootstraprP r? zCreated Cloud VM '%s'z '%s' VM creation details: %szcreated instancezsalt/cloud/{}/createdZcreated) rA Zis_profile_configuredrB r r r+ r� r� rO r� �infor< r- r� r r r r5 Z wait_for_iprJ r r r� r� r� �pprintZpformatrK )r� Z vm_requestZ err_messager@ rf r5 r r r r �create s� ���� � �� � ���� � � �� � �� rF c C s� |du ri }|dkrt d��tdd�}t| dd�}|d d �d �r-|d d d d }n |d d d d }| i i}t�d� |j�|d | �}|�� t �dd�du rct d | t � � d�d t � t jdt� t ddd�}|�r:|�dt jdt� t ddd��} | r�t�d� |� d�} | d } | d } || | | d �|| d < |�d �r�t| | d!�dd"�|| d#< nt|d | | d$�dd"�|| d#< |�d%t jd&t� t ddd��} | �r:t�d'� i || d(< |d d( D ]T}|�d i ��d �p�|�d i ��d �}|� d�} | d } | d } || | | d �|| d( |d) < |�d ��r)t| | d!�dd"�|| d#< q�t|d | | d$�dd"�|| d#< q�t jd*t� t ddd�}|�r}||d g d+�|| d,< |d- d. }|D ] }|d � d�d/ }|| d, d# �t||d) d0�dd"�� �q\|S )1z� Destroy a VM. CLI Examples: .. code-block:: bash salt-cloud -d myminion salt-cloud -a destroy myminion service_name=myservice Nr zEThe destroy action must be called with -d, --destroy, -a or --action.rY r% rP r? r� r� Z managed_diskr} r r� z Deleting VMr� Zupdate_cachedirFTz cloud.delete_minion_cachedir�:r � cleanup_disksr� � delete_vhd� cleanup_vhdsz Deleting vhdr ���r� )Z delete_disksrI � container�blobZ delete_diskr )rL rM )r r1 rf )r� rL rM Zdelete_data_disks�cleanup_data_diskszDeleting data_disksr r[ �cleanup_interfaces)rO r� rf Zcleanup_networkr� r� � )r� r� )r r) r� r+ r� r� r� r� rx rB r� r r2 rA rJ r r1 � delete_blob�delete_managed_diskrW r� )r[ r1 r ri Z node_datar r rr rH rJ �compsrL rM rN ZdiskZdatavhdrO Zifacesr� r� r r r �destroy� s� � ���� � ��� �� �� � ��� �� ��� rT c C s� | dkrt d��tdd�}i }z|j�� }tjj�|�}|D ]}|||d < q W |S tyJ } ztjj� d|j � d|j i}W Y d}~|S d}~ww )z8 List storage accounts within the subscription. rP zGThe list_storage_accounts function must be called with -f or --function�storager% r[ r9 N) r r) �storage_accountsrd r r r Zpaged_object_to_listr r/ r0 )r1 �storconnr Zaccounts_queryZaccountsZaccountr5 r r r �list_storage_accounts2 s& � ����rX c C sT t jdt� tdd�} zt�d�}t|| pd�}W |S ttfy) t d� | ���w )z+ Get the cloud environment object. rG FrE zmsrestazure.azure_cloudZAZURE_PUBLIC_CLOUDz0The azure {} cloud environment is not available.) rA rJ r rB � importlib� import_moduler� r � ImportErrorr r� )rG Zcloud_env_modulerL r r r r- K s � ���r- c C s� | � d�ptjdt� tdd�}| � d�p tjdt� tdd�}| � d�p,tjdt� tdd�}| � d�p;tjdt� tdd�}|sBtd��|sHtd��|sftd d �}|j�||�}d d � |j D �}t t |� � ��}t � }|jj}t||||d �S )z- Get the block blob storage service. r� FrE � sas_tokenr � storage_key�"A resource group must be specifiedz#A storage account must be specifiedrU r% c S s i | ]}|j |j�qS r )Zkey_namer )rq rt r r r ru y s z+_get_block_blob_service.<locals>.<dictcomp>)r\ �endpoint_suffix)r+ rA rJ r rB r r) rV Z list_keysr� �next�iter�valuesr- r. r/ r ) r r� r\ r r] rW Z storage_keysrL r_ r r r �_get_block_blob_service] s> � � � � ��rc c C s� |du ri }d|vrt d��t|�}i }z |�|d �jD ]}|jj|jj�� |jjd�||j < q W |S t yM } zt � t |�� W Y d}~|S d}~ww )z List blobs. NrL � A container must be specified)� blob_type� last_modified�server_encrypted)r rc � list_blobsro Z propertiesre rf � isoformatrg r[ r� r� r� r- )r1 r �storageservicer rM r5 r r r rh � s& �� ���rh c C sL |du ri }d|vrt d��d|vrt d��t|�}|�|d |d � dS )z) Delete a blob from a container. NrL rd rM zA blob must be specifiedT)r rc rQ )r1 r rj r r r rQ � s rQ c C sf t dd�}z|j�|d |d � W dS ty2 } zt�d|�d�t|�� W Y d}~dS d}~ww ) z6 Delete a managed disk from a resource group. rY r% r� rM z#Error deleting managed disk %s - %sNFT)r) Zdisksr� r� r� r� r+ r- )r1 r ri r5 r r r rR � s ����rR c C s� |du ri }| dkrt d��tdd�}t� }i }|D ]6}z |jj|d�}W n ty0 i }Y nw |D ] }|�� }|||d < t||d d�d �||d d < q3q|S ) z List virtual networks. NrP �=The avail_sizes function must be called with -f or --functionr� r% r� r[ )r� r� r� r� )r r) r� Zvirtual_networksrd r r. � list_subnets) r1 r r� r� r r� ZnetworksZ network_objr� r r r �list_virtual_networks� s. � � ��rm c C sJ |du ri }| dkrt d��tdd�}|�d�p!tjdt� tdd�}|s0d |v r0d|vr0|d }|s6t d ��|�d�du rItjdt� tdd�|d< d|vsS|d du rWt d ��i }|j�||d �}|D ]>}|� � ||j < i ||j d < |j D ]#}|j � d �}|d } |� � ||j d | < |j ||j d | d< qw|||j d< qd|S )z, List subnets in a virtual network. NrP rk r� r% r� FrE r� r^ z A "network" must be specifiedr� r r� r� )r r) r+ rA rJ r rB r� rd r. r[ r� r} r2 ) r1 r r� r� r r� r� r� rS r[ r r r rl � s@ � � � rl c C s� |du ri }d|vrt d��d|vrt d��tdd�}ttd�}|�d �p.tjd t� td d �}|s5t d ��|�d �p<t � }|sCt d��|�dd�}|�dd�}|�dd�}|�dd�} |�di �} |�d�} t | t �sot d��d| vr{d| vr{t d ��t � d |d � i } z#|||||| | | d �} |j�||d |d | �}|�� } | �� } W | S ty� } ztjj�dd �|j�� d |ji} W Y d}~| S d}~ww )!a� .. versionadded:: 2019.2.0 Create or update a VM extension object "inside" of a VM object. required kwargs: .. code-block:: yaml extension_name: myvmextension virtual_machine_name: myvm settings: {"commandToExecute": "hostname"} optional kwargs: .. code-block:: yaml resource_group: < inferred from cloud configs > location: < inferred from cloud configs > publisher: < default: Microsoft.Azure.Extensions > virtual_machine_extension_type: < default: CustomScript > type_handler_version: < default: 2.0 > auto_upgrade_minor_version: < default: True > protected_settings: < default: None > Nr z#An extension name must be specifiedr z(A virtual machine name must be specifiedrY r% �VirtualMachineExtensionr� FrE r^ rN z A location must be specifiedr` r r r r r r Tr r z#VM extension settings are not validr �scriptzYVM extension settings are not valid. Either commandToExecute or script must be specified.zCreating VM extension %s)rN r` r r r r r z/Error attempting to create the VM extension: {}r� )r r) r� r( r+ rA rJ r rB rO r� r� r� rD Zvirtual_machine_extensionsr� rr r. r r r r r/ r� r0 )r1 r ri rn r� rN r` r r r r r r r: r� r5 r r r r9 sx � � �� � � ���r9 c C �^ |dkrt d��tdd�}tjdt� tdd�}i }|srt� }|D ];}z|jj| |d�}|� � |� � }|� � }W n" t y[ } zd | j v rLW Y d } ~ q d | j i}W Y d } ~ q d } ~ ww |sptjj�dd �| �� d d �| �i}|S z|jj| |d�}|� � |� � }|� � }W |S t y� } ztjj�dd �| | j �� d | j i}W Y d } ~ |S d } ~ ww )z� .. versionadded:: 2019.2.0 Stop (deallocate) a VM CLI Examples: .. code-block:: bash salt-cloud -a stop myminion r z3The stop action must be called with -a or --action.rY r% r� FrE �r' r� � was not foundNr� �,Unable to find virtual machine with name: {}z Error attempting to stop {}: {})r r) rA rJ r rB r� r� Z deallocaterx rr r. r r0 r r r r/ r� � r[ r1 ri r� r r� r� �instancer; r5 r r r �stop� s\ �� �� ��� ����rv c C rp )z� .. versionadded:: 2019.2.0 Start a VM CLI Examples: .. code-block:: bash salt-cloud -a start myminion r z4The start action must be called with -a or --action.rY r% r� FrE rq rr Nr� rs z Error attempting to start {}: {})r r) rA rJ r rB r� r� �startrx rr r. r r0 r r r r/ r� rt r r r rw � sb � �� �� ��� � ���rw )NNr )R�__doc__rY Zloggingr3 Zos.pathrE r� r� � functoolsr Zmultiprocessingr Zmultiprocessing.poolr Z salt.cacher Z salt.configrA Z salt.loaderZsalt.utils.azurearmZsalt.utils.cloudZsalt.utils.filesZsalt.utils.stringutilsZsalt.utils.yamlZ salt.versionZsalt.exceptionsr r r r rD Zazure.mgmt.compute.modelsZmgmtrY Zmodelsr( Zazure.mgmt.network.modelsr� r� Z#azure.storage.blob.blockblobservicer Z msrestazure.azure_exceptionsr r[ r Z getLogger�__name__r� r r r! r6 r: r r r) rO rX rz r| r� r� r� r� r� r� r� r� r< rF rT rX r- rc rh rQ rR rm rl r9 rv rw r r r r �<module> s� a � / ! P U $! 1 v ! + . �e <
Save