EVOLUTION-MANAGER
Edit File: StatStruct.cpython-36.pyc
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 _>