Edit File: discover.cpython-36.pyc
3 u1�W�, � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlZddlZddlZddl Z ddl Z ddl m Z ddl m Z d d � Ze jej�G d d � d e��ZG d d� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZdS )z� pyudev.discover =============== Tools to discover a device given limited information. .. moduleauthor:: mulhern <amulhern@redhat.com> � )�absolute_import)�division)�print_function)�unicode_literalsN)�Devices)�DeviceNotFoundErrorc s t j� �� fdd��}|S )z\ Allow Device discovery methods to return None instead of raising an exception. c s$ y � | |�S t k r dS X dS )z� Returns result of calling ``func`` on ``args``, ``kwargs``. Returns None if ``func`` raises :exc:`DeviceNotFoundError`. N)r )�args�kwargs)�func� � /usr/lib/python3.6/discover.py�the_func1 s z wrap_exception.<locals>.the_func)� functools�wraps)r r r )r r �wrap_exception+ s r c @ sL e Zd ZdZeejdd� ��Zeejdd� ��Zedd� �Z edd � �Z d S ) � HypothesiszM Represents a hypothesis about the meaning of the device identifier. c C s t � �dS )a� Match the given string according to the hypothesis. The purpose of this method is to obtain a value corresponding to ``value`` if that is possible. It may use a regular expression, but in general it should just return ``value`` and let the lookup method sort out the rest. :param str value: the string to inspect :returns: the matched thing or None if unmatched :rtype: the type of lookup's key parameter or NoneType N)�NotImplementedError)�cls�valuer r r �matchD s zHypothesis.matchc C s t � �dS )aN Lookup the given string according to the hypothesis. :param Context context: the pyudev context :param key: a key with which to lookup the device :type key: the type of match's return value if not None :returns: a list of Devices obtained :rtype: frozenset of :class:`Device` N)r )r �context�keyr r r �lookupU s zHypothesis.lookupc C s dS )z� A potentially expensive method that may allow an :class:`Hypothesis` to find devices more rapidly or to find a device that it would otherwise miss. :param Context context: the pyudev context Nr )r r r r r �setupc s zHypothesis.setupc C s$ | j |�}|dk r | j||�S t� S )a Get any devices that may correspond to the given string. :param Context context: the pyudev context :param str value: the value to look for :returns: a list of devices obtained :rtype: set of :class:`Device` N)r r � frozenset)r r r r r r r � get_devicesn s zHypothesis.get_devicesN) �__name__� __module__� __qualname__�__doc__� classmethod�abc�abstractmethodr r r r r r r r r > s r c @ sL e Zd ZdZedd� �Zedd� �Zedd� �Zedd � �Zed d � �Z d S ) �DeviceNumberHypothesisz� Represents the hypothesis that the device is a device number. The device may be separated into major/minor number or a composite number. c C s8 t jd�}|j|�}|o6tjt|jd��t|jd���S )z� Match the number under the assumption that it is a major,minor pair. :param str value: value to match :returns: the device number or None :rtype: int or NoneType z#^(?P<major>\d+)(\D+)(?P<minor>\d+)$�major�minor)�re�compiler �os�makedev�int�group)r r Zmajor_minor_rer r r r �_match_major_minor� s z)DeviceNumberHypothesis._match_major_minorc C s& t jd�}|j|�}|o$t|jd��S )z� Match the number under the assumption that it is a single number. :param str value: value to match :returns: the device number or None :rtype: int or NoneType z^(?P<number>\d+)$Znumber)r& r' r r* r+ )r r Z number_rer r r r � _match_number� s z$DeviceNumberHypothesis._match_numberc C s | j |�p| j|�S )z� Match the number under the assumption that it is a device number. :returns: the device number or None :rtype: int or NoneType )r, r- )r r r r r r � s z DeviceNumberHypothesis.matchc C s |j }tjtjj|d��S )z� Find subsystems in /sys/dev. :param Context context: the context :returns: a lis of available subsystems :rtype: list of str Zdev)�sys_pathr( �listdir�path�join)r r r. r r r �find_subsystems� s z&DeviceNumberHypothesis.find_subsystemsc s8 t tj��� ��fdd�| j� �D �}tdd� |D ��S )z� Lookup by the device number. :param Context context: the context :param int key: the device number :returns: a list of matching devices :rtype: frozenset of :class:`Device` c 3 s | ]}�� |��V qd S )Nr )�.0�s)r r r r r � <genexpr>� s z0DeviceNumberHypothesis.lookup.<locals>.<genexpr>c s s | ]}|d k r|V qd S )Nr )r3 �rr r r r5 � s )r r Zfrom_device_numberr2 r )r r r �resr )r r r r r � s z DeviceNumberHypothesis.lookupN) r r r r r r, r- r r2 r r r r r r# | s r# c @ s( e Zd ZdZedd� �Zedd� �ZdS )�DevicePathHypothesiszG Discover the device assuming the identifier is a device path. c C s |S )z� Match ``value`` under the assumption that it is a device path. :returns: the device path or None :rtype: str or NoneType r )r r r r r r � s zDevicePathHypothesis.matchc C s( t tj�||�}|dk r"t|f�S t� S )z� Lookup by the path. :param Context context: the context :param str key: the device path :returns: a list of matching devices :rtype: frozenset of :class:`Device` N)r r Z from_pathr )r r r r7 r r r r � s zDevicePathHypothesis.lookupN)r r r r r r r r r r r r8 � s r8 c @ s4 e Zd ZdZedd� �Zedd� �Zedd� �ZdS ) �DeviceNameHypothesiszf Discover the device assuming the input is a device name. Try every available subsystem. c s< |j � d}� fdd�|D �}dd� |D �}tdd� |D ��S ) z� Find all subsystems in sysfs. :param Context context: the context :rtype: frozenset :returns: subsystems in sysfs �bus�class� subsystemc 3 s | ]}t jj� |�V qd S )N)r( r0 r1 )r3 �name)r. r r r5 � s z7DeviceNameHypothesis.find_subsystems.<locals>.<genexpr>c s s | ]}t jj|�r|V qd S )N)r( r0 �isdir)r3 �dr r r r5 � s c s s" | ]}t j|�D ] }|V qqd S )N)r( r/ )r3 r? �nr r r r5 � s )r: r; r< )r. r )r r ZdirnamesZabsnamesZ realnamesr )r. r r2 � s z$DeviceNameHypothesis.find_subsystemsc C s |S )z� Match ``value`` under the assumption that it is a device name. :returns: the device path or None :rtype: str or NoneType r )r r r r r r � s zDeviceNameHypothesis.matchc s8 t tj��� ��fdd�| j� �D �}tdd� |D ��S )z� Lookup by the path. :param Context context: the context :param str key: the device path :returns: a list of matching devices :rtype: frozenset of :class:`Device` c 3 s | ]}�� |��V qd S )Nr )r3 r4 )r r r r r r5 s z.DeviceNameHypothesis.lookup.<locals>.<genexpr>c s s | ]}|d k r|V qd S )Nr )r3 r6 r r r r5 s )r r � from_namer2 r )r r r r7 r )r r r r r s zDeviceNameHypothesis.lookupN)r r r r r r2 r r r r r r r9 � s r9 c @ sZ e Zd ZdZdddddddd d d d g Zed d� �Zedd� �Zedd� �Zedd� �Z dS )�DeviceFileHypothesisz� Discover the device assuming the value is some portion of a device file. The device file may be a link to a device node. z/devz/dev/disk/by-idz/dev/disk/by-labelz/dev/disk/by-partlabelz/dev/disk/by-partuuidz/dev/disk/by-pathz/dev/disk/by-uuidz/dev/input/by-pathz /dev/mapperz/dev/mdz/dev/vgc C s: |j � }dd� |D �}dd� |D �}ttdd� |D ���S )a7 Get all directories that may contain links to device nodes. This method checks the device links of every device, so it is very expensive. :param Context context: the context :returns: a sorted list of directories that contain device links :rtype: list c s s | ]}t |j�r|V qd S )N)�list� device_links)r3 r? r r r r5 5 s z5DeviceFileHypothesis.get_link_dirs.<locals>.<genexpr>c s s | ]}|j D ] }|V q qd S )N)rD )r3 r? �lr r r r5 6 s c s s | ]}t jj|�V qd S )N)r( r0 �dirname)r3 rE r r r r5 7 s )Z list_devices�sorted�set)r r �devicesZdevices_with_linksZlinksr r r � get_link_dirs( s z"DeviceFileHypothesis.get_link_dirsc C s | j |�| _dS )z� Set the link directories to be used when discovering by file. Uses `get_link_dirs`, so is as expensive as it is. :param Context context: the context N)rJ � _LINK_DIRS)r r r r r r 9 s zDeviceFileHypothesis.setupc C s |S )Nr )r r r r r r D s zDeviceFileHypothesis.matchc sr t tj��d�kr4�� ��}|dk r.t|f�S t� S �fdd�| jD �}� �fdd�|D �}tdd� |D ��S dS )a� Lookup the device under the assumption that the key is part of the name of a device file. :param Context context: the context :param str key: a portion of the device file name It is assumed that either it is the whole name of the device file or it is the basename. A device file may be a device node or a device link. �/Nc 3 s | ]}t jj|� �V qd S )N)r( r0 r1 )r3 Zld)r r r r5 [ s z.DeviceFileHypothesis.lookup.<locals>.<genexpr>c 3 s | ]}�� |�V qd S )Nr )r3 �f)r r r r r5 \ s c s s | ]}|d k r|V qd S )Nr )r3 r? r r r r5 ] s )r r Zfrom_device_filer rK )r r r Zdevice�filesrI r )r r r r r H s zDeviceFileHypothesis.lookupN) r r r r rK r rJ r r r r r r r rB s rB c @ s4 e Zd ZdZeeeegZdd� Z dd� Z dd� Z dS ) � Discoveryz1 Provides discovery methods for devices. c C s | j | _d S )N)� _HYPOTHESES� _hypotheses)�selfr r r �__init__m s zDiscovery.__init__c C s x| j D ]}|j|� qW dS )z Set up individual hypotheses. May be an expensive call. :param Context context: the context N)rQ r )rR r Zhypr r r r p s zDiscovery.setupc s t � �fdd�| jD ��S )z� Get the devices corresponding to value. :param Context context: the context :param str value: some identifier of the device :returns: a list of corresponding devices :rtype: frozenset of :class:`Device` c 3 s$ | ] }|j � ��D ] }|V qqd S )N)r )r3 �hr? )r r r r r5 � s z(Discovery.get_devices.<locals>.<genexpr>)r rQ )rR r r r )r r r r { s zDiscovery.get_devicesN) r r r r rB r9 r# r8 rP rS r r r r r r rO ` s rO )r Z __future__r r r r r! r r( r& ZsixZ pyudev.devicer r r Z add_metaclass�ABCMeta�objectr r# r8 r9 rB rO r r r r �<module> s& =K /M