Python-检查用户是否具有管理员权限

Question

我正在Python 3.x中编写一个小程序作为一个自学项目，我的想法是让程序允许用户输入两个文本条目字段，然后将用户的输入插入到两个特定注册表项的值中。

有没有一种简单的方法来检查当前用户是否可以访问注册表？我宁愿它清楚地告诉用户，他/她需要管理员权限，而不是让程序发疯和崩溃，因为它试图访问一个受限制的区域。

我希望在程序启动后，在给用户提供任何输入选项之前，对其进行检查。这需要什么代码？

编辑：如果不明显，这是针对Windows平台的。

Answer 1

这是一篇关于how to determine if an application requires elevated privileges的短文。

您可以使用pywin32或ctypes执行CreateProcess（）调用。

我建议使用ctypes，因为它现在是python中的标准，这里有一个good example of using CreateProcess with ctypes。

Answer 2

使用pywin32时，应该可以使用以下方法…：

import pythoncom
import pywintypes
import win32api
from win32com.shell import shell

if shell.IsUserAnAdmin():
   ...

是的，pywin32似乎确实支持Python 3。

Answer 3

我需要一个简单的Windows/Posix解决方案来测试用户是否对文件系统具有根/管理员权限，而不需要安装第三方解决方案。我知道依赖环境变量时存在漏洞，但它们符合我的目的。可以扩展这种方法来读/写注册表。

我在WinXP、WinVista和Wine上用Python 2.6/2.7进行了测试。如果有人知道这对Pyton 3.x和/或Win7不起作用，请告知。谢谢。

def has_admin():
    import os
    if os.name == 'nt':
        try:
            # only windows users with admin privileges can read the C:\windows\temp
            temp = os.listdir(os.sep.join([os.environ.get('SystemRoot','C:\\windows'),'temp']))
        except:
            return (os.environ['USERNAME'],False)
        else:
            return (os.environ['USERNAME'],True)
    else:
        if 'SUDO_USER' in os.environ and os.geteuid() == 0:
            return (os.environ['SUDO_USER'],True)
        else:
            return (os.environ['USERNAME'],False)