o 8g+^w,@sdZddlmZddlmZddlmZddlmZddlZddlZddlZddl Z ddl Z ddl m Z ddl mZd d Ze ejGd d d eZGd ddeZGdddeZGdddeZGdddeZGdddeZdS)z pyudev.discover =============== Tools to discover a device given limited information. .. moduleauthor:: mulhern )absolute_import)division)print_function)unicode_literalsNDeviceNotFoundError)Devicescstfdd}|S)z\ Allow Device discovery methods to return None instead of raising an exception. cs&z|i|WStyYdSw)z Returns result of calling ``func`` on ``args``, ``kwargs``. Returns None if ``func`` raises :exc:`DeviceNotFoundError`. Nr)argskwargsfunc1/usr/lib/python3/dist-packages/pyudev/discover.pythe_func0s  z wrap_exception..the_func) functoolswraps)r rr r rwrap_exception*s rc@sLeZdZdZeejddZeejddZeddZ edd Z d S) HypothesiszM Represents a hypothesis about the meaning of the device identifier. cCt)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 NotImplementedErrorclsvaluer r rmatchDszHypothesis.matchcCr)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` r)rcontextkeyr r rlookupUs 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 rsetupcs zHypothesis.setupcCs$||}|dur|||StS)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__ classmethodabcabstractmethodrrrr!r r r rr>s    rc@sLeZdZdZeddZeddZeddZedd 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. cCs8td}||}|ott|dt|dS)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+)$majorminor)recompilerosmakedevintgroup)rrmajor_minor_rerr r r_match_major_minors z)DeviceNumberHypothesis._match_major_minorcCs&td}||}|ot|dS)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+)$number)r,r-rr0r1)rr number_rerr r r _match_numbers 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 )r3r6rr r rrszDeviceNumberHypothesis.matchcCs|j}ttj|dS)z Find subsystems in /sys/dev. :param Context context: the context :returns: a lis of available subsystems :rtype: list of str dev)sys_pathr.listdirpathjoin)rrr8r r rfind_subsystemss z&DeviceNumberHypothesis.find_subsystemsc8ttjfdd|D}tdd|DS)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` c3|] }|VqdSNr .0srr rr r z0DeviceNumberHypothesis.lookup..cs|] }|dur|VqdSr?r rArr r rrDrE)rrfrom_device_numberr<r rrrresr rCrr zDeviceNumberHypothesis.lookupN) r"r#r$r%r&r3r6rr<rr r r rr)|s     r)c@s(eZdZdZeddZeddZdS)DevicePathHypothesiszG Discover the device assuming the identifier is a device path. cC|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 rrzDevicePathHypothesis.matchcCs(ttj||}|durt|fStS) 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)rr from_pathr rJr r rrs zDevicePathHypothesis.lookupN)r"r#r$r%r&rrr r r rrMs   rMc@s4eZdZdZeddZeddZeddZdS) DeviceNameHypothesiszf Discover the device assuming the input is a device name. Try every available subsystem. cs<|jd}fdd|D}dd|D}tdd|DS)z Find all subsystems in sysfs. :param Context context: the context :rtype: frozenset :returns: subsystems in sysfs )busclass subsystemc3s|] }tj|VqdSr?r.r:r;)rAnamer8r rrDz7DeviceNameHypothesis.find_subsystems..css |] }tj|r|VqdSr?)r.r:isdirrAdr r rrDcss$|] }t|D]}|Vq qdSr?)r.r9)rAr\nr r rrDs")r8r )rrdirnamesabsnames realnamesr rXrr<s  z$DeviceNameHypothesis.find_subsystemscCrN)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 rrrOzDeviceNameHypothesis.matchcr=)rPc3r>r?r r@rCr rrD rEz.DeviceNameHypothesis.lookup..csrFr?r rGr r rrD rE)rr from_namer<r rJr rCrrrLzDeviceNameHypothesis.lookupN)r"r#r$r%r&r<rrr r r rrRs   rRc@sHeZdZdZgdZeddZeddZeddZed d Z d S) 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|DS)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|jr|VqdSr?)list device_linksr[r r rrD*rYz5DeviceFileHypothesis.get_link_dirs..css |] }|jD]}|VqqdSr?)re)rAr\lr r rrD+r]css|] }tj|VqdSr?)r.r:dirname)rArfr r rrD,rE) list_devicessortedset)rrdevicesdevices_with_linkslinksr r r get_link_dirss 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)rn _LINK_DIRSrr r rr.s zDeviceFileHypothesis.setupcCrNr?r rr r rr9szDeviceFileHypothesis.matchcsnttjdvr}|durt|fStSfdd|jD}fdd|D}tdd|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. /Nc3s|] }tj|VqdSr?rV)rAld)rr rrDPrYz.DeviceFileHypothesis.lookup..c3s|]}|VqdSr?r )rAf)rr r rrDQscsrFr?r r[r r rrDRrE)rrfrom_device_filer ro)rrrdevicefilesrkr rCrr=s  zDeviceFileHypothesis.lookupN) r"r#r$r%ror&rnrrrr r r rrcs   rcc@s4eZdZdZeeeegZddZ ddZ ddZ dS) Discoveryz1 Provides discovery methods for devices. cCs |j|_dSr?) _HYPOTHESES _hypotheses)selfr r r__init__`s zDiscovery.__init__cCs|jD]}||qdS)z Set up individual hypotheses. May be an expensive call. :param Context context: the context N)rxr)ryrhypr r rrcs  zDiscovery.setupcstfdd|jDS)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]}|Vq qdSr?)r!)rAhr\rrr rrDws   z(Discovery.get_devices..)r rx)ryrrr r}rr!ns zDiscovery.get_devicesN) r"r#r$r%rcrRr)rMrwrzrr!r r r rrvUs rv)r% __future__rrrrr'rr.r,sixpyudev._errorsr pyudev.devicerr add_metaclassABCMetaobjectrr)rMrRrcrvr r r rs(      =G/F