File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/runners/__pycache__/virt.cpython-310.pyc
o
�N�g!E � @ s� d Z ddlZddlZddlZddlZddlZddlZddl Zddl
m Z e� e
�Zd*dd�Zd+dd �Zd,d
d �Zd-d d
�Zdd� Zd.dd�Z d/dd�Zd+dd�Zd d � Zd d � Zd d!� Zd0d"d#�Zd$d%� Zd&d'� Z d*d(d)�Z dS )1z#
Control virtual machines via Salt
� N)�SaltClientError� c C sP d}d}| � � D ] \}}||krqt|t�sq|�dd�|kr%|d }|}q|S )zO
Determine what the most resource free host is based on the given data
r r Zfreemem)�items�
isinstance�dict�get)�dataZomit�hostZbestmem�hv_�comps� r �E/opt/saltstack/salt/lib/python3.10/site-packages/salt/runners/virt.py�_determine_host s
�r Fc C sh |D ]/}t || t�s q| || �di �v r1|| || d | ii}|s-t�|dd�d� | S qi S )z.
Scan the query data for the named VM
�vm_info�nested�r Z outputter�progress)r r r �
__jid_event__�
fire_event)�namer �quietr
�retr r r
�_find_vm% s
��r c C s( |rt �d� i }tj�td ��y}zX|jdddd�D ]M}|s qt|t�s&qi }t t
|� � ��}| r7| |kr7qt|| t�s?qd|| vrFqt|| d t�sPq|| d ||< |� |� |sht
�|dd �d
� qW n ty� } z
t|� W Y d }~nd }~ww |W d � S 1 s�w Y d S ) z�
Query the virtual machines. When called without options all hosts
are detected and a full query is returned. A single host can be
passed in to specify an individual host to query.
�0'quiet' is deprecated. Please migrate to --quiet� conf_file�virtual:physicalzvirt.full_info�grain�Ztgt_typer Z
virt_queryr r N)�log�warning�salt�client�get_local_client�__opts__�cmd_iterr r �next�iter�keys�updater r r �print)r r r r! �info�chunk�id_� client_errorr r r
�query7 sJ
�
�����$�r. c C s: |rt �d� i }tj�td ���}|jdddd�D ]o}|s qt|t�s%qi }t t
|� � ��}| r6| |kr6qt|| t�s>qd|| vrEqt|| d t�sOqi }|| d � � D ]\} }
|
d |v rm||
d �
| � qY| g||
d < qY|||< |�|� |s�t�|d d
�d � q|W d � S 1 s�w Y d S )
z�
List the virtual machines on each host, this is a simplified query,
showing only the virtual machine names belonging to each host.
A single host can be passed in to specify an individual host
to list.
r r r z virt.vm_infor r r �stater r r N)r r r r! r" r# r$ r r r% r& r' r �appendr( r r ) r r Zhyperr r! r* r+ r, r �key�valr r r
�list_ sH
�
��$�r3 c C s t tdd��} t| � | S )z�
Return the host to use for the next autodeployed VM. This queries
the available host and executes some math the determine the most
"available" next host.
T�r )r r. r) )r r r r
� next_host� s r5 c C sF t | dd�}|D ]}d|| v r|| �d� qt�|dd�d� |S )zD
Return information about the host connected to this master
Tr4 r r r r )r. �popr r )r r r, r r r
� host_info� s �r7 �kvmT�default�base�
seed.apply�Nonec C st t �ddid� t|dd�}|D ] }d|| v r/| || d v r/t �dd| � d�id� d S q|d
u r8t|�}||vs>|sLt �dd |� d �id� d S d
}d
}|r�t �dd
id� tjj�� \}}tj �
t d d| �}tjj �
|d��}|�tjj�|�� W d
� n1 s�w Y tj�t d ���}t �dd| � d|� �id� z |j|d| ||gd|||| |
| |||||
| ||d�d�}W n ty� } z
t|� W Y d
}~nd
}~ww t|�}|s�t �dd| � d�id� W d
� d S |D ] }|| d du �rtd �| || d �� W d
� d S q�t �dd| � d |� �id� W d
� d S 1 �s3w Y d
S ) aX
This routine is used to create a new virtual machine. This routines takes
a number of options to determine what the newly created virtual machine
will look like.
name
The mandatory name of the new virtual machine. The name option is
also the minion id, all minions must have an id.
cpu
The number of cpus to allocate to this new virtual machine.
mem
The amount of memory to allocate to this virtual machine. The number
is interpreted in megabytes.
image
The network location of the virtual machine image, commonly a location
on the salt fileserver, but http, https and ftp can also be used.
hypervisor
The hypervisor to use for the new virtual machine. Default is `kvm`.
host
The host to use for the new virtual machine, if this is omitted
Salt will automatically detect what host to use.
seed
Set to `False` to prevent Salt from seeding the new virtual machine.
nic
The nic profile to use, defaults to the "default" nic profile which
assumes a single network interface per VM associated with the "br0"
bridge on the master.
install
Set to False to prevent Salt from installing a minion on the new VM
before it spins up.
disk
The disk profile to use
saltenv
The Salt environment to use
enable_vnc
Whether a VNC screen is attached to resulting VM. Default is `False`.
seed_cmd
If seed is `True`, use this execution module function to seed new VM.
Default is `seed.apply`.
enable_qcow
Clone disk image as a copy-on-write qcow2 image, using downloaded
`image` as backing file.
serial_type
Enable serial console. Set to 'pty' for serial console or 'tcp' for
telnet.
Default is 'None'
�messagezSearching for hostsr Tr4 r �Virtual machine z is already deployed�failNzHost z was not foundzMinion will be preseededZpki_dirZminions�wr z Creating VM z on host z virt.init�X )�image�nic�
hypervisor�start�disk�saltenv�seed�install�pub_key�priv_key�seed_cmd�
enable_vnc� enable_qcow� serial_type)�timeout�kwarg�VM z was not initialized.r Fz/VM {} initialization failed. Returned error: {}z initialized on host �good)r r r. r r �utilsZcloudZgen_keys�os�path�joinr# �filesZfopen�writeZ stringutilsZto_strr! r" r$ r r) r% �format)r ZcpuZmemrB rD r rH rC rI rE rF rG rM rL rN rO r �noderJ rK Z accepted_keyZfp_r! �cmd_retr- r Z minion_idr r r
�init� s� O �� ���
�����& ���'�&�r] c C s t dd�}t| ||�S )z0
Return the information on the named VM
Tr4 )r. r )r r r r r r
r C s
r c C s� i }t j�td ��g}t| dd�}|s(t�dd| � d�id� W d � d S tt|� � ��}z!|j
|d
| gd d �}|D ]}|� |� q=t�dd
| � �id� W n t yg } z
t
|� W Y d}~nd}~ww |W d � S 1 stw Y dS )z5
Force power down and restart an existing VM
r Tr4 r= �Failed to find VM z to resetr Nr? z
virt.resetrA �rP z Reset VM )r r! r" r# r r r r% r& r' r$ r( r r) )r r r! r r r\ �compr- r r r
�resetK s* �� ��$�ra c C sB i }t j�td ���}t| dd�}|s(t�dd| � d�id� W d � d S tt|� � ��}|| | d
d krKt
d | � d
�� W d � dS z |j |d| gdd�}W n! t yx } zd| � d|� �W Y d}~W d � S d}~ww |D ]}|�
|� q{t�dd| � �id� W d � dS 1 s�w Y dS )z'
Start a named virtual machine
r Tr4 r= r^ z to startr Nr? r/ ZrunningrR z is already running� bad statez
virt.startrA r_ r> z not started: z Started VM rS )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
rE b s4 ��� ��
$�rE c C s6 i }t j�td ���}t| dd�}|s$td| � d�� W d � dS tt|�� ��}|| | d d krGtd
| � d �� W d � d S z |j |d
| gdd�}W n t
yr } zd� | |�W Y d}~W d � S d}~ww |D ]}|� |� qut
�dd| � �id� W d � dS 1 s�w Y dS )z4
Force power down the named virtual machine
r Tr4 r^ z to destroyNr? r/ �shutdownrR z is already shutdownrb z virt.stoprA r_ z.Virtual machine {} could not be forced off: {}r= zPowered off VM r rS )r r! r" r# r r) r% r& r' r$ r rZ r( r r rc r r r
� force_off| s4 �� ��� $�re c
C sP i }t j�td ���}t| dd�}|s(t�dd| � d�id� W d � d S tt|� � ��}z
|j
|d
| dgd d �}W n t y\ } zd
� | |�W Y d}~W d � S d}~ww |D ]}|�
|� q_|r�t�d| � t j�t��
} | �| � W d � n1 s�w Y t�dd| � �id� W d � dS 1 s�w Y dS )z
Destroy the named VM
r Tr4 �errorr^ z to purger Nr? z
virt.purgerA r_ z*Virtual machine {} could not be purged: {}zDeleting key %sr= z
Purged VM rS )r r! r" r# r r r r% r& r' r$ r rZ r( r �debugr1 ZKey�
delete_key)
r rh r r! r r r\ r- r` Zskeyr r r
�purge� s: �����
�$�ri c C sF i }t j�td ���}t| dd�}|s(t�dd| � d�id� W d � d S tt|� � ��}|| | d
d krOt�dd | � d
�id� W d � dS z |j
|d| gdd�}W n t yz } zd� | |�W Y d}~W d � S d}~ww |D ]}|�
|� q}t�dd| � �id� W d � dS 1 s�w Y dS )z
Pause the named VM
r Tr4 rf r^ � to pauser Nr? r/ �pausedrR z is already pausedrb z
virt.pauserA r_ z*Virtual machine {} could not be pasued: {}r= z
Paused VM rS �r r! r" r# r r r r% r& r' r$ r rZ r( rc r r r
�pause� s< ������� $�rm c C sF i }t j�td ���}t| dd�}|s(t�dd| � d�id� W d � d S tt|� � ��}|| | d
d krOt�dd | � d
�id� W d � dS z |j
|d| gdd�}W n t yz } zd� | |�W Y d}~W d � S d}~ww |D ]}|�
|� q}t�dd| � �id� W d � dS 1 s�w Y dS )z
Resume a paused VM
r Tr4 rf r^ rj r Nz not foundr/ rk rR z is not pausedrb z virt.resumerA r_ z+Virtual machine {} could not be resumed: {}r= z Resumed VM rS rl rc r r r
�resume� s8 ��� ���
$�rn c
C s� t j�td ���}tdd�}t| |dd�}ztt|��}W n ty9 t �
dd| � d�id� Y W d � d S w || | d
}|sYt �
dd| � d�id� W d � d S |s`t ||�}||vryt �
dd |� d �id� W d � d S z|� |d
|dg� |j
|d| |gtd�}W n ty� } zd�| |�W Y d}~W d � S d}~ww d�| ||�} t �
d| id� W d � dS 1 s�w Y dS )z�
Migrate a VM from one host to another. This routine will just start
the migration and display information on how to look up the progress.
r Tr4 rf z Named VM z was not found to migrater Nr �disksz Target host z
not foundz virt.seed_non_shared_migratez virt.migrate)Z copy_storagez,Virtual machine {} could not be migrated: {}z�The migration of virtual machine {} to host {} has begun, and can be tracked via jid {}. The ``salt-run virt.query`` runner can also be used, the target VM will be shown as paused until the migration is complete.r= )r r! r" r# r. r r% r& �
StopIterationr r r �cmdZ cmd_async�allr rZ )
r �targetr! r Z origin_dataZ origin_hostro Zjidr- �msgr r r
�migrate� s\
����
��
����
�"�ru )r )F)NF)NFN)N) r8 NTr9 TTr9 r: Fr; Fr< )T) �__doc__�loggingZos.pathrU Z salt.clientr Zsalt.keyZsalt.utils.cloudZsalt.utils.filesZsalt.utils.stringutilsZsalt.exceptionsr � getLogger�__name__r r r r. r3 r5 r7 r] r ra rE re ri rm rn ru r r r r
�<module> sL
(-
�
!