以root运行Python脚本，如何正确写入postgres用户的.bash_profile？

问题原因 #

你用subprocess.getoutput("sudo -u postgres echo ~")得到/root，是因为~在root的shell环境中就被展开了，并没有交给postgres用户的shell处理，所以拿到的是root的主目录。

正确解决方案 #

方案1：用Python内置模块直接获取（推荐） #

不需要调用外部命令，直接通过pwd模块读取系统用户数据库，高效且可靠：

import pwd
import subprocess

# 获取postgres用户的主目录
postgres_home = pwd.getpwnam('postgres').pw_dir
# 拼接.bash_profile路径
bash_profile_path = f"{postgres_home}/.bash_profile"

# 写入配置示例（以添加PATH为例）
content_to_add = "export PATH=$PATH:/usr/pgsql-15/bin"
subprocess.run(
    ["sudo", "-u", "postgres", "bash", "-c", f'echo "{content_to_add}" >> "{bash_profile_path}"'],
    check=True
)

方案2：修正shell命令，让~在postgres用户的shell中展开 #

通过指定bash解释器执行命令，确保~由postgres用户的shell处理：

import subprocess

# 获取正确的postgres主目录
postgres_home = subprocess.getoutput('sudo -u postgres bash -c "echo ~"').strip()
bash_profile_path = f"{postgres_home}/.bash_profile"

# 写入配置示例
subprocess.run(
    ["sudo", "-u", "postgres", "sh", "-c", f'echo "export PATH=$PATH:/usr/pgsql-15/bin" >> "{bash_profile_path}"'],
    check=True
)

方案3：用getent命令读取用户信息（跨平台兼容） #

getent命令可以直接读取系统用户数据库，提取主目录字段：

import subprocess

# 提取postgres用户的第6个字段（主目录）
postgres_home = subprocess.getoutput('getent passwd postgres | cut -d: -f6').strip()
bash_profile_path = f"{postgres_home}/.bash_profile"

# 写入配置示例
subprocess.run(
    ["sudo", "-u", "postgres", "echo", 'export PATH=$PATH:/usr/pgsql-15/bin', ">>", bash_profile_path],
    check=True
)

注意事项 #

• 写入.bash_profile时，必须以postgres用户身份执行，避免权限问题，所以所有写入操作都要加上sudo -u postgres。
• 如果.bash_profile不存在，上述命令会自动创建该文件。

内容的提问来源于stack exchange，提问作者madrarua