为什么 os.path.expanduser 不返回主目录？

Question

我正在制作一个 python 桌面应用程序，它将日志作为 .csv 文件保存在 Windows 上用户的 Documents 文件夹中。该应用程序是用 python 2.7 和 kivy 1.8.0 编写的，使用 pyinstaller 2.1 打包为 Windows 程序，安装程序是使用 Inno Setup Compiler 制作的。在这篇文章中，我会将用户的真实姓名替换为 USER。

我有以下几行代码：

DOCUMENTS = os.path.expanduser('~\\Documents\\')
print DOCUMENTS
with open(DOCUMENTS + 'data_log.csv', 'ab') as f:
    do stuff

在我的计算机和我测试过的另一台计算机上，程序按预期运行。 DOCUMENTS 评估为“C:\Users\USER\Documents\”。但是，在我尝试过的其他三台计算机上，DOCUMENTS 的计算结果为“C:\Users\USER\AppData\Roaming\SPB_16.6\Documents\”。然后程序在尝试创建 data_log.csv 时崩溃，并出现以下错误：

IOError: [Errno 2] No such file or directory: 'C:\\Users\\USER\\AppData\Roaming\\SPB_16.6\\Documents\\data_log.csv'

首先，为什么 os.path.expanduser 可能在某些系统上行为不端，而在其他系统上却没有？

其次，即使在错误的目录下，如果文件不存在，open() 也应该创建文件，那为什么会导致它崩溃呢？

我已经弄清楚是什么导致了这个问题。在大多数系统上，HOME 是 None，所以 os.path.expanduser 使用 USERPROFILE 代替。但是，在极少数情况下，HOME 设置为 C:\SPB\ 或 C:\Users\USER\AppData\Roaming\SPB_16.6 之类的内容。我的解决方案是使用os.environ直接访问USERPROFILE，而不是使用os.path.expanduser。

Answer 1

来自expanduser的文档：

在 Windows 上，如果设置了 HOME 和 USERPROFILE，将使用，否则 将使用 HOMEPATH 和 HOMEDRIVE 的组合。一个初始的~用户 通过从创建的最后一个目录组件中剥离来处理 上面派生的用户路径。

如您所见，代码极其简单（转储为inspect）：

def expanduser(path):
    """Expand ~ and ~user constructs.

    If user or $HOME is unknown, do nothing."""
    if path[:1] != '~':
        return path
    i, n = 1, len(path)
    while i < n and path[i] not in '/\\':
        i = i + 1

    if 'HOME' in os.environ:
        userhome = os.environ['HOME']
    elif 'USERPROFILE' in os.environ:
        userhome = os.environ['USERPROFILE']
    elif not 'HOMEPATH' in os.environ:
        return path
    else:
        try:
            drive = os.environ['HOMEDRIVE']
        except KeyError:
            drive = ''
        userhome = join(drive, os.environ['HOMEPATH'])

    if i != 1: #~user
        userhome = join(dirname(userhome), path[1:i])

    return userhome + path[i:]

expanduser 本身不会出错。您需要检查程序中的这些环境变量，看看它们是否包含正确的值。

    import os
    for var in ('HOME', 'USERPROFILE', 'HOMEPATH', 'HOMEDRIVE'):
        print os.environ.get(var)

open 失败的一个可能原因是您尝试打开文件的目录不存在，或者您没有访问该文件的权限。