使用wmi检测光驱是否有光盘

# 使用提醒:

# 1. xbot包提供软件自动化、数据表格、Excel、日志、AI等功能

# 2. package包提供访问当前应用数据的功能，如获取元素、访问全局变量、获取资源文件等功能

# 3. 当此模块作为流程独立运行时执行main函数

# 4. 可视化流程中可以通过"调用模块"的指令使用此模块

import xbot

from xbot import print, sleep

from .import package

import wmi

def main(args):

   print(CDROM_NotEmpty('Y:'))

   print(getCDROMSpaceSize('Y'))

   print(checkDisk('Y:'))

   pass

con = wmi.WMI()

def CDROM_NotEmpty(drive):

   for cdrom in con.Win32_CDROMDrive():

       if cdrom.Drive == drive:

           return cdrom.Medialoaded

def getCDROMSpaceSize(drive):

   for disk in con.Win32_LogicalDisk():

       if disk.Caption == drive:

           return disk.size

def checkDisk(drive):

   try:

       isEmpty = CDROM_NotEmpty(drive)

       spaceSize = getCDROMSpaceSize(drive)

   except BaseException:

       print('failed检测光驱光盘异常')

   else:

       if isEmpty == True and spaceSize == None:

           return 0

       elif isEmpty == False and spaceSize == None:

           return -1

       elif isEmpty == True and int(spaceSize) > 0:

           return 1

0：检测光驱有空光盘

-1 ：检测光驱无光盘

1：检测光驱有光盘且光盘有内容