python通过查找注册表的方式获取盘符(磁盘列表)

来源:互联网 发布:申请域名注册备案平台 编辑:程序博客网 时间:2024/06/06 02:02
# coding = utf-8import _winregdef GetPartitionNames():    res = []    deviceID = ''    subKey = 'SYSTEM\MountedDevices'        key = _winreg.OpenKey(_winreg.HKEY_LOCAL_MACHINE, subKey)    #print key    i = 0    try:        while True:            name,value,type = _winreg.EnumValue(key,i)            if name.startswith('\\DosDevices\\'):                res.append((name, repr(value)[1:17]))                if name.startswith('\\DosDevices\\C'):                    deviceID = repr(value)[1:17]            i+=1    except WindowsError:        pass    res = filter(lambda item : item[1] == deviceID, res)    res = zip(*res)[0]    index = res[0].rindex('\\')    res = sorted([item[index+1:-1] for item in res])    return res    def test():    print GetPartitionNames()if __name__ == '__main__':    test()
0 0
原创粉丝点击