Current File : //snap/core20/2379/lib/python3/dist-packages/pyudev/__pycache__/discover.cpython-38.pyc
U
u1�W�, � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlZddlZddlZddl Z ddl
Z
ddl m Z ddl m
Z
d d
� Ze
�ej�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 �� �� fdd��}|S )z\
Allow Device discovery methods to return None instead of raising an
exception.
c s( z � | |�W S t k
r" Y dS X dS )z�
Returns result of calling ``func`` on ``args``, ``kwargs``.
Returns None if ``func`` raises :exc:`DeviceNotFoundError`.
Nr )�args�kwargs��func� �1/usr/lib/python3/dist-packages/pyudev/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`
Nr )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$ | � |�}|dk r | �||�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 �d�}|�|�}|o6t�t|�d��t|�d���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 �d�}|�|�}|o$t|�d��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/ r0 )r r Z number_rer r
r
r �
_match_number� s
z$DeviceNumberHypothesis._match_numberc C s | � |�p| �|�S )z�
Match the number under the assumption that it is a device number.
:returns: the device number or None
:rtype: int or NoneType
)r1 r2 r r
r
r r � s z DeviceNumberHypothesis.matchc C s |j }t�tj�|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 r3 r
r
r �find_subsystems� s z&DeviceNumberHypothesis.find_subsystemsc s8 t tj��� ��fdd�| �� �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 r8 r
�r: �rr
r
r r= � s )r r Zfrom_device_numberr7 r �r r r �resr
r< r r � s
z DeviceNumberHypothesis.lookupN)
r! r" r# r$ r% r1 r2 r r7 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 � s zDevicePathHypothesis.matchc C s( t tj�||�}|dk r"t|f�S t� S )��
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
r r � s
zDevicePathHypothesis.lookupN)r! r" r# r$ r% r r r
r
r
r rB � s
rB 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
)Zbus�classZ subsystemc 3 s | ]}t j�� |�V qd S r8 �r- r5 r6 )r: �name�r3 r
r r= � s z7DeviceNameHypothesis.find_subsystems.<locals>.<genexpr>c s s | ]}t j�|�r|V qd S r8 )r- r5 �isdir�r: �dr
r
r r= � s c s s" | ]}t �|�D ]
}|V qqd S r8 )r- r4 )r: rK �nr
r
r r= � s )r3 r )r r ZdirnamesZabsnamesZ realnamesr
rH r r7 � 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 � s zDeviceNameHypothesis.matchc s8 t tj��� ��fdd�| �� �D �}tdd� |D ��S )rC c 3 s | ]}�� |��V qd S r8 r
r9 r< r
r r= s z.DeviceNameHypothesis.lookup.<locals>.<genexpr>c s s | ]}|d k r|V qd S r8 r
r> r
r
r r= s )r r � from_namer7 r r@ r
r< r r s
zDeviceNameHypothesis.lookupN)r! r" r# r$ r% r7 r r r
r
r
r rD � s
rD 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: |� � }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 r8 )�list� device_linksrJ r
r
r r= 5 s
z5DeviceFileHypothesis.get_link_dirs.<locals>.<genexpr>c s s | ]}|j D ]
}|V q qd S r8 )rP )r: rK �lr
r
r r= 6 s c s s | ]}t j�|�V qd S r8 )r- r5 �dirname)r: rQ r
r
r r= 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 | � |�| _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)rV �
_LINK_DIRSr r
r
r r 9 s zDeviceFileHypothesis.setupc C s |S r8 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 j�|� �V qd S r8 rF )r: Zld)r r
r r= [ s z.DeviceFileHypothesis.lookup.<locals>.<genexpr>c 3 s | ]}�� |�V qd S r8 r
)r: �f)r r r
r r= \ s c s s | ]}|d k r|V qd S r8 r
rJ r
r
r r= ] s )r r Zfrom_device_filer rW )r r r Zdevice�filesrU r
r< r r H s
zDeviceFileHypothesis.lookupN)
r! r" r# r$ rW r% rV r r r r
r
r
r rN s* �
rN c @ s4 e Zd ZdZeeeegZdd� Z dd� Z
dd� Z dS ) � Discoveryz1
Provides discovery methods for devices.
c C s | j | _d S r8 )� _HYPOTHESES� _hypotheses)�selfr
r
r �__init__m s zDiscovery.__init__c C s | j D ]}|�|� qdS )z
Set up individual hypotheses.
May be an expensive call.
:param Context context: the context
N)r] r )r^ 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$ | ] }|� � ��D ]
}|V qqd S r8 )r )r: �hrK �r r r
r r= � s z(Discovery.get_devices.<locals>.<genexpr>)r r] )r^ r r r
ra r r { s �zDiscovery.get_devicesN) r! r" r# r$ rN rD r( rB r\ r_ r r r
r
r
r r[ ` s � r[ )r$ Z
__future__r r r r r& r r- r+ ZsixZ
pyudev.devicer r r Z
add_metaclass�ABCMeta�objectr r( rB rD rN r[ r
r
r
r �<module> s&
=K /M
Mini Shell