3
�<�e3 � @ sP d dl Z d dlZd dlmZ dgZejj� ZG dd� de�Z G dd� de�Z
dS )� N)�consts� Inventoryc @ s\ e Zd ZdZddd�Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z ddd�Z dS )r z�
Inventory object can handle information about available hardware devices. It also informs the plugins
about related hardware events.
NTc C s� |d k r|| _ n
tj� | _ |d kr(tj}|j| j �| _|d krDtj}|rxy| jj|� W n t k
rv t
j d� Y nX |d kr�t � }|| _
d | _i | _d S )NzLcannot set udev monitor receive buffer size, we are probably running inside zIcontainer or with limited capabilites, TuneD functionality may be limitedz�cannot set udev monitor receive buffer size, we are probably running inside container or with limited capabilites, TuneD functionality may be limited)�
_udev_context�pyudevZContextZMonitorZ from_netlink�
_udev_monitorr ZCFG_DEF_UDEV_BUFFER_SIZE�set_receive_buffer_size�EnvironmentError�log�warn�_MonitorObserverFactory�_monitor_observer_factory�_monitor_observer�_subscriptions)�selfZ udev_contextZudev_monitor_clsZmonitor_observer_factory� buffer_sizer � r � /usr/lib/python3.6/inventory.py�__init__ s&
zInventory.__init__c C s: yt jj| j||�S tk
r4 t jj| j||�S X dS )z9Get a pyudev.Device object for the sys_name (e.g. 'sda').N)r ZDevices� from_namer �AttributeErrorZDevice)r � subsystemZsys_namer r r �
get_device) s zInventory.get_devicec C s | j j|d�S )z)Get list of devices on a given subsystem.)r )r Z list_devices)r r r r r � get_devices1 s zInventory.get_devicesc C sx |j | jkrd S xb| j|j D ]R\}}y|||� W q tk
rn } z tjd| � tj|� W Y d d }~X q X q W d S )Nz+Exception occured in event handler of '%s'.)r r � Exceptionr �errorZ exception)r ZeventZdevice�plugin�callback�er r r �_handle_udev_event5 s z Inventory._handle_udev_eventc C s\ t jd||f � ||f}|| jkr6| j| j|� n"|g| j|< | jj|� | jj� dS )z7Register handler of device events on a given subsystem.zadding handler: %s (%s)N)r �debugr �appendr Z filter_by�start)r r r r �
callback_datar r r � subscribe@ s
zInventory.subscribec C s6 | j d kr2tjd� | jj| j| j�| _ | j j� d S )Nzstarting monitor observer)r
r r r �creater r r! )r r r r �start_processing_eventsL s
z!Inventory.start_processing_eventsc C s( | j d k r$tjd� | j j� d | _ d S )Nzstopping monitor observer)r
r r �stop)r r r r �stop_processing_eventsR s
z Inventory.stop_processing_eventsc C sJ xD| j | D ]6}|\}}||kr tjd||f � | j | j|� q W d S )Nzremoving handler: %s (%s))r r r �remove)r r r r" Z_pluginr r r r �_unsubscribe_subsystemX s
z Inventory._unsubscribe_subsystemc C sf g }xF| j D ]<}|dks ||kr | j||� t| j | �dkr |j|� q W x|D ] }| j |= qRW dS )z4Unregister handler registered with subscribe method.Nr )r r) �lenr )r r r Zempty_subsystemsZ
_subsystemr r r � unsubscribe_ s
zInventory.unsubscribe)NNNNT)N)
�__name__�
__module__� __qualname__�__doc__r r r r r# r% r' r) r+ r r r r r s
c @ s e Zd Zdd� ZdS )r c O s t j||�S )N)r ZMonitorObserver)r �args�kwargsr r r r$ l s z _MonitorObserverFactory.createN)r, r- r. r$ r r r r r k s r ) r Z
tuned.logsZtunedr �__all__Zlogs�getr �objectr r r r r r �<module> s
b