我正在编写一个小程序作为Python 3.x中的自学项目.我的想法是让程序允许两个字段的文本输入到用户,然后将用户的输入插入两个特定注册表项的值.
是否有一种简单的方法来检查当前用户是否可以访问注册表?我宁愿它干净地告诉用户他/她需要管理员权限而不是程序疯狂和崩溃,因为它试图访问受限区域.
在用户获得任何输入选项之前,我希望在程序启动后立即进行此检查.这需要什么代码?
编辑:如果不明显,这适用于Windows平台.
我需要一个简单的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)
使用pywin32,类似下面的内容应该有效......:
import pythoncom import pywintypes import win32api from win32com.shell import shell if shell.IsUserAnAdmin(): ...
是的,似乎pywin32支持Python 3.