File manager - Edit - /opt/saltstack/salt/lib/python3.10/site-packages/libcloud/compute/drivers/__pycache__/libvirt_driver.cpython-310.pyc
Back
o ;j9 � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dl m Z d dl m Z m Z d dlmZmZmZ d dlmZ d dlmZ zd dlZdZW n eyU d ZY nw G d d � d e�ZdS ) � N)�join)� defaultdict)�ET� ensure_string)�Node� NodeState� NodeDriver)�Provider)�is_public_subnetTFc @ s e Zd ZdZejZdZdZe j e j e j e j e j e j e j e j d�Zd=dd�Zdd � Zd d � Zd d � Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd>d!d"�Z d#d$� Z d%d&� Z d'd(� Z d)d*� Z d+d,� Z!d-d.� Z"d/d0� Z#d1d2� Z$d3d4� Z%d5d6� Z&d7d8� Z'd9d:� Z(d;d<� Z)dS )?�LibvirtNodeDriverzv Libvirt (http://libvirt.org/) node driver. To enable debug mode, set LIBVIR_DEBUG environment variable. ZLibvirtzhttp://libvirt.org/)r � � � � � � � Nc C s� t std��|| _|| _|| _|dur7d| jv r7|du r$|du r$td��tjtjg| jdg}t� ||d�| _ nt� |�| _ |du rI| j � � | _dS dS )aJ :param uri: Hypervisor URI (e.g. vbox:///session, qemu:///system, etc.). :type uri: ``str`` :param key: the username for a remote libvirtd server :type key: ``str`` :param secret: the password for a remote libvirtd server :type key: ``str`` z0Libvirt driver requires 'libvirt' Python packageNz+tcpz]The remote Libvirt instance requires authentication, please set 'key' and 'secret' parametersr ) � have_libvirt� RuntimeError�_uri�_key�_secret�libvirt�VIR_CRED_AUTHNAME�VIR_CRED_PASSPHRASE�_cred_callbackZopenAuth� connection�openZgetInfo)�selfZuri�keyZsecretZauth� r �[/opt/saltstack/salt/lib/python3.10/site-packages/libcloud/compute/drivers/libvirt_driver.py�__init__< s& � � �zLibvirtNodeDriver.__init__c C s@ |D ]}|d t jkr| j|d< q|d t jkr | j|d< qdS )a� Callback for the authentication scheme, which will provide username and password for the login. Reference: ( http://bit.ly/1U5yyQg ) :param cred: The credentials requested and the return :type cred: ``list`` :param user_data: Custom data provided to the authentication routine :type user_data: ``list`` :rtype: ``int`` r r )r r r r r )r ZcredZ user_dataZ credentialr r r! r ` s �z LibvirtNodeDriver._cred_callbackc C s | j �� }| j|d�}|S )N)�domains)r ZlistAllDomains� _to_nodes�r r# Znodesr r r! � list_nodest s z LibvirtNodeDriver.list_nodesc C s | j |d�}|jdd�dkS )N��noder )�flags)�_get_domain_for_nodeZreboot�r r( �domainr r r! � reboot_nodey s z LibvirtNodeDriver.reboot_nodec C � | j |d�}|�� dkS �Nr' r )r* Zdestroyr+ r r r! � destroy_node} � z LibvirtNodeDriver.destroy_nodec C r. r/ )r* �creater+ r r r! � start_node� r1 z LibvirtNodeDriver.start_nodec C r. r/ )r* �shutdownr+ r r r! � stop_node� r1 zLibvirtNodeDriver.stop_nodec C � | j |d�S )z� Start a stopped node. :param node: Node which should be used :type node: :class:`Node` :rtype: ``bool`` r' )r3 �r r( r r r! � ex_start_node� s z LibvirtNodeDriver.ex_start_nodec C r6 )z� Shutdown a running node. Note: Usually this will result in sending an ACPI event to the node. :param node: Node which should be used :type node: :class:`Node` :rtype: ``bool`` r' )r5 r7 r r r! �ex_shutdown_node� s z"LibvirtNodeDriver.ex_shutdown_nodec C r. )z� Suspend a running node. :param node: Node which should be used :type node: :class:`Node` :rtype: ``bool`` r' r )r* Zsuspendr+ r r r! �ex_suspend_node� � z!LibvirtNodeDriver.ex_suspend_nodec C r. )z� Resume a suspended node. :param node: Node which should be used :type node: :class:`Node` :rtype: ``bool`` r' r )r* Zresumer+ r r r! �ex_resume_node� r; z LibvirtNodeDriver.ex_resume_nodec C � | j |d�}| j|d�}|S )z� Retrieve Node object for a domain with a provided uuid. :param uuid: Uuid of the domain. :type uuid: ``str`` )�uuid�r, )�_get_domain_for_uuid�_to_node)r r> r, r( r r r! �ex_get_node_by_uuid� � z%LibvirtNodeDriver.ex_get_node_by_uuidc C r= )z� Retrieve Node object for a domain with a provided name. :param name: Name of the domain. :type name: ``str`` )�namer? )�_get_domain_for_namerA )r rD r, r( r r r! �ex_get_node_by_name� rC z%LibvirtNodeDriver.ex_get_node_by_namer c s� t j�|�r t j�|�std��| j|d�}| j�� }|j|dd�}t j |d�}|r/|d }nd}d� t t � � �|�} t|| �} t| d��� � fd d �} |�| d � W d � n1 s\w Y z|�� W | S tyr Y | S w ) a� Take a screenshot of a monitoring of a running instance. :param node: Node to take the screenshot of. :type node: :class:`libcloud.compute.base.Node` :param directory: Path where the screenshot will be saved. :type directory: ``str`` :param screen: ID of the monitor to take the screenshot of. :type screen: ``int`` :return: Full path where the screenshot has been saved. :rtype: ``str`` z$Invalid value for directory argumentr' r )�stream�screen)�typez.pngzscreenshot-{}{}�wbc s � � |� d S )N)�write)rG �bufZopaque��fpr r! rK � s z8LibvirtNodeDriver.ex_take_node_screenshot.<locals>.writeN)�os�path�exists�isdir� ValueErrorr* r Z newStreamZ screenshot� mimetypesZguess_all_extensions�format�int�time�pjoinr ZrecvAllZfinish� Exception) r r( � directoryrH r, rG Z mime_type� extensions� extensionrD Z file_pathrK r rM r! �ex_take_node_screenshot� s, � ��z)LibvirtNodeDriver.ex_take_node_screenshotc C s | j �� }|S )zN Return a system hostname on which the hypervisor is running. )r Z getHostname)r �hostnamer r r! �ex_get_hypervisor_hostname s z,LibvirtNodeDriver.ex_get_hypervisor_hostnamec C sL | j �� }t�|�}g d�}i }|D ]}|�|�}| j|d�}|||< q|S )zS Retrieve hypervisor system information. :rtype: ``dict`` )Zbios�system� processorZ memory_device)�element)r Z getSysinfor �XML�find� _get_entries)r �xml�etreeZ attributesZsysinfoZ attributerb �entriesr r r! �ex_get_hypervisor_sysinfo s z+LibvirtNodeDriver.ex_get_hypervisor_sysinfoc s � fdd�|D �}|S )Nc s g | ]}� j |d ��qS )r? )rA )�.0r, �r r r! � <listcomp> s z/LibvirtNodeDriver._to_nodes.<locals>.<listcomp>r r% r rk r! r$ s zLibvirtNodeDriver._to_nodesc C s� |� � \}}}}}| j�|tj�}g g }}| �|�} | D ]} t| �r)|�| � q |�| � q |�� |� � | j � � |d ||d�} t |� � |�� |||| | d�} |�� | _| S )Ni )r> Zos_type�typesZ used_memory� vcpu_count� used_cpu_time)�idrD �state� public_ips� private_ipsZdriver�extra)�info�NODE_STATE_MAP�getr �UNKNOWN� _get_ip_addresses_for_domainr �appendZ UUIDStringZOSTyper ZgetTyper �IDrD Z_uuid) r r, rq Zmax_memZmemoryrn ro rr rs � ip_addresses� ip_addressrt r( r r r! rA " s4 � � zLibvirtNodeDriver._to_nodec C s� g }t �� dkr |S d| jvr|S | j|d�}i }z ddg}tj|tjtjd�}|�� \}}| j|d�}W n1 t yf } z%| j dkr\d d g}tj|tjtjd�}|�� \}}| j |d �}W Y d } ~ nd } ~ ww |D ]} | |v rx|| } |� | � qi|S ) a2 Retrieve IP addresses for the provided domain. Note: This functionality is currently only supported on Linux and only works if this code is run on the same machine as the VMs run on. :return: IP addresses for the provided domain. :rtype: ``list`` �Linuxz///r? Zarpz-an)�stdout�stderr)� arp_outputr �ipZneigh)� ip_outputN) �platformr` r � _get_mac_addresses_for_domain� subprocess�Popen�PIPE� communicate�_parse_ip_table_arp�OSError�errno�_parse_ip_table_neigh�extend) r r, �resultZ mac_addresses� arp_table�cmd�childr �_�e� mac_addressr| r r r! ry E s4 �� �z.LibvirtNodeDriver._get_ip_addresses_for_domainc C sB |� � }t�|�}|�d�}g }|D ] }|�d�}|�|� q|S )zR Parses network interface MAC addresses from the provided domain. z&devices/interface[@type='network']/mac�address)ZXMLDescr rc �findallrw rz )r r, rf rg Zelemsr� �elemr� r r r! r� p s z/LibvirtNodeDriver._get_mac_addresses_for_domainc C s | j �|j�}|S )zE Return libvirt domain object for the provided node. )r �lookupByUUIDStringr> r+ r r r! r* s z&LibvirtNodeDriver._get_domain_for_nodec C � | j �|�}|S )zE Return libvirt domain object for the provided uuid. )r r� )r r> r, r r r! r@ � � z&LibvirtNodeDriver._get_domain_for_uuidc C r� )zE Return libvirt domain object for the provided name. )r Z lookupByName)r rD r, r r r! rE � r� z&LibvirtNodeDriver._get_domain_for_namec C s4 |� d�}i }|D ]}|�d�}|j}|||< q |S )zE Parse entries dictionary. :rtype: ``dict`` �entryrD )r� rw �text)r rb �elementsr� rD �valuer r r! re � s z LibvirtNodeDriver._get_entriesc C � t �d�}| �||�S )z� Sets up the regexp for parsing out IP addresses from the 'arp -an' command and pass it along to the parser function. :return: Dictionary from the parsing function :rtype: ``dict`` z.*?\((.*?)\) at (.*?)\s+��re�compile�_parse_mac_addr_table)r r� Z arp_regexr r r! r� � � z%LibvirtNodeDriver._parse_ip_table_arpc C r� )z� Sets up the regexp for parsing out IP addresses from the 'ip neighbor' command and pass it along to the parser function. :return: Dictionary from the parsing function :rtype: ``dict`` z(.*?)\s+.*lladdr\s+(.*?)\s+r� )r r� Zip_regexr r r! r� � r� z'LibvirtNodeDriver._parse_ip_table_neighc C sZ t |��d�}tt�}|D ] }|�|�}|sq |�� }|d }|d } || �|� q |S )z� Parse the command output and return a dictionary which maps mac address to an IP address. :return: Dictionary which maps mac address to IP address. :rtype: ``dict`` � r r )r �splitr �list�match�groupsrz ) r Z cmd_outputZ mac_regex�linesr� �liner� r� r} r� r r r! r� � s z'LibvirtNodeDriver._parse_mac_addr_table)NN)r )*�__name__� __module__� __qualname__�__doc__r ZLIBVIRTrI rD Zwebsiter Z TERMINATEDZRUNNINGZPENDINGrx rv r" r r& r- r0 r3 r5 r8 r9 r: r< rB rF r] r_ ri r$ rA ry r� r* r@ rE re r� r� r� r r r r! r & sR � $ /#+ r )rO r� rW r� rT r� Zos.pathr rX � collectionsr Zlibcloud.utils.py3r r Zlibcloud.compute.baser r r Zlibcloud.compute.typesr Zlibcloud.utils.networkingr r r � ImportErrorr r r r r! �<module> s& �
| ver. 1.4 |
Github
|
.
| PHP 8.2.30 | Generation time: 0 |
proxy
|
phpinfo
|
Settings