check_external_drive.py 文件源码

def onDeviceChange (self, hwnd, msg, wparam, lparam):
    global DRIVE
    global FLAG
    dev_broadcast_hdr = DEV_BROADCAST_HDR.from_address (lparam)

    if wparam == DBT_DEVICEARRIVAL:
      if dev_broadcast_hdr.dbch_devicetype == DBT_DEVTYP_VOLUME:
        dev_broadcast_volume = DEV_BROADCAST_VOLUME.from_address (lparam)
        drive_letter = drive_from_mask (dev_broadcast_volume.dbcv_unitmask)
        DRIVE = chr (ord ("A") + drive_letter)
        print "[*] Drive Found: ", DRIVE + ":\\"
        FLAG = True
        win32gui.PostQuitMessage(0)
        #win32gui.PostQuitMessage(0)
    return 1