How to run script with elevated privilege on windows

Я пишу приложение pyqt, которое требует выполнения задачи администратора. Я бы предпочел запустить свой скрипт с повышенными привилегиями. Я знаю, что этот вопрос задают много раз в SO или на другом форуме. Но решение, которое предлагают люди, состоит в том, чтобы взглянуть на этот вопрос SO Запросить повышение прав доступа из скрипта Python?

Однако я не могу выполнить пример кода, приведенный по ссылке. Я поместил этот код поверх основного файла и попытался выполнить его.

import os
import sys
import win32com.shell.shell as shell
ASADMIN = 'asadmin'

if sys.argv[-1] != ASADMIN:
    script = os.path.abspath(sys.argv[0])
    params = ' '.join([script] + sys.argv[1:] + [ASADMIN])
    shell.ShellExecuteEx(lpVerb='runas', lpFile=sys.executable, lpParameters=params)
    sys.exit(0)
print "I am root now."

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

Обновление от 19-02-2023

Обновление приведенного ниже скрипта теперь доступно в виде пакета Python того же автора. Вы можете установить его из PyPI, который находится по адресу https://pypi.org/project/pyuac /, а исходный код / домашняя страница находятся по адресу https://github.com/Preston-Landers/pyuac. Установите его с помощью:

pip install pyuac
pip install pypiwin32

Прямое использование пакета заключается:

import pyuac

def main():
    print("Do stuff here that requires being run as an admin.")
    # The window will disappear as soon as the program exits!
    input("Press enter to close the window. >")

if __name__ == "__main__":
    if not pyuac.isUserAdmin():
        print("Re-launching as admin!")
        pyuac.runAsAdmin()
    else:        
        main()  # Already an admin here.

Или, если вы хотите использовать декоратор:

from pyuac import main_requires_admin

@main_requires_admin
def main():
    print("Do stuff here that requires being run as an admin.")
    # The window will disappear as soon as the program exits!
    input("Press enter to close the window. >")

if __name__ == "__main__":
    main()

Оригинальный ответ

Спасибо всем за ваш ответ. Мой скрипт заработал с модулем / скриптом, написанным Престоном Ландерсом в 2010 году. После двух дней работы в Интернете я смог найти скрипт. Он был глубоко спрятан в списке рассылки pywin32. С помощью этого скрипта проще проверить, является ли пользователь администратором, и, если нет, запросить права UAC / admin. Он предоставляет выходные данные в отдельных окнах для отображения того, что делает код. Пример использования кода также включен в скрипт. Для удобства всех, кто ищет UAC в Windows, взгляните на этот код. Его можно использовать примерно так из вашего основного скрипта:-

import admin

if not admin.isUserAdmin():
    admin.runAsAdmin()

Фактический код (в модуле):-

#!/usr/bin/env python
# -*- coding: utf-8; mode: python; py-indent-offset: 4; indent-tabs-mode: nil -*-
# vim: fileencoding=utf-8 tabstop=4 expandtab shiftwidth=4

# (C) COPYRIGHT © Preston Landers 2010
# Released under the same license as Python 2.6.5

 
import sys, os, traceback, types
 
def isUserAdmin():
   
    if os.name == 'nt':
        import ctypes
        # WARNING: requires Windows XP SP2 or higher!
        try:
            return ctypes.windll.shell32.IsUserAnAdmin()
        except:
            traceback.print_exc()
            print "Admin check failed, assuming not an admin."
            return False
    elif os.name == 'posix':
        # Check for root on Posix
        return os.getuid() == 0
    else:
        raise RuntimeError, "Unsupported operating system for this module: %s" % (os.name,)
   
def runAsAdmin(cmdLine=None, wait=True):
 
    if os.name != 'nt':
        raise RuntimeError, "This function is only implemented on Windows."
   
    import win32api, win32con, win32event, win32process
    from win32com.shell.shell import ShellExecuteEx
    from win32com.shell import shellcon
   
    python_exe = sys.executable
 
    if cmdLine is None:
        cmdLine = [python_exe] + sys.argv
    elif type(cmdLine) not in (types.TupleType,types.ListType):
        raise ValueError, "cmdLine is not a sequence."
    cmd = '"%s"' % (cmdLine[0],)
    # XXX TODO: isn't there a function or something we can call to massage command line params?
    params = " ".join(['"%s"' % (x,) for x in cmdLine[1:]])
    cmdDir = ''
    showCmd = win32con.SW_SHOWNORMAL
    #showCmd = win32con.SW_HIDE
    lpVerb = 'runas'  # causes UAC elevation prompt.
   
    # print "Running", cmd, params
 
    # ShellExecute() doesn't seem to allow us to fetch the PID or handle
    # of the process, so we can't get anything useful from it. Therefore
    # the more complex ShellExecuteEx() must be used.
 
    # procHandle = win32api.ShellExecute(0, lpVerb, cmd, params, cmdDir, showCmd)
 
    procInfo = ShellExecuteEx(nShow=showCmd,
                              fMask=shellcon.SEE_MASK_NOCLOSEPROCESS,
                              lpVerb=lpVerb,
                              lpFile=cmd,
                              lpParameters=params)
 
    if wait:
        procHandle = procInfo['hProcess']    
        obj = win32event.WaitForSingleObject(procHandle, win32event.INFINITE)
        rc = win32process.GetExitCodeProcess(procHandle)
        #print "Process handle %s returned code %s" % (procHandle, rc)
    else:
        rc = None
 
    return rc
 
def test():
    rc = 0
    if not isUserAdmin():
        print "You're not an admin.", os.getpid(), "params: ", sys.argv
        #rc = runAsAdmin(["c:\\Windows\\notepad.exe"])
        rc = runAsAdmin()
    else:
        print "You are an admin!", os.getpid(), "params: ", sys.argv
        rc = 0
    x = raw_input('Press Enter to exit.')
    return rc
 
 
if __name__ == "__main__":
    sys.exit(test())

в комментариях к ответу, из которого вы взяли код, кто-то говорит, что ShellExecuteEx не отправляет свой стандартный вывод обратно в исходную оболочку. таким образом, вы не увидите "Я теперь root", хотя код, вероятно, работает нормально.

вместо того, чтобы что-то печатать, попробуйте записать в файл:

import os
import sys
import win32com.shell.shell as shell
ASADMIN = 'asadmin'

if sys.argv[-1] != ASADMIN:
    script = os.path.abspath(sys.argv[0])
    params = ' '.join([script] + sys.argv[1:] + [ASADMIN])
    shell.ShellExecuteEx(lpVerb='runas', lpFile=sys.executable, lpParameters=params)
    sys.exit(0)
with open("somefilename.txt", "w") as out:
    print >> out, "i am root"

а затем загляните в файл.

Я нашел очень простое решение этой проблемы.

1. Создайте ярлык для python.exe


2. Измените целевой ярлык на что-то вроде C:\xxx\...\python.exe your_script.py


3. Нажмите "продвинуть ..." на панели свойств ярлыка и выберите опцию "запуск от имени администратора"

Я не уверен, подходят ли значения этих параметров, поскольку я использую китайскую версию Windows.

Вот решение, для которого требовался только модуль ctypes. Поддержка программы в оболочке pyinstaller.

#!python
# coding: utf-8
import sys
import ctypes

def run_as_admin(argv=None, debug=False):
    shell32 = ctypes.windll.shell32
    if argv is None and shell32.IsUserAnAdmin():
        return True

    if argv is None:
        argv = sys.argv
    if hasattr(sys, '_MEIPASS'):
        # Support pyinstaller wrapped program.
        arguments = map(unicode, argv[1:])
    else:
        arguments = map(unicode, argv)
    argument_line = u' '.join(arguments)
    executable = unicode(sys.executable)
    if debug:
        print 'Command line: ', executable, argument_line
    ret = shell32.ShellExecuteW(None, u"runas", executable, argument_line, None, 1)
    if int(ret) <= 32:
        return False
    return None


if __name__ == '__main__':
    ret = run_as_admin()
    if ret is True:
        print 'I have admin privilege.'
        raw_input('Press ENTER to exit.')
    elif ret is None:
        print 'I am elevating to admin privilege.'
        raw_input('Press ENTER to exit.')
    else:
        print 'Error(ret=%d): cannot elevate privilege.' % (ret, )