File: //snap/core18/current/usr/lib/python3/dist-packages/probert/__pycache__/storage.cpython-36.pyc
3
YQ2X � @ sV d dl Z d dlZd dlZd dlZd dlmZ e jd�ZG dd� d�ZG dd� d�Z dS )� N)�udev_get_attributeszprobert.storagec @ s^ e Zd ZdZdd� Zddd�Zedd� �Zed d
� �Zed d � �Z ed
d� �Z
edd� �Z dS )� StorageInfoa& properties:
.type = [disk, partition, etc.}
.name = /dev/sda
.size = 123012034 (bytes)
.serial = abcdefghijkl
.vendor = Innotec
.model = SolidStateRocketDrive
.devpath = /devices
.is_virtual =
.raw = {raw dictionary}
c C s: |\| _ |j| j �| _| jd | _t| jd d �| _d S )N�DEVTYPE�attrs�size)�name�get�raw�type�intr )�selfZ
probe_data� r
�1/usr/lib/python3/dist-packages/probert/storage.py�__init__& s zStorageInfo.__init__�
Unknown valuec C sF x@|D ]8}y
| j | S tk
r< tjdj|| j�� Y qX qW |S )Nz&Failed to get key {} from interface {})r �KeyError�log�debug�formatr )r �keys�missing�keyr
r
r �
_get_hwvalues- s
zStorageInfo._get_hwvaluesc C s4 dddg}| j |dd�}|dkr0| jjd�d }|S )zm Some disks don't have ID_VENDOR_* instead the vendor
is encoded in the model: SanDisk_A223JJ3J3 ZID_VENDOR_FROM_DATABASEZ ID_VENDORZ ID_VENDOR_IDzUnknown Vendor)r r �_r )r �model�split)r r �vr
r
r �vendor8 s zStorageInfo.vendorc C s dddg}| j |dd�S )NZID_MODEL_FROM_DATABASEZID_MODELZ ID_MODEL_IDz
Unknown Model)r r )r )r r r
r
r r F s zStorageInfo.modelc C s ddg}| j |dd�S )NZ ID_SERIALZID_SERIAL_SHORTzUnknown Serial)r r )r )r r r
r
r �serialO s zStorageInfo.serialc C s dg}| j |dd�S )NZDEVPATHzUnknown devpath)r r )r )r r r
r
r �devpathW s zStorageInfo.devpathc C s | j jd�S )Nz/devices/virtual/)r �
startswith)r r
r
r �
is_virtual\ s zStorageInfo.is_virtualN)r ) �__name__�
__module__� __qualname__�__doc__r r �propertyr r r r r! r
r
r
r r s
r c @ sR e Zd Zi fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Zddd�Z dd� Z
dS )�Storagec C s || _ tj� | _d S )N)�results�pyudevZContext�context)r r( r
r
r r b s zStorage.__init__c sD y&| j jd��� ��fdd��j� D �S ttfk
r> g S X d S )N�storagec s g | ]}�| � �kr|�qS r
r
)�.0�device)�keynamer+ �valuer
r �
<listcomp>i s z.Storage.get_devices_by_key.<locals>.<listcomp>)r( r r r �AttributeError)r r. r/ r
)r. r+ r/ r �get_devices_by_keyf s
zStorage.get_devices_by_keyc C s. y| j jd�j� S ttfk
r( g S X d S )Nr+ )r( r r r r1 )r r
r
r � get_devicesn s zStorage.get_devicesc s< y | j dd�}� fdd�|D �S ttfk
r6 g S X dS )z
/dev/sda r � partitionc s g | ]}|j � �r|�qS r
)r )r, �part)r- r
r r0 x s z*Storage.get_partitions.<locals>.<listcomp>N)r2 r r1 )r r- Z
partitionsr
)r- r �get_partitionst s
zStorage.get_partitionsc sD y&| j jd�� � fdd�| jdd�D �S ttfk
r> g S X d S )Nr+ c s g | ]}� | d dkr|�qS )r �diskr
)r, r7 )r+ r
r r0 � s z%Storage.get_disks.<locals>.<listcomp>�MAJOR�8)r( r r2 r r1 )r r
)r+ r � get_disks} s
zStorage.get_disksc C s@ y"| j jd�j|�}|jd�jd�S ttfk
r: dS X d S )Nr+ r r �0)r( r r r1 )r r- Zhwinfor
r
r �get_device_size� s
zStorage.get_device_sizeFc C s� t jjdt jj|��}t jj|d�}t|��}t|j� j� �}W dQ R X |}t jjt jj|d��s�t jjt j
d|�d �}t jjd|�}t jj|dd�} t jj| �r�t| ��}
|t|
j� j� �9 }W dQ R X |S )z device='/dev/sda' z/sys/class/blockr NZqueuez[\d+]r Zlogical_block_size) �os�path�join�basename�openr �read�strip�exists�rer ) r r- Z is_partitionZ
device_dirZ
blockdev_size�dr Z logsize_baseZ
parent_devZ logical_size�sr
r
r �_get_device_size� s
zStorage._get_device_sizec C sv i }xf| j jdd�D ]T}|d d krt|�}t| j|d ��|d< t|�||d < ||d jd|i� qW || _|S )
N�block)Z subsystemr8 �1�7ZDEVNAMEr r )rJ rK )r* Z list_devicesr �strrH �dict�updater( )r r+ r- r r
r
r �probe� s z
Storage.probeN)F) r" r# r$ r r2 r3 r6 r: r< rH rO r
r
r
r r' a s
r' )
Zloggingr= rE r) Z
probert.utilsr Z getLoggerr r r' r
r
r
r �<module> s
G