Linux下Python如何在用户登录后以桌面会话上下文启动进程？

1. 检测用户桌面会话的简便方法 #

无需依赖DBus，可通过Linux系统的/run/user/目录特性实现——登录桌面的用户会生成一个以其UID命名的目录，会话结束时该目录会被删除。结合X11/Wayland会话的标识文件，能准确判断用户是否登录桌面。

获取目标用户UID #

先通过用户名获取对应UID，避免硬编码：

import pwd

TARGET_USER = "username"

def get_user_uid(username):
    try:
        return pwd.getpwnam(username).pw_uid
    except KeyError:
        raise ValueError(f"用户 {username} 不存在")

等待用户登录桌面 #

循环检测会话目录及会话标识文件，每5秒检查一次以降低资源占用：

import os
import time

def wait_for_user_desktop_session(uid):
    run_dir = f"/run/user/{uid}"
    while True:
        # 检查用户会话目录存在
        if os.path.exists(run_dir):
            # 验证是否存在活跃的X11或Wayland会话
            xauth_path = os.path.join(run_dir, "gdm/Xauthority")
            wayland_display = os.path.join(run_dir, "wayland-0")
            if os.path.exists(xauth_path) or os.path.exists(wayland_display):
                return
        time.sleep(5)

2. 切换会话上下文并启动进程 #

要让进程在用户桌面会话中正常运行（支持GLFW调用），需设置核心环境变量并切换用户身份：

实现上下文切换与进程启动 #

import subprocess

def start_program_in_user_session(uid, username, program_path):
    # 获取用户家目录
    user_home = pwd.getpwuid(uid).pw_dir
    
    # 配置桌面会话必需的环境变量
    env = os.environ.copy()
    env.update({
        "DISPLAY": ":0",  # 多数桌面环境默认值，若不符可通过ps命令查询Xorg进程确认
        "XAUTHORITY": f"/run/user/{uid}/gdm/Xauthority",
        "HOME": user_home,
        "USER": username,
        "PATH": f"/usr/local/bin:/usr/bin:/bin:{user_home}/.local/bin"
    })
    
    # 切换到目标用户身份（需脚本以root权限运行）
    os.setuid(uid)
    
    # 启动目标程序，使用subprocess替代os.popen更安全
    subprocess.Popen(program_path, env=env, shell=False)

3. 完整脚本示例 #

import pwd
import os
import time
import subprocess

TARGET_USER = "username"
PROGRAM_TO_START = "/path/to/your/glfw-program"

def get_user_uid(username):
    try:
        return pwd.getpwnam(username).pw_uid
    except KeyError:
        raise ValueError(f"用户 {username} 不存在")

def wait_for_user_desktop_session(uid):
    run_dir = f"/run/user/{uid}"
    while True:
        if os.path.exists(run_dir):
            xauth_path = os.path.join(run_dir, "gdm/Xauthority")
            wayland_display = os.path.join(run_dir, "wayland-0")
            if os.path.exists(xauth_path) or os.path.exists(wayland_display):
                return
        time.sleep(5)

def start_program_in_user_session(uid, username, program_path):
    user_home = pwd.getpwuid(uid).pw_dir
    env = os.environ.copy()
    env.update({
        "DISPLAY": ":0",
        "XAUTHORITY": f"/run/user/{uid}/gdm/Xauthority",
        "HOME": user_home,
        "USER": username,
        "PATH": f"/usr/local/bin:/usr/bin:/bin:{user_home}/.local/bin"
    })
    os.setuid(uid)
    subprocess.Popen(program_path, env=env, shell=False)

if __name__ == "__main__":
    try:
        target_uid = get_user_uid(TARGET_USER)
        print(f"等待用户 {TARGET_USER} 登录桌面...")
        wait_for_user_desktop_session(target_uid)
        print(f"用户已登录，启动程序 {PROGRAM_TO_START}")
        start_program_in_user_session(target_uid, TARGET_USER, PROGRAM_TO_START)
    except Exception as e:
        print(f"错误: {str(e)}")

注意事项 #

• 脚本需以root权限运行（开机自动进程通常由systemd以root身份启动）
• 若用户使用Wayland会话，可添加WAYLAND_DISPLAY: "wayland-0"到环境变量中
• 若DISPLAY值不是:0，可通过ps aux | grep Xorg命令查询Xorg进程的参数确认准确值
• 优先使用subprocess.Popen而非os.popen，避免shell注入风险

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