查看getpass模块

import getpass
getpass.getuser()
'kostya'

可用性:Unix, Windows

p.s.每个评论下面“这个函数查看各种环境变量的值来确定用户名。因此，该函数不应用于访问控制目的(或任何其他目的，因为它允许任何用户模仿任何其他用户)。”

查看getpass模块

import getpass
getpass.getuser()
'kostya'

可用性:Unix, Windows

p.s.每个评论下面“这个函数查看各种环境变量的值来确定用户名。因此，该函数不应用于访问控制目的(或任何其他目的，因为它允许任何用户模仿任何其他用户)。”

窗口:

os.environ.get('USERNAME')

Linux:

os.environ.get('USER')

两个:

os.environ.get('USER', os.environ.get('USERNAME'))

请注意，用户可以修改环境变量，因此这是一个潜在的安全漏洞。使用这种方法，攻击者可以很容易地伪造他们的用户名。

Psutil提供了一种可移植的方法，它不像getpass解决方案那样使用环境变量。它不太容易出现安全问题，应该是目前公认的答案。

import psutil

def get_username():
    return psutil.Process().username()

在底层，它结合了unix的基于getpuid的方法和Windows的GetTokenInformation方法。

以上方法在我的案例中都不起作用(向下滚动到实际的解决方案)。 我得到的所有解决方案的问题是错误的用户名时运行命令与sudo:

psutil soulution:

$ python3
>>> import psutil
>>> psutil.Process().username()
'ubuntu' # OK!

$ sudo su
$ python3
>>> import psutil
>>> psutil.Process().username()
'root' # OK!

$ sudo python3
>>> import psutil
>>> psutil.Process().username()
'root' # WRONG, should be ubuntu!

getpass解决方案:

$ python3
>>> import getpass
>>> getpass.getuser()
'ubuntu' # OK!

$ sudo su
$ python3
>>> import getpass
>>> getpass.getuser()
'root' # OK!

$ sudo python3
>>> import getpass
>>> getpass.getuser()
'root' # WRONG, should be ubuntu!

PWD + os。getuid解决方案:

$ python3
>>> import os, pwd
>>> pwd.getpwuid( os.getuid() )[ 0 ]
'ubuntu' # OK!

$ sudo su
$ python3
>>> import os, pwd
>>> pwd.getpwuid( os.getuid() )[ 0 ]
'root' # OK!

$ sudo python3
>>> import getpass
>>> getpass.getuser()
'root' # WRONG, should be ubuntu!

操作系统。Getlogin的工作方式有点不同，但仍然是错误的:

$ python3
>>> import os
>>> os.getlogin()
'ubuntu' # OK!

$ sudo su
$ python3
>>> import os
>>> os.getlogin()
'ubuntu' # WRONG, should be root!


$ sudo python3
>>> import os
>>> os.getlogin()
'ubuntu' # OK!

操作系统。Getenv给出了相同的结果:

$ python3
>>> import os
>>> os.getenv('SUDO_USER', os.getenv('USER'))
'ubuntu' # OK!

$ sudo su
$ python3
>>> import os
>>> os.getenv('SUDO_USER', os.getenv('USER'))
'ubuntu' # WRONG, should be root!


$ sudo python3
>>> import os
>>> os.getenv('SUDO_USER', os.getenv('USER'))
'ubuntu' # OK!

在sudo python3情况下，切换SUDO_USER和USER会给出错误的结果。

实际解决方案(不可移植)

解决方案有点棘手，依赖于默认的根主目录位置，但适用于所有情况:

$ python3
>>> import os
>>> 'root' if os.path.expanduser('~') == '/root' else os.getenv('SUDO_USER', os.getenv('USER'))
'ubuntu' # OK!

$ sudo su
$ python3
>>> import os
>>> 'root' if os.path.expanduser('~') == '/root' else os.getenv('SUDO_USER', os.getenv('USER'))
'root' #  OK!

$ sudo python3
>>> import os
>>> 'root' if os.path.expanduser('~') == '/root' else os.getenv('SUDO_USER', os.getenv('USER'))
'ubuntu' # OK!

如果你需要这个来获取用户的home目录，下面可以被认为是可移植的(至少win32和linux)，标准库的一部分。

>>> os.path.expanduser('~')
'C:\\Documents and Settings\\johnsmith'

此外，你可以解析这样的字符串，只得到最后的路径组件(即。用户名)。

看:os.path.expanduser