python 自动检测 U 盘拔插

工作中用到，记录一下。
自动检测 U 盘拔插，输出 U 盘盘符。

import subprocess
import os
import re,time
from shutil import copyfile

def sh(command, print_msg=True):
    p = subprocess.Popen(
        command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
    try:
        result = p.stdout.read().decode('GBK')
    except:
        result = p.stdout.read().decode('utf-8')
    if print_msg:
        print(result)
    return result


def usbpath():
    if os.name == 'nt':
        disks = sh("wmic logicaldisk get deviceid, description",
                   print_msg=False).split('\n')
        print(disks)
        for disk in disks:
            if 'Removable' in disk:
                return re.search(r'\w:', disk).group()
    elif os.name == 'posix':
        return sh('ll -a /media')[-1].strip()
    else:
        return sh('ls /Volumes')[-1].strip()

print(usbpath())