在 Python 中获取主目录的正确跨平台方法是什么？

Question

我需要获取当前登录用户的主目录位置。目前，我一直在 Linux 上使用以下内容：

os.getenv("HOME")

但是，这在 Windows 上不起作用。正确的跨平台方法是什么？

Answer 1

这并不真正符合问题的条件（它被标记为cross-platform），但也许这对某人有用。

如何获取有效用户的主目录（Linux特定）。

假设您正在编写一个安装程序脚本或其他一些解决方案，要求您在某些本地用户下执行某些操作。您很可能会在安装程序脚本中通过更改有效用户来完成此操作，但 os.path.expanduser("~") 仍将返回 /root。

参数需要有想要的用户名：

os.path.expanduser(f"~{USERNAME}/")

请注意，在不更改 EUID 的情况下，上述方法可以正常工作，但如果前面描述的场景适用，下面的示例显示了如何使用它：

import os
import pwd
import grp

class Identity():

    def __init__(self, user: str, group: str = None):
        self.uid = pwd.getpwnam(user).pw_uid
        if not group:
            self.gid = pwd.getpwnam(user).pw_gid
        else:
            self.gid = grp.getgrnam(group).gr_gid

    def __enter__(self):
        self.original_uid = os.getuid()
        self.original_gid = os.getgid()
        os.setegid(self.uid)
        os.seteuid(self.gid)

    def __exit__(self, type, value, traceback):
        os.seteuid(self.original_uid)
        os.setegid(self.original_gid)

if __name__ == '__main__':

    with Identity("hedy", "lamarr"):
        homedir = os.path.expanduser(f"~{pwd.getpwuid(os.geteuid())[0]}/")
        with open(os.path.join(homedir, "install.log"), "w") as file:
            file.write("Your home directory contents have been altered")

Answer 2

我发现 pathlib 模块也支持这个。

from pathlib import Path
>>> Path.home()
WindowsPath('C:/Users/XXX')

Answer 3

我知道这是一个旧线程，但我最近需要这个用于大型项目（Python 3.8）。它必须适用于任何主流操作系统，因此我采用了@Max 在 cmets 中编写的解决方案。

代码：

import os
print(os.path.expanduser("~"))

输出窗口：

PS C:\Python> & C:/Python38/python.exe c:/Python/test.py
C:\Users\mXXXXX

输出 Linux (Ubuntu)：

rxxx@xx:/mnt/c/Python$ python3 test.py
/home/rxxx

我还在 Python 2.7.17 上对其进行了测试，结果也可以。

Answer 4

您想使用os.path.expanduser。
这将确保它适用于所有平台：

from os.path import expanduser
home = expanduser("~")

如果您使用的是 Python 3.5+，则可以使用 pathlib.Path.home():

from pathlib import Path
home = str(Path.home())