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

Question

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

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

我希望它在程序启动后立即进行此检查，然后再为用户提供任何输入选项。这需要什么代码？

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

Answer 1

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

我在 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)