File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/grains/__pycache__/smartos.cpython-310.pyc
o
�N�g � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddl Zddl
ZdZ dej j
jiZe�e�Zdd� Zdd� Zd d
� Zd d � Zd
d� Zdd� ZdS )z�
SmartOS grain provider
:maintainer: Jorge Schrauwen <sjorge@blackdot.be>
:maturity: new
:depends: salt.utils, salt.module.cmdmod
:platform: SmartOS
.. versionadded:: 2017.7.0
� N�smartos�cmd.runc C s t jj�� rtS dS )z*
Only load when we are on SmartOS
F)�salt�utils�platformZ
is_smartos�__virtualname__� r r �G/opt/saltstack/salt/lib/python3.10/site-packages/salt/grains/smartos.py� __virtual__ s r
c C s� i } i }t d d��d�D ] }tttg d�|�d����}|||d < ||d d= q
t|�| d< d| d < d| d
< dddd �| d < |D ]I}|| d
�� dkrX| d d7 < n|| d
�� dkrj| d
d7 < || d | d vr~d| d || d < | d || d d7 < qCtjj �
t d d��}|d | d< |d | d< |d r�|d | d< |d | d< |d | d < |d | d< | S ) z?
Return useful information from a SmartOS compute node
r z&vmadm list -p -o uuid,alias,state,type�
)�uuid�alias�state�type�:r Zcomputenode_vms_totalr Zcomputenode_vms_runningZcomputenode_vms_stopped)ZKVMZLXZOSZcomputenode_vms_typer Zrunning� �stoppedr �sysinfoz SDC VersionZcomputenode_sdc_versionz
VM CapableZcomputenode_vm_capablezCPU VirtualizationZcomputenode_vm_hw_virtZ ManufacturerZ manufacturerZProductZ productnameZUUID) �__salt__�split�dict�list�zip�len�lowerr r �json�loads)�grainsZvmsZvmr r r r �_smartos_computenode_data) s6
r c C s� i } t d d��� �d�}|d | d< |d | d< t�d�}d | d
< tj�d �r[tj j
� d d ��#}|D ]}tj j �
|�}|�|�}|rJ|�d�| d
< q3W d
� | S 1 sVw Y | S )z7
Return useful information from a SmartOS zone
r zzoneadm list -pr r Zzoneidr Zzonenamez Image:\s(.+)�Unknown� imageversionz /etc/product�rN)r �stripr �re�compile�os�path�isfiler r �files�fopen� stringutils�
to_unicode�match�group)r Zzoneinfor �fp_�liner, r r r �_smartos_zone_data^ s&
��
��r0 c C s� ddd�} t �d�}t �d�}ddg}|D ]M}tj�|�rbtjj�|d��5}|D ]*}tjj �
|�}|� |�}|rR|� d�| d < |� |� d��}|rP|� d�| d
< nq(W d � n1 s]w Y q| S ) z)
SmartOS zone pkgsrc information
r )�
pkgsrcversion�
pkgsrcpathz
PKG_PATH=(.+)z;^https?://pkgsrc.joyent.com/packages/SmartOS/(.+)/(.+)/All$z /opt/local/etc/pkg_install.confz /opt/tools/etc/pkg_install.confr! r r2 r1 N)
r# r$ r% r& r' r r r( r) r* r+ r, r- ) r r2 r1 Zpkg_install_pathsZ pkg_installr. r/ Zmatch_pkgsrcpathZmatch_pkgsrcversionr r r �_smartos_zone_pkgsrc_datay s: �
��
��� ��
r3 c C s� dg i} t �d�}ddg}|D ]:}tj�|�rItjj�|d��"}|D ]}tjj �
|�� � }|� |�r9| d �
|� q"W d � n1 sDw Y q| S )z(
SmartOS zone pkgin information
Zpkgin_repositoriesz ^(?:https|http|ftp|file)://.*$z&/opt/local/etc/pkgin/repositories.confz&/opt/tools/etc/pkgin/repositories.confr! N)r# r$ r% r& r' r r r( r) r* r+ r" r, �append)r Z pkginrepoZrepositories_pathZ repositoriesr. r/ r r r �_smartos_zone_pkgin_data� s$ �
�
�� ��r5 c C sz i } t jj�� rt jjj| t� dd�} nt jj�� r%t jjj| t� dd�} t jjj| t � dd�} t jjj| t
� dd�} | S )z$
Provide grains for SmartOS
T)Z merge_lists) r r r Zis_smartos_zoneZ
dictupdate�updater0 Zis_smartos_globalzoner r5 r3 )r r r r r � s � ���)�__doc__�loggingr% r# Zsalt.modules.cmdmodr Zsalt.utils.dictupdateZsalt.utils.jsonZsalt.utils.pathZsalt.utils.platformZsalt.utils.stringutilsr �modulesZcmdmod�runr � getLogger�__name__�logr
r r0 r3 r5 r r r r r �<module> s(
�
5.