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

os.getenv("HOME")

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


当前回答

这并不是真正的问题(它被标记为跨平台),但也许这对某些人来说是有用的。

如何获取有效用户的主目录(特定于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")

其他回答

这可以使用标准库的一部分pathlib来完成,它将路径视为具有方法的对象,而不是字符串。

Path.expanduser () Path.home ()

from pathlib import Path

home: str = str(Path('~').expanduser())

我知道这是一个旧线程,但我最近需要这个大型项目(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上测试了它,结果也可以。

你需要使用os.path.expanduser。 这将确保它适用于所有平台:

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

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

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

这并不是真正的问题(它被标记为跨平台),但也许这对某些人来说是有用的。

如何获取有效用户的主目录(特定于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")

我发现pathlib模块也支持这一点。

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