如何从使用 pkexec 运行的脚本中获取当前用户名？

Question

I'm executing a python script as root with pkexec and I'm using working_dir = os.getenv('HOME') to get the username but it always returns root instead of test1 which is the current user.

How can I get the user that ran pkexec instead?

Script is located in /usr/bin if that information is any use.

Answer 1

The sudo man page describes the environment variables which hold the information of the user who invoked it.

import os
print os.environ["SUDO_USER"]

Answer 2

Thanks to that other guy, in the end, below is what I did to get the current user:

import os
import pwd

user = pwd.getpwuid(int(os.environ["PKEXEC_UID"])).pw_name
working_dir = '/home/' + user

Answer 3

For those looking for a pure bash solution, this finds the user run with sudo or pkexec:

if test -z "$SUDO_USER"
then
    USER=$(getent passwd $PKEXEC_UID | cut -d: -f1)
else
    USER=$SUDO_USER
fi