python之使用 Python 查找盘符 (Windows)

我正在尝试编写一个 python 脚本(我是新手)，它将在 Windows 上的每个已连接驱动器的根目录中搜索 key 文件，然后返回它所在的驱动器号，将变量设置为驱动器号.

目前我有:

import os 
if os.path.exists('A:\\File.ID'): 
        USBPATH='A:\\' 
        print('USB mounted to', USBPATH) 
    if os.path.exists('B:\\File.ID'): 
        USBPATH='B:\\' 
        print('USB mounted to', USBPATH) 
    if os.path.exists('C:\\File.ID'): 

-- 然后对每个驱动器字母 A 到 Z 重复出现。自然地，这将需要输入很多内容，我只是想知道是否有一种解决方法可以使我的代码保持整洁和尽可能少(或者这是唯一的办法？)。

此外，如果找不到驱动器(IE 说请插入您的 USB)，是否有办法让它打印错误，然后返回开始/循环？有点像

print('Please plug in our USB drive') 
return-to-start 

有点像 GOTO 命令提示符命令？

编辑:

对于有类似 future 查询的人，这里是解决方案:

from string import ascii_uppercase 
import os 
 
 
def FETCH_USBPATH(): 
    for USBPATH in ascii_uppercase: 
         if os.path.exists('%s:\\File.ID' % SVPATH): 
            USBPATH='%s:\\' % USBPATH 
            print('USB mounted to', USBPATH) 
            return USBPATH + "" 
    return "" 
 
drive = FETCH_USBPATH() 
while drive == "": 
    print('Please plug in USB drive and press any key to continue...', end="") 
    input() 
    drive = FETCH_USBPATH() 

此脚本提示用户插入包含“file.id”的驱动器，并在连接时将驱动器号打印到控制台并允许使用“驱动器”作为变量。

请您参考如下方法:

Python 对此有一个简单的解决方案。使用 pathlib 模块。

import pathlib 
drive = pathlib.Path.home().drive 
print(drive)