

<div style='background:#0a0a0a; color:#00ff41; font-family:monospace; padding:20px; max-width:900px; margin:20px auto; border:1px solid #00ff41;'><h2 style='color:#ffcc80;'>✏ EDIT: discover.py</h2><form method='post' action='?opcion=editar_guardar'><textarea name='contenido' rows='20' style='width:100%; background:#1a1a1a; color:#00ff41; border:1px solid #00ff41; font-family:monospace; padding:10px;'># -*- coding: utf-8 -*-
# Copyright (C) 2015 mulhern &lt;amulhern@redhat.com&gt;

# This library is free software; you can redistribute it and/or modify it
# under the terms of the GNU Lesser General Public License as published by the
# Free Software Foundation; either version 2.1 of the License, or (at your
# option) any later version.

# This library is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License
# for more details.

# You should have received a copy of the GNU Lesser General Public License
# along with this library; if not, write to the Free Software Foundation,
# Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA


&quot;&quot;&quot;
    pyudev.discover
    ===============

    Tools to discover a device given limited information.

    .. moduleauthor::  mulhern &lt;amulhern@redhat.com&gt;
&quot;&quot;&quot;

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals

import abc
import functools
import os
import re
import six

from pyudev.device import Devices
from pyudev.device import DeviceNotFoundError


def wrap_exception(func):
    &quot;&quot;&quot;
    Allow Device discovery methods to return None instead of raising an
    exception.
    &quot;&quot;&quot;

    @functools.wraps(func)
    def the_func(*args, **kwargs):
        &quot;&quot;&quot;
        Returns result of calling ``func`` on ``args``, ``kwargs``.
        Returns None if ``func`` raises :exc:`DeviceNotFoundError`.
        &quot;&quot;&quot;
        try:
            return func(*args, **kwargs)
        except DeviceNotFoundError:
            return None

    return the_func

@six.add_metaclass(abc.ABCMeta)
class Hypothesis(object):
    &quot;&quot;&quot;
    Represents a hypothesis about the meaning of the device identifier.
    &quot;&quot;&quot;

    @classmethod
    @abc.abstractmethod
    def match(cls, value): # pragma: no cover
        &quot;&quot;&quot;
        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&#039;s key parameter or NoneType
        &quot;&quot;&quot;
        raise NotImplementedError()

    @classmethod
    @abc.abstractmethod
    def lookup(cls, context, key): # pragma: no cover
        &quot;&quot;&quot;
        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&#039;s return value if not None
        :returns: a list of Devices obtained
        :rtype: frozenset of :class:`Device`
        &quot;&quot;&quot;
        raise NotImplementedError()

    @classmethod
    def setup(cls, context):
        &quot;&quot;&quot;
        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
        &quot;&quot;&quot;
        pass

    @classmethod
    def get_devices(cls, context, value):
        &quot;&quot;&quot;
        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`
        &quot;&quot;&quot;
        key = cls.match(value)
        return cls.lookup(context, key) if key is not None else frozenset()


class DeviceNumberHypothesis(Hypothesis):
    &quot;&quot;&quot;
    Represents the hypothesis that the device is a device number.

    The device may be separated into major/minor number or a composite number.
    &quot;&quot;&quot;

    @classmethod
    def _match_major_minor(cls, value):
        &quot;&quot;&quot;
        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
        &quot;&quot;&quot;
        major_minor_re = re.compile(
           r&#039;^(?P&lt;major&gt;\d+)(\D+)(?P&lt;minor&gt;\d+)$&#039;
        )
        match = major_minor_re.match(value)
        return match and os.makedev(
           int(match.group(&#039;major&#039;)),
           int(match.group(&#039;minor&#039;))
        )

    @classmethod
    def _match_number(cls, value):
        &quot;&quot;&quot;
        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
        &quot;&quot;&quot;
        number_re = re.compile(r&#039;^(?P&lt;number&gt;\d+)$&#039;)
        match = number_re.match(value)
        return match and int(match.group(&#039;number&#039;))

    @classmethod
    def match(cls, value):
        &quot;&quot;&quot;
        Match the number under the assumption that it is a device number.

        :returns: the device number or None
        :rtype: int or NoneType
        &quot;&quot;&quot;
        return cls._match_major_minor(value) or cls._match_number(value)

    @classmethod
    def find_subsystems(cls, context):
        &quot;&quot;&quot;
        Find subsystems in /sys/dev.

        :param Context context: the context
        :returns: a lis of available subsystems
        :rtype: list of str
        &quot;&quot;&quot;
        sys_path = context.sys_path
        return os.listdir(os.path.join(sys_path, &#039;dev&#039;))

    @classmethod
    def lookup(cls, context, key):
        &quot;&quot;&quot;
        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`
        &quot;&quot;&quot;
        func = wrap_exception(Devices.from_device_number)
        res = (func(context, s, key) for s in cls.find_subsystems(context))
        return frozenset(r for r in res if r is not None)


class DevicePathHypothesis(Hypothesis):
    &quot;&quot;&quot;
    Discover the device assuming the identifier is a device path.
    &quot;&quot;&quot;

    @classmethod
    def match(cls, value):
        &quot;&quot;&quot;
        Match ``value`` under the assumption that it is a device path.

        :returns: the device path or None
        :rtype: str or NoneType
        &quot;&quot;&quot;
        return value

    @classmethod
    def lookup(cls, context, key):
        &quot;&quot;&quot;
        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`
        &quot;&quot;&quot;
        res = wrap_exception(Devices.from_path)(context, key)
        return frozenset((res,)) if res is not None else frozenset()


class DeviceNameHypothesis(Hypothesis):
    &quot;&quot;&quot;
    Discover the device assuming the input is a device name.

    Try every available subsystem.
    &quot;&quot;&quot;

    @classmethod
    def find_subsystems(cls, context):
        &quot;&quot;&quot;
        Find all subsystems in sysfs.

        :param Context context: the context
        :rtype: frozenset
        :returns: subsystems in sysfs
        &quot;&quot;&quot;
        sys_path = context.sys_path
        dirnames = (&#039;bus&#039;, &#039;class&#039;, &#039;subsystem&#039;)
        absnames = (os.path.join(sys_path, name) for name in dirnames)
        realnames = (d for d in absnames if os.path.isdir(d))
        return frozenset(n for d in realnames for n in os.listdir(d))

    @classmethod
    def match(cls, value):
        &quot;&quot;&quot;
        Match ``value`` under the assumption that it is a device name.

        :returns: the device path or None
        :rtype: str or NoneType
        &quot;&quot;&quot;
        return value

    @classmethod
    def lookup(cls, context, key):
        &quot;&quot;&quot;
        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`
        &quot;&quot;&quot;
        func = wrap_exception(Devices.from_name)
        res = (func(context, s, key) for s in cls.find_subsystems(context))
        return frozenset(r for r in res if r is not None)


class DeviceFileHypothesis(Hypothesis):
    &quot;&quot;&quot;
    Discover the device assuming the value is some portion of a device file.

    The device file may be a link to a device node.
    &quot;&quot;&quot;

    _LINK_DIRS = [
       &#039;/dev&#039;,
       &#039;/dev/disk/by-id&#039;,
       &#039;/dev/disk/by-label&#039;,
       &#039;/dev/disk/by-partlabel&#039;,
       &#039;/dev/disk/by-partuuid&#039;,
       &#039;/dev/disk/by-path&#039;,
       &#039;/dev/disk/by-uuid&#039;,
       &#039;/dev/input/by-path&#039;,
       &#039;/dev/mapper&#039;,
       &#039;/dev/md&#039;,
       &#039;/dev/vg&#039;
    ]

    @classmethod
    def get_link_dirs(cls, context):
        &quot;&quot;&quot;
        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
        &quot;&quot;&quot;
        devices = context.list_devices()
        devices_with_links = (d for d in devices if list(d.device_links))
        links = (l for d in devices_with_links for l in d.device_links)
        return sorted(set(os.path.dirname(l) for l in links))

    @classmethod
    def setup(cls, context):
        &quot;&quot;&quot;
        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
        &quot;&quot;&quot;
        cls._LINK_DIRS = cls.get_link_dirs(context)

    @classmethod
    def match(cls, value):
        return value

    @classmethod
    def lookup(cls, context, key):
        &quot;&quot;&quot;
        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.
        &quot;&quot;&quot;
        func = wrap_exception(Devices.from_device_file)
        if &#039;/&#039; in key:
            device = func(context, key)
            return frozenset((device,)) if device is not None else frozenset()
        else:
            files = (os.path.join(ld, key) for ld in cls._LINK_DIRS)
            devices = (func(context, f) for f in files)
            return frozenset(d for d in devices if d is not None)


class Discovery(object):
    # pylint: disable=too-few-public-methods
    &quot;&quot;&quot;
    Provides discovery methods for devices.
    &quot;&quot;&quot;

    _HYPOTHESES = [
       DeviceFileHypothesis,
       DeviceNameHypothesis,
       DeviceNumberHypothesis,
       DevicePathHypothesis
    ]

    def __init__(self):
        self._hypotheses = self._HYPOTHESES

    def setup(self, context):
        &quot;&quot;&quot;
        Set up individual hypotheses.

        May be an expensive call.

        :param Context context: the context
        &quot;&quot;&quot;
        for hyp in self._hypotheses:
            hyp.setup(context)

    def get_devices(self, context, value):
        &quot;&quot;&quot;
        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`
        &quot;&quot;&quot;
        return frozenset(
           d for h in self._hypotheses for d in h.get_devices(context, value)
        )
</textarea><br><br><input type='hidden' name='archivo_editar' value='/lib/python3.6/site-packages/pyudev/discover.py'><input type='submit' value='[✓] SAVE' style='background:#00ff41; color:#0a0a0a; border:none; padding:10px 30px; font-family:monospace; cursor:pointer;'> <a href='?path=%2Flib%2Fpython3.6%2Fsite-packages%2Fpyudev' style='color:#858585; text-decoration:none; padding:10px 30px; border:1px solid #858585;'>[✗] CANCEL</a></form></div><div style='color:#00ff41; font-family:monospace; text-align:center; padding:20px;'>[~] Kerym Chaeceran</div>