Python 调用系统API 实现程序自动以管理员权限运行的方法

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

核心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。

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