我需要获取当前登录用户的主目录的位置。目前,我一直在Linux上使用以下:
os.getenv("HOME")
但是,这在Windows上不起作用。正确的跨平台方式是什么?
我需要获取当前登录用户的主目录的位置。目前,我一直在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())
我知道这是一个旧线程,但我最近需要这个大型项目(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
>>> Path.home()
WindowsPath('C:/Users/XXX')
这并不是真正的问题(它被标记为跨平台),但也许这对某些人来说是有用的。
如何获取有效用户的主目录(特定于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())