如何在执行期间获取python脚本的用户名

Question

假设有一个脚本sth.py

当我使用“root”帐户登录时。输入：

python sth.py

我想得到结果：'root'

当输入sudo -u work python sth.py时，我想得到结果：'work'

如何获得它？

谢谢〜

Answer 1

import getpass
getpass.getuser()

使用环境变量的解决方案通常不可移植。

Answer 2

在linux上，应设置环境变量LOGNAME和USER：

import os
print os.environ['LOGNAME']