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

Самый простой способ - это изменить ключ в начале, возможно, на значение заглушки - если это не удается, поймайте ошибку и сообщите пользователю.

Answer 3

С pywin32, что-то вроде должно работать ...:

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

if shell.IsUserAnAdmin(): 
    ... 

И да, это кажется pywin32 делает поддержка Python 3.

Answer 4

мне нужен простой, Windows/Posix решение для тестирования если пользователь имеет права root/admin для файловой системы без установки стороннего решения. Я понимаю, что есть уязвимости при использовании переменных среды, но они соответствуют моей цели. Возможно, можно расширить этот подход для чтения/записи реестра.

Я тестировал это с помощью Python 2.6/2.7 на WinXP, WinVista и Wine. Если кто-нибудь знает, что это не будет работать на 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)