1. 技术文章
  2. Retrieve USB information using pyudev with device name

Retrieve USB information using pyudev with device name

由 半城伤御伤魂 提交于 2019-11-30 16:21:03
rand0m

PyUSB would be a good address for detailed USB information indeed. But I use pyudev to monitor the insertion of USB removable devices. So I tried to do all I need with one library.. Here is a code that works but is ugly (as you can see I can extract all information I want using pyudev except the usb size/capacity):

import glib

from pyudev import Context, Monitor
import pyudev
import subprocess 

def get_block_infos(dev_name):
    dev = pyudev.Device.from_device_file(context, dev_name)


    try:
        objProc = subprocess.Popen('lsblk --nodeps %s | grep -v SIZE  | awk \'{ print $4 }\'' % dev.get('DEVNAME'), shell=True, bufsize=0, executable="/bin/bash", stdin=None, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    except OSError, e:
        print(e)

    #  stdOut.communicate() --> dimension [0]: stdout, dimenstion [1]: stderr
    stdOut = objProc.communicate()

    print('<BLOCK INFORMATION>')
    print('Device name: %s' % dev.get('DEVNAME'))
    print('Device type: %s' % dev.get('DEVTYPE'))
    print('Bus system: %s' % dev.get('ID_BUS'))
    print('Partition label: %s' % dev.get('ID_FS_LABEL'))
    print('FS: %s' % dev.get('ID_FS_SYSTEM_ID'))
    print('FS type: %s' % dev.get('ID_FS_TYPE'))
    print('Device usage: %s' % dev.get('ID_FS_USAGE'))
    print('Device model: %s' % dev.get('ID_MODEL'))
    print('Partition type: %s' % dev.get('ID_PART_TABLE_TYPE'))
    print('USB driver: %s' % dev.get('ID_USB_DRIVER'))
    print('Path id: %s' % dev.get('ID_PATH'))
    print('Capacity: %s' % stdOut[0].strip())
    print('</BLOCK INFORMATION>')

def get_usb_infos(dev_path):
    print('<USB INFORMATION>')

    usb_removable_device = None

    # deprecated and removed from dbus
    # print(pyudev.Device.from_path(context, id_path))  
    # because I found no other documented way, I iterate
    # over all  devices and match my pci path..
    for device in context.list_devices(subsystem='usb'):
        usb_dev_path = device.get('DEVPATH')

        if dev_path.startswith(usb_dev_path):
            # this matches the usb hub,
            # the usb controller and
            # in third place the usb stick
            # so lets watch out for the
            # longest/most precise match

        try:
            if len(usb_dev_path) > len(usb_removable_device.get('DEVPATH')):
                usb_removable_device = device
        except AttributeError:
            # ignore because in first loop
            # usb_removable_device is None and
            # there is no usb_removable_device.get() attrib
            usb_removable_device = device

        # get more information with usb_removable_device.items()
        print('Vendor: %s' % usb_removable_device.get('ID_VENDOR_FROM_DATABASE'))
        print('</USB INFORMATION>')

try:
    from pyudev.glib import MonitorObserver

    def device_event(observer, device):
        get_block_infos(device.get('DEVNAME'))
        get_usb_infos(device.get('DEVPATH'))
except:
    from pyudev.glib import GUDevMonitorObserver as MonitorObserver

    def device_event(observer, action, device):
        get_block_infos(device.get('DEVNAME'))
        get_usb_infos(device.get('DEVPATH'))


context = Context()
monitor = Monitor.from_netlink(context)

monitor.filter_by(subsystem='block')
observer = MonitorObserver(monitor)

observer.connect('device-event', device_event)
monitor.start()

glib.MainLoop().run()

An example output of this script would be:

<BLOCK INFORMATION>
Device name: /dev/sdb
Device type: disk
Bus system: usb
Partition label: CentOS-6.5-x86_64-LiveCD
FS: LINUX
FS type: iso9660
Device usage: filesystem
Device model: Patriot_Memory
Partition type: dos
USB driver: usb-storage
Path id: pci-0000:00:1d.0-usb-0:1.2:1.0-scsi-0:0:0:0
Capacity: 14.5G
</BLOCK INFORMATION>
<USB INFORMATION>
Vendor: Kingston Technology Company Inc.
</USB INFORMATION>
<BLOCK INFORMATION>
Device name: /dev/sdb1
Device type: partition
Bus system: usb
Partition label: CentOS-6.5-x86_64-LiveCD
FS: LINUX
FS type: ext4
Device usage: filesystem
Device model: Patriot_Memory
Partition type: dos
USB driver: usb-storage
Path id: pci-0000:00:1d.0-usb-0:1.2:1.0-scsi-0:0:0:0
Capacity: 14.4G
</BLOCK INFORMATION>
<USB INFORMATION>
Vendor: Kingston Technology Company Inc.
</USB INFORMATION>
标签
python
usb
subsystem
pyudev
易学教程内所有资源均来自网络或用户发布的内容，如有违反法律规定的内容欢迎反馈！
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!