我需要获取当前登录用户的主目录的位置。目前,我一直在 Linux 上使用以下内容:
os.getenv("HOME")
但是,这在 Windows 上不起作用。正确的跨平台方法是什么?
您想使用 os.path.expanduser。
这将确保它适用于所有平台:
from os.path import expanduser
home = expanduser("~")
如果您使用的是 Python 3.5+,您可以使用 pathlib.Path.home():
from pathlib import Path
home = str(Path.home())
但通常最好不要将
Path.home()
转换为字符串。这样使用更自然:
with open(Path.home() / ".ssh" / "known_hosts") as f:
lines = f.readlines()
我发现pathlib模块也支持这个。
from pathlib import Path
>>> Path.home()
WindowsPath('C:/Users/XXX')
我知道这是一个旧线程,但我最近需要它来进行大型项目(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 上测试了它,它也有效。
pathlib
来完成,它是标准库的一部分,并将路径视为带有方法的对象,而不是字符串。
from pathlib import Path
home: str = str(Path('~').expanduser())
这并不真正符合这个问题(它被标记为
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")