使用 Python 查找驱动器号（Windows）

Question

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

目前我有：

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 命令提示符命令？

编辑：

对于未来有类似疑问的人，以下是解决方案：

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”的驱动器，并在连接时将驱动器号打印到控制台并允许使用“驱动器”作为变量。

Answer 1

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

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

Answer 2

由于您想反复检查驱动器是否存在，您可能希望将其作为单独的函数移动，像这样

from string import ascii_uppercase
from os import path


def get_usb_drive():
    for drive in ascii_uppercase:
        if path.exists(path.join(drive, "File.ID")):
            return drive + ":\\"
    return ""

然后，如果您希望程序反复提示用户插入设备，您可能希望循环运行它，像这样

drive = get_usb_drive()
while drive == "":
    print('Please plug in our USB drive and press any key to continue...',end="")
    input()
    drive = get_usb_drive()

最初，我们将尝试使用get_usb_drive() 获取驱动器，如果找不到，它将返回一个空字符串。我们迭代直到get_usb_drive()返回的值是一个空字符串，并提示用户插入设备并等待按键。

注意：我们使用os.path.join 创建实际的文件系统路径，而不是所有手动字符串连接。

Answer 3

使用循环并生成路径名：

import os
import string

for l in string.ascii_uppercase:
    if os.path.exists('%s:\\File.ID' % l):
        USBPATH='%s:\\' % l
        print('USB mounted to', USBPATH)
        break

Answer 4

最简单的方法是：

from pathlib import Path
root = Path(__file__).anchor  # 'C:\' or '\' on unix.

适用于所有系统。 然后你可以这样做：

some_path = Path(root).joinpath('foo', 'bar')  # C:\foo\bar or \foo\bar on unix.

它在控制台中不起作用，因为使用文件路径。