4

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

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

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

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

4

4 回答 4

12

我需要一个简单的 Windows/Posix 解决方案来测试用户是否具有文件系统的 root/admin 权限，而无需安装第 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)
于 2012-05-20T07:43:42.850 回答
7

使用pywin32，类似以下内容应该可以工作......：

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

if shell.IsUserAnAdmin():
   ...

是的，看起来 pywin32 确实支持 Python 3。

于 2010-06-01T01:14:42.557 回答
4

这是一篇关于如何确定应用程序是否需要提升权限的简短文章。

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

我建议使用 ctypes，因为它现在在 python 中是标准的，这里有一个使用 CreateProcess 和 ctypes 的好例子。

于 2010-06-01T01:08:10.917 回答
1

最直接的方法是尝试在开始时更改密钥，可能是存根值 - 如果失败，捕获错误并告诉用户。

于 2010-06-01T01:08:17.893 回答