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

时间:2010-06-01 01:00:11

标签: python windows registry privileges

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

是否有一种简单的方法可以检查当前用户是否可以访问注册表?我宁愿它干净地告诉用户他/她需要管理员权限而不是程序疯狂和崩溃,因为它试图访问受限区域。

在用户获得任何输入选项之前,我希望在程序启动后立即进行此检查。这需要什么代码?

编辑:如果不明显,则适用于Windows平台。

4 个答案:

答案 0 :(得分:7)

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

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

答案 1 :(得分:6)

使用pywin32,以下内容应该有效......:

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

if shell.IsUserAnAdmin():
   ...

是的,似乎pywin32支持Python 3。

答案 2 :(得分:4)

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

您可以使用pywin32或ctypes进行CreateProcess()调用。

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

答案 3 :(得分:1)

最简单的方法是尝试在开头更改密钥,也许是更改存根值 - 如果失败,请捕获错误并告诉用户。