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

Question

我正在用 Python 在系统级的 Linux 中做一个项目。所以，我想知道如果我以普通用户身份运行我的代码并且如果我正在访问系统文件，那么它应该具有 root 权限，那么我如何提示输入 root 密码并以 super 用户身份运行更多代码。我想知道，如何以 super 用户身份运行 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 时间戳，以便下次运行脚本时再次需要密码。