Current File : //snap/core/17212/usr/lib/python3/dist-packages/probert/__pycache__/storage.cpython-35.pyc
YQ2X � @ sy d d l Z d d l Z d d l Z d d l Z d d l m Z e j d � Z Gd d � d � Z Gd d � d � Z d S)� N)�udev_get_attributeszprobert.storagec @ s� e Z d Z d Z d d � Z d d d � Z e d d � � Z e d d
� � Z e d d � � Z e d
d � � Z
e d d � � Z d S)� 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 sO | \ | _ | j | j � | _ | j d | _ t | j d 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__z
Unknown valuec C sU xN | D]F } y | j | SWq t k
rL t j d j | | j � � Yq Xq W| 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 sM d d d g } | j d | d d � } | d k rI | j j d � 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_IDr r zUnknown Vendor�_r )r �model�split)r r �vr
r
r �vendor8 s zStorageInfo.vendorc C s% d d d g } | j d | d d � S)NZID_MODEL_FROM_DATABASEZID_MODELZ ID_MODEL_IDr r z
Unknown Model)r )r r r
r
r r F s zStorageInfo.modelc C s" d d g } | j d | d d � S)NZ ID_SERIALZID_SERIAL_SHORTr r zUnknown Serial)r )r r r
r
r �serialO s zStorageInfo.serialc C s d g } | j d | d d � S)NZDEVPATHr r zUnknown devpath)r )r r r
r
r �devpathW s zStorageInfo.devpathc C s | j j d � S)Nz/devices/virtual/)r �
startswith)r r
r
r �
is_virtual\ s zStorageInfo.is_virtualN) �__name__�
__module__� __qualname__�__doc__r r �propertyr r r r r r
r
r
r r s r c @ sv e Z d Z i d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d d � Z d d � Z
d S)�Storagec C s | | _ t j � | _ d S)N)�results�pyudevZContext�context)r r' r
r
r r b s zStorage.__init__c s\ y9 | j j d � � � � � f d d � � j � D� SWn t t f k
rW g SYn Xd S)N�storagec s* g | ] } � | � � k r | � q Sr
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 j d � j � SWn t t f k
r8 g SYn Xd S)Nr* )r' r r r r0 )r r
r
r � get_devicesn s zStorage.get_devicesc sP y- | j d d � } � f d d � | D� SWn t t f k
rK g SYn Xd S)z
/dev/sda r � partitionc s% g | ] } | j � � r | � q Sr
)r )r+ �part)r, r
r r/ x s z*Storage.get_partitions.<locals>.<listcomp>N)r1 r r0 )r r, Z
partitionsr
)r, r �get_partitionst s
zStorage.get_partitionsc s\ y9 | j j d � � � f d d � | j d d � D� SWn t t f k
rW g SYn Xd S)Nr* c s* g | ] } � | d d k r | � q S)r �diskr
)r+ r6 )r* r
r r/ � s z%Storage.get_disks.<locals>.<listcomp>�MAJOR�8)r' r r1 r r0 )r r
)r* r � get_disks} s
'zStorage.get_disksc C sX y5 | j j d � j | � } | j d � j d � SWn t t f k
rS d SYn Xd S)Nr* r r �0)r' r r r0 )r r, Zhwinfor
r
r �get_device_size� s
zStorage.get_device_sizeFc C s" t j j d t j j | � � } t j j | d � } t | � � } t | j � j � � } Wd QRX| } t j j t j j | d � � s� t j j t j
d | � d � } t j j d | � } t j j | d d � } t j j | � r t | � �# }
| t |
j � j � � 9} Wd QRX| 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 s� i } x� | j j d d � D]n } | d d k r t | � } t | j | d � � | d <t | � | | d <| | d j d | i � q W| | _ | S)
NZ subsystem�blockr7 �1�7ZDEVNAMEr r )rI rJ )r) Z list_devicesr �strrG �dict�updater' )r r* r, r r
r
r �probe� s z
Storage.probeN) r! r" r# r r1 r2 r5 r9 r; rG rN r
r
r
r r& a s r& )
Zloggingr< rD r( Z
probert.utilsr Z getLoggerr r r& r
r
r
r �<module> s G
Mini Shell