python 脚本中的身份验证以 root 身份运行

Question

我正在 Linux 上用 Python 做一个系统级的项目。 所以，我想知道我是否以普通用户身份运行我的代码并且 如果我正在访问系统文件，那么它应该具有 root 权限， 那么我如何提示输入 root 密码并以超级用户身份运行进一步的代码。 我想知道如何以超级用户身份运行 python 脚本并提示密码。

任何帮助将不胜感激。 先感谢您..

Answer 1

您可以做的另一件事是，如果脚本未以 root 身份执行，则让脚本自动调用 sudo：

import os
import sys

euid = os.geteuid()
if euid != 0:
    print "Script not started as root. Running sudo.."
    args = ['sudo', sys.executable] + sys.argv + [os.environ]
    # the next line replaces the currently-running process with the sudo
    os.execlpe('sudo', *args)

print 'Running. Your euid is', euid

输出：

Script not started as root. Running sudo..
[sudo] password for bob:
Running. Your euid is 0

使用 sudo -k 进行测试，以清除 sudo 时间戳，以便下次脚本运行时运行它将再次要求密码。

Answer 2

import os
euid = os.geteuid() 
if euid != 0:
  raise EnvironmentError, "need to be root"
  exit()

这不会在脚本中间提示，而是强制用户使用 sudo 或以 root 身份重新运行它