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

最新推荐文章于 2024-07-11 02:53:54 发布

septwolves2015

最新推荐文章于 2024-07-11 02:53:54 发布

阅读量2k

点赞数

分类专栏： python

本文链接：https://blog.csdn.net/septwolves2015/article/details/52995298

版权

python 专栏收录该内容

17 篇文章 1 订阅

订阅专栏

# coding = utf-8

import _winreg

def 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()