ChatGPT解决这个技术问题 Extra ChatGPT

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

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

os.getenv("HOME")

但是,这在 Windows 上不起作用。什么是正确的跨平台方式来做到这一点?

这被标记为 How to find the real user home directory using python 的重复项,但我投票决定重新打开,因为此答案适用于 Python 3,而旧答案则不适用。

A
Arsen Khachaturyan

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

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

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

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

应该注意的是,如果用户登录到 Windows 上的域并将其配置文件主文件夹设置在活动目录中,那么这将报告映射的网络文件夹而不是本地主目录
如果有人只想要主目录而不是用户目录,您可以尝试查找用户目录 (ud) 并执行以下操作:hd=os.sep.join(ud.split(os.sep)[:-1])
我想知道为什么没有其他人在这个问题中提到它,但是如果您需要知道另一个用户的主目录在哪里,您可以使用 os.path.expanduser('~username')。可能只适用于 Linux。
@Dipyan 不,那是 pathlib 的 Python 2 反向移植。对于 Python 3,pathlib 是正确的。
结果是一样的。如果您通常使用 pathlib,您可能更喜欢 pathlib 解决方案(并省略 str 的调用)。如果您只想将路径作为字符串,它们都做同样的事情。
J
Jaeyoon Jeong

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

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

u
user56700

我知道这是一个旧线程,但我最近需要这个用于大型项目(Python 3.8)。它必须适用于任何主流操作系统,因此我选择了@Max 在评论中写的解决方案。

代码:

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 上对其进行了测试,这也可以。


您的答案与 dcolish 的答案有何不同?
T
Tammi

这并不真正符合问题的条件(它被标记为 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")