U u1‘Wô,ã@sìdZddlmZddlmZddlmZddlmZddlZddlZddlZddl Z ddl Z ddl m Z ddl m Z d d „Ze  ej¡Gd d „d eƒƒZGd d„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZdS)zš pyudev.discover =============== Tools to discover a device given limited information. .. moduleauthor:: mulhern é)Úabsolute_import)Údivision)Úprint_function)Úunicode_literalsN)ÚDevices©ÚDeviceNotFoundErrorcst ˆ¡‡fdd„ƒ}|S)z\ Allow Device discovery methods to return None instead of raising an exception. cs(z ˆ||ŽWStk r"YdSXdS)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_func1s z wrap_exception..the_func)Ú functoolsÚwraps)r rr r rÚwrap_exception+s rc@sLeZdZdZeejdd„ƒƒZeejdd„ƒƒZedd„ƒZ edd „ƒZ d S) Ú HypothesiszM Represents a hypothesis about the meaning of the device identifier. cCs 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ÚmatchDszHypothesis.matchcCs 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ÚlookupUs zHypothesis.lookupcCsdS)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 ©rrr r rÚsetupcs zHypothesis.setupcCs$| |¡}|dk r| ||¡Stƒ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)rrÚ frozenset)rrrrr r rÚ get_devicesns zHypothesis.get_devicesN) Ú__name__Ú __module__Ú __qualname__Ú__doc__Ú classmethodÚabcÚabstractmethodrrrr r r r rr>s   rc@sLeZdZdZedd„ƒ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. cCs8t 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\d+)(\D+)(?P\d+)$ÚmajorÚminor)ÚreÚcompilerÚosÚmakedevÚintÚgroup)rrZmajor_minor_rerr r rÚ_match_major_minorƒs ÿ   þz)DeviceNumberHypothesis._match_major_minorcCs&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\d+)$Znumber)r+r,rr/r0)rrZ number_rerr r rÚ _match_number•s  z$DeviceNumberHypothesis._match_numbercCs| |¡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 )r1r2rr r rr¢szDeviceNumberHypothesis.matchcCs|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)rrr3r r rÚfind_subsystems¬s z&DeviceNumberHypothesis.find_subsystemscs8ttjƒ‰‡‡‡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` c3s|]}ˆˆ|ˆƒVqdS©Nr ©Ú.0Ús©rr rr rÚ Ãsz0DeviceNumberHypothesis.lookup..css|]}|dk r|VqdSr8r ©r:Úrr r rr=Äs)rrZfrom_device_numberr7r©rrrÚresr r<rr¸s zDeviceNumberHypothesis.lookupN) r!r"r#r$r%r1r2rr7rr r r rr(|s    r(c@s(eZdZdZedd„ƒZedd„ƒZdS)ÚDevicePathHypothesiszG Discover the device assuming the identifier is a device path. cCs|S)z› Match ``value`` under the assumption that it is a device path. :returns: the device path or None :rtype: str or NoneType r rr r rrÌszDevicePathHypothesis.matchcCs(ttjƒ||ƒ}|dk r"t|fƒStƒ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)rrZ from_pathrr@r r rrÖs zDevicePathHypothesis.lookupN)r!r"r#r$r%rrr r r rrBÇs  rBc@s4eZdZdZedd„ƒZedd„ƒZedd„ƒZdS) ÚDeviceNameHypothesiszf Discover the device assuming the input is a device name. Try every available subsystem. cs<|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 subsystemc3s|]}tj ˆ|¡VqdSr8©r-r5r6)r:Úname©r3r rr=ösz7DeviceNameHypothesis.find_subsystems..css|]}tj |¡r|VqdSr8)r-r5Úisdir©r:Údr r rr=÷s css"|]}t |¡D] }|VqqdSr8)r-r4)r:rKÚnr r rr=øs )r3r)rrZdirnamesZabsnamesZ realnamesr rHrr7ës z$DeviceNameHypothesis.find_subsystemscCs|S)z› Match ``value`` under the assumption that it is a device name. :returns: the device path or None :rtype: str or NoneType r rr r rrúszDeviceNameHypothesis.matchcs8ttjƒ‰‡‡‡fdd„| ˆ¡Dƒ}tdd„|DƒƒS)rCc3s|]}ˆˆ|ˆƒVqdSr8r r9r<r rr=sz.DeviceNameHypothesis.lookup..css|]}|dk r|VqdSr8r r>r r rr=s)rrÚ from_namer7rr@r r<rrs zDeviceNameHypothesis.lookupN)r!r"r#r$r%r7rrr r r rrDäs  rDc @sZeZdZdZdddddddd 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/vgcCs:| ¡}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 css|]}t|jƒr|VqdSr8)ÚlistÚ device_linksrJr r rr=5s z5DeviceFileHypothesis.get_link_dirs..css|]}|jD] }|Vq qdSr8)rP)r:rKÚlr r rr=6scss|]}tj |¡VqdSr8)r-r5Údirname)r:rQr r rr=7s)Z list_devicesÚsortedÚset)rrÚdevicesZdevices_with_linksZlinksr r rÚ get_link_dirs(s z"DeviceFileHypothesis.get_link_dirscCs| |¡|_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_DIRSrr r rr9s zDeviceFileHypothesis.setupcCs|Sr8r rr r rrDszDeviceFileHypothesis.matchcsrttjƒ‰dˆkr4ˆˆˆƒ}|dk r.t|fƒStƒS‡fdd„|jDƒ}‡‡fdd„|Dƒ}tdd„|DƒƒSdS)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. ú/Nc3s|]}tj |ˆ¡VqdSr8rF)r:Zld)rr rr=[sz.DeviceFileHypothesis.lookup..c3s|]}ˆˆ|ƒVqdSr8r )r:Úf)rr r rr=\scss|]}|dk r|VqdSr8r rJr r rr=]s)rrZfrom_device_filerrW)rrrZdeviceÚfilesrUr r<rrHs  zDeviceFileHypothesis.lookupN) r!r"r#r$rWr%rVrrrr r r rrNs*õ   rNc@s4eZdZdZeeeegZdd„Z dd„Z dd„Z dS) Ú Discoveryz1 Provides discovery methods for devices. cCs |j|_dSr8)Ú _HYPOTHESESÚ _hypotheses)Úselfr r rÚ__init__mszDiscovery.__init__cCs|jD]}| |¡qdS)z Set up individual hypotheses. May be an expensive call. :param Context context: the context N)r]r)r^rZhypr r rrps zDiscovery.setupcst‡‡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` c3s$|]}| ˆˆ¡D] }|VqqdSr8)r )r:ÚhrK©rrr rr=„sz(Discovery.get_devices..)rr])r^rrr rarr {s ÿzDiscovery.get_devicesN) r!r"r#r$rNrDr(rBr\r_rr r r r rr[`sü r[)r$Z __future__rrrrr&rr-r+ZsixZ pyudev.devicerrrZ add_metaclassÚABCMetaÚobjectrr(rBrDrNr[r r r rÚs&       =K/M