Sindbad~EG File Manager
3
x?ag�( � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlT ddlZddlZddl Z ddl
Z
dd l m Z dd
l m Z G d d � d e
�ZG d
d� de
�ZG dd� de �ZdS )z�
StatStruct -- classes for sampling statistics from a VDO via ioctls
$Id: //eng/vdo-releases/aluminum/src/python/vdo/statistics/StatStruct.py#1 $
� )�absolute_import)�division)�print_function)�unicode_literals)�*N� )�Field)� LabeledValuec @ sP e Zd ZdZdd� Zdd� Zdd� Zedd d
��Zedd d ��Z ed
d� �Z
dS )�SampleszD
An object which represents a collection of samples from a VDO.
c s" � d | _ � fdd�|D �| _dS )a�
Create a new set of samples by sampling a VDO device.
:param assays: The types of samples to take
:param devices: The device to sample (a dictionary containing
the user-supplied name and the name to use for sampling)
:param mustBeVDO: If set to False, errors resulting from the device not
being a VDO will be suppressed
�userc s" g | ]}|j tjj� d ���qS )�sample)r �os�path�basename)�.0�assay)�device� � /usr/lib/python3.6/StatStruct.py�
<listcomp>6 s z$Samples.__init__.<locals>.<listcomp>N)r �samples)�self�assaysr � mustBeVDOr )r r �__init__+ s
zSamples.__init__c C s | j S )z\
Get the name of the device which was sampled.
:return: The name of the device
)r )r r r r � getDevice9 s zSamples.getDevicec C s | j S )zS
Get the list of samples for the device.
:return: The list of samples
)r )r r r r �
getSamplesA s zSamples.getSamplesTc C sl y t | ||�S tk
rf } z>|d }|jdkrF|rBtdj|���dS tdj||���W Y dd}~X nX dS )a�
Assay a device.
:param assays: The types of samples to take
:param devices: The device to sample (a dictionary containing
the user-supplied name and the name to use for sampling)
:param mustBeVDO: If set to False, errors resulting from the device not
being a VDO will be suppressed
:return: The results of the assays or None if the device is not a VDO and
mustBeVDO is set to False
r � zDevice {0} is not a VDONz Error sampling device {0}: {1})r
�IOError�errno� Exception�format)r r r Zioer r r r r I s
z
Samples.assayc s t d� �fdd�|D ��S )a�
Assay a list of devices.
:param assays: The types of samples to take
:param devices: The devices to sample (a list of dictionaries containing
the user-supplied name and the name to use for sampling)
:param mustBeVDO: If set to False, errors resulting from the device not
being a VDO will be suppressed
:return: The results of the assays or None if the device is not a VDO and
mustBeVDO is set to False
Nc s g | ]}t j� |���qS r )r
r )r r )r r r r r o s z(Samples.assayDevices.<locals>.<listcomp>)�filter)r Zdevicesr r )r r r � assayDevicesa s zSamples.assayDevicesc C s
| |d�S )a�
Returns a dictionary used for sampling purposes.
The dictionary is structured as:
{ "user" : <user-specified name>,
"sample" : <sample name> }
The user-specified name is used for display purposes.
The sample name is used to perform the actual sampling.
The two names may be identical.
:param user: user-specified name
:param sample: the name to use for sampling
:return: A sampling dictionary.
)r r r )r r r r r �samplingDevicer s zSamples.samplingDeviceN)T)T) �__name__�
__module__� __qualname__�__doc__r r r � staticmethodr r# r$ r r r r r
' s r
c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�SamplezB
An object which represents a single sample (ioctl) of a VDO.
c C s || _ || _dS )z�
Create a new sample.
:param statStruct: The structure representing the type of this sample
:param sample: The sampled values
N)�
statStructr )r r+ r r r r r � s zSample.__init__c C s | j S )z�
Get the object representing the type of this sample.
:return: The StatStruct which represents the type of this sample
)r+ )r r r r �getType� s zSample.getTypec C s | j j| j�S )z
Get the sampled values as a collection of LabeledValues.
:return: A LabeledValue representing the sampled values
)r+ �labeledr )r r r r r- � s zSample.labeledc C sD t |t�s| j| S | j}x |dd� D ] }|| }q(W ||d S )a
Get the value of a named statistic.
:param statName: The name of the statistic, either as a string, or as
a list of strings, one for each level of the sample
hierarchy
:return: The value of the named statistic
Nr ���r. )�
isinstance�listr )r �statName�stats�namer r r �getStat� s
zSample.getStatc C s | j |�|j |�kS )ad
Check whether the value of a given statistic is the same in this sample
and some other sample.
:param other: The other sample
:param statName: The name of the statistic as would be specified to
getStat()
:return: True if the value of the named statistic is the same in this and
the other sample
)r4 )r �otherr1 r r r � statEqual� s zSample.statEqualN) r% r& r' r( r r, r- r4 r6 r r r r r* � s
r* c sV e Zd ZdZi Z� fdd�Zdd� Zdd� Zdd � Zd
d � Z d d
� Z
ddd�Z � Z S )�
StatStructz�
Base class for objects representing a VDO statistics structure. This object
can be used to sample a VDO via an ioctl and convert the result from the
C format in the ioctl to a Sample object.
c s� |j dd�}|dkr |d nd| _|j dd�| _|j dd�| _|j dd�| _|| _tt| �j|| j � f|� t
� | _ x|D ]}|| j |j < qzW dS )a�
Create a new statistics structure.
:param name: The name of statistics structure
:param fields: A list of Field objects specifying the format of the C
structure returned by the ioctl
:param **kwargs: Keyword args which may be:
labelPrefix: The prefix to prepend to the label for
each field in this structure
ioctl: The value of the ioctl to use for sampling
a VDO
Field: Any of the keyword arguments for the Field
base class
� labelPrefixN� � �ioctl�procFile�procRoot)
�popr8 r; r< r= �fields�superr7 r �
_getCClass�dict� fieldsByNamer3 )r r3 r? �kwargsr8 �field)� __class__r r r � s
zStatStruct.__init__c C sR t | �jd }| jj|�}|sNdd� | jD �}t t|�tfd|i�}|| j|< |S )z�
Get the Structure class which represents the C struct for a StatStruct
or Field. If the Structure class hasn't yet been made for the given type,
it will be created.
:return: The class defined by the specified set of Fields
z.cc S s g | ]}|j r|j|jf�qS r )ZinStructr3 �cType)r rE r r r r � s z)StatStruct._getCClass.<locals>.<listcomp>Z_fields_)�typer% �cClasses�getr? �strZ Structure)r Z classNamerG Z fieldListr r r rA � s
zStatStruct._getCClassc C sH | j � }tjjd| j|| j�}t|d��}|j|� W dQ R X | j|�S )z�
Get a sample from a VDO via an ioctl.
:param name: The name of the proc directory from which to read
:return: The sample
z/proc�rbN) rG r
r �joinr= r< �open�readinto�_extract)r r3 r2 ZprocPath�fdr r r r � s
zStatStruct.samplec C s2 t � }x | jD ]}|j|| �||j< qW t| |�S )z�
Extract the sampled values from the return of an ioctl call.
:param stats: The structure returned from an ioctl
:return: The sample as a Sample
)rB r? �
extractSampler3 r* )r r2 r rE r r r rP
s zStatStruct._extractc s^ t |� j|�}� jdkr2||kr2� fdd�|D �S t� }x � jD ]}|j|� �||j< q@W |S )z
:inherit:
r c s g | ]}� j |� ��qS r )rR )r �s)r r r r s z,StatStruct.extractSample.<locals>.<listcomp>)�getattrr3 ZlengthrB r? rR )r r2 �parentZmyStatsr rE r )r r rR s zStatStruct.extractSamplec C s | j | j|| �S )a
Get the value of one of the fields at the current level of the C structure
from the ioctl.
:param stats: The current level of the C structure
:param fieldName: The name of the field to get
:return: The value of the specified field
)rC rR )r r2 Z fieldNamer r r �getSampleValue( s
zStatStruct.getSampleValuer: c sV � �j 7 � � �j }t�t�r4� �fdd��D �}n� �fdd��jD �}tj||�S )z
:inherit:
c s g | ]}�j |� ��qS r )r- )r rS )�prefixr r r r < s z&StatStruct.labeled.<locals>.<listcomp>c s$ g | ] }|j r|j�|j � ��qS r )Zdisplayr- r3 )r rE )rW r r r r ? s )r8 �labelr/ r0 r? r Zmake)r r rW rX Z
labeledFieldsr )rW r r r r- 4 s
zStatStruct.labeled)r: )
r% r& r' r( rI r rA r rP rR rV r- �
__classcell__r r )rF r r7 � s
r7 )r( Z
__future__r r r r Zctypes� collectionsZfcntlr
�sysr r �objectr
r* r7 r r r r �<module> s _>
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists