获取windows和linux下U盘路径

windows下安装psutil包

 pip install psutil

如果失败去以下网址下载whl包安装

https://www.lfd.uci.edu/~gohlke/pythonlibs/#psutil

获取linux和windows下的U盘路径

# 只获取第一个可移动U盘路径
import psutil
    def getUDiskPath():
        if system_name == 'Windows':
            # 获取所有盘符信息
            disk_list = psutil.disk_partitions()
            u_path = [disk.device for disk in disk_list if disk.opts == 'rw,removable']
            if u_path:
                return u_path[0]
        elif system_name == 'linux':
            import pyudev       
            context = pyudev.Context()
            removable = [device for device in context.list_devices(subsystem='block', DEVTYPE='disk') if device.attributes.asstring('removable') == "1"]
            for device in removable:
              partitions = [device.device_node for device in context.list_devices(subsystem='block', DEVTYPE='partition', parent=device)]
              #print("All removable partitions: {}".format(",
              #".join(partitions)))
              #print("Mounted removable partitions:")
              for p in psutil.disk_partitions():
                  if p.device in partitions:
                      return p.mountpoint