File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/grains/__pycache__/disks.cpython-310.pyc
o
�N�g: � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZej j
j ej j
j ej j
j
d�Ze�e�Zdd� ZG dd� d�Zdd � Zd
d � ejD �Zd d
� Zdd� Zdd� ZdS )z
Detect disks
� N)�cmd.runz cmd.run_all�cmd.powershellc C sD t jj�� r t� S t jj�� rt� S t jj�� rt� S t �
d� dS )z%
Return list of disk devices
z Disk grain does not support OSN) �salt�utils�platformZ
is_freebsd�
_freebsd_geomZis_linux� _linux_disksZ
is_windows�_windows_disks�log�trace� r r �E/opt/saltstack/salt/lib/python3.10/site-packages/salt/grains/disks.py�disks s r c
@ s` e Zd ZdZdZdZdZdZdZdZ dZ
d Z d
Z ed e d e d
e diZ
edededede diZdS )� _geomconstsz Geom nameZ MediasizeZ
SectorsizeZ
StripesizeZ StripeoffsetZdescrZlunidZlunname�identZ rotationrateZ device_modelZ
serial_numberZ media_RPMZWWN)�re_intz(\d+)�try_intN)�__name__�
__module__� __qualname__�GEOMNAMEZ MEDIASIZEZ
SECTORSIZEZ
STRIPESIZEZ STRIPEOFFSETZDESCRZLUNIDZLUNNAMEZIDENT� ROTATIONRATE�_aliases�
_datatypesr r r r
r ' s, ��r c C s~ | dkrzt |�W S ty Y d S w | tu r=| d dkr=t�| d |�}|r;zt |�d��W S ty: Y d S w d S |S )Nr r r � )�int�
ValueError�tuple�re�search�group)Zdatatype�datar r r r
�
_datavalueD s
� �r" c C s g | ] }|� d �stj| �qS )�_)�
startswithr �__dict__)�.0�keyr r r
�
<listcomp>V s
��r( c sV t jj�d�} i g d�� td | � d��}|�d�}� fdd�}|D ]}||� q"� S )N�geom�r �ssdsr z
disk listz
c s� i }| � d�D ].}tD ])}t�|� d�|�}|r4ttj�|�|�d��}|||< |tj v r4||tj | < q q|�
tj �}|� d�rCd S |� d |< |�tj
�dkr`t�d| � � d �|� d S d S ) N�
z:\s(.*)r Zcdr r �"Device %s reports itself as an SSDr+ )�split�
_geom_attribsr r r" r r �getr r �popr r$ r r
r �append)�device�tmp�lineZattribr �value�name��retr r
�parse_geom_attribsb s* �
��
�z)_freebsd_geom.<locals>.parse_geom_attribs)r r �path�which�__salt__r. )r) Zdevicesr: r3 r r8 r
r [ s
r c C s� g g d�} t � d�D ]h}tjj�|��d�}zS|shtjj�|d ��>}|�d�d }|� d�}|dkrA| d �
|� t � d
|� n|d krS| d �
|� t � d
|� nt � d|� W d � n1 scw Y W q
t
yr Y q
w | S )zJ
Return list of disk devices and work out if they are SSD or HDD.
r* z /sys/block/*z../devices/virtual/z/queue/rotational�/� r �0r+ r- �1r z"Device %s reports itself as an HDDzHUnable to identify device %s as an SSD or HDD. It does not report 0 or 1N)�globr r r; �readlinkr$ �filesZfopenr. �readr2 r
r �OSError)r9 �entryZvirtualZentry_fpr3 �flagr r r
r ~ s2
�� �� �r c C s� d} g g d�}t d | �}|st�d� |S t|t�r |g}|D ]2}|d �� dvr5t�d|d � �� q d|d � �}|d
�|� |d �� d krR|d �|� q |S )
Nz-Get-PhysicalDisk | Select DeviceID, MediaTyper* r zNo physical discs foundZ MediaType)Zhdd�ssdZscm� unspecifiedzUnknown media type: z\\.\PhysicalDriveZDeviceIDr rI r+ )r= r
r �
isinstance�dict�lowerr2 )�cmdr9 Z
drive_info�driver3 r r r
r � s$
�r )�__doc__rB �loggingr Zsalt.modules.cmdmodr Zsalt.utils.filesZsalt.utils.pathZsalt.utils.platform�modulesZcmdmodZ
_run_quietZ_run_all_quietZ
powershellr= � getLoggerr r
r r r" r% r/ r r r r r r r
�<module> s, �
�#