Python 实现程序自动以管理员权限运行的方法

由于Windows的安全机制,Python写的脚本缺少了管理员权限,运行就会受到一些限制。
文章介绍Python 脚本自动以管理员权限运行的方法,
也就是如果脚本不是以管理员运行,就自动提升到管理员权限。

目录

    • 核心API函数
    • 代码实现(1)
    • 代码实现(2)

核心API函数

将python提升到管理员权限运行需要调用ShellExecute这个API函数。
ShellExecute函数的用法如下(这里仅作了解 ):

ShellExecute(hWnd,Operation,FileName,Parameters,Directory,ShowCmd)
hWnd: 整数 (指定父窗口句柄,如果没有可用None代替)
Operation: 字符串 (指定动作, 譬如: open、runas、print、edit、explore、find)
FileName: 字符串 (指定要打开的文件或程序)
Parameters: 字符串 (给要打开的程序指定参数)
Directory: 字符串 (缺省目录)
ShowCmd: 整数 (程序启动后的窗口显示方式,
正常为1、最小化为2、最大化为3)

代码实现(1)

使用Python自带的ctypes模块,调用API函数的代码:

import ctypes, sys

def is_admin():
    try:
        return ctypes.windll.shell32.IsUserAnAdmin()
    except:
        return False

if is_admin():
    # 主程序写在这里
else:
    # 以管理员权限重新运行程序
    ctypes.windll.shell32.ShellExecuteW(None,"runas", sys.executable, __file__, None, 1)

如果是Python 2,把最后一行改成这样:

ctypes.windll.shell32.ShellExecuteW(None, u"runas", unicode(sys.executable), unicode(__file__), None, 1)

如果使用pyinstaller等库打包python程序,则应将ShellExecuteW的第四个参数替换为""

另外,在ShellExecuteW的第4个参数中可加入其他的命令行参数,
在第6个参数中可指定窗口显示方式,如最小化(为2)、最大化(为3)等。

本方法优点:

  • 不需要外部库,仅使用自带的ctypes模块。
  • 适用于Python 2和Python 3。

本方法缺点:

在命令行窗口(cmd)中运行python时,会另外打开一个命令行窗口,造成输出的消息一闪而过。
(不过,据了解这似乎是Windows自身的安全机制造成的,而不是程序的bug,可能是用于防止漏洞)

代码实现(2)

相比于Python自带的ctypes库,win32api库更加广为人知、使用方便。
使用pip的安装方法:pip install pywin32
使用win32api模块的代码:

import sys, os, win32api

def is_admin():
    # 由于win32api中没有IsUserAnAdmin函数,所以用了这种方法
    try:
        # 在c:\windows目录下新建一个文件test01.txt
        testfile=os.path.join(os.getenv("windir"),"test01.txt")
        open(testfile,"w").close()
    except OSError: # 不成功
        return False
    else: # 成功
        os.remove(testfile) # 删除文件
        return True

print(is_admin())
if is_admin():
    # 主程序写在这里
else:
    # 以管理员权限重新运行程序
    win32api.ShellExecute(None,"runas", sys.executable, __file__, None, 1)

如果是Python 2,将except OSError这一行改为except IOError
本方法优点:

  • 适用于Python 2和Python 3。

如果读者有更好的实现程序自动以管理员权限运行的方法,欢迎在评论区补充!点赞收藏

你可能感兴趣的:(Python,python,系统安全)